57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
|
package tidb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"go-common/library/conf/env"
|
||
|
|
"go-common/library/log"
|
||
|
|
"go-common/library/naming"
|
||
|
|
"go-common/library/naming/discovery"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _schema = "tidb://"
|
||
|
|
|
||
|
|
func (db *DB) nodeList() (nodes []string) {
|
||
|
|
var (
|
||
|
|
insMap map[string][]*naming.Instance
|
||
|
|
ins []*naming.Instance
|
||
|
|
ok bool
|
||
|
|
)
|
||
|
|
if insMap, ok = db.dis.Fetch(context.Background()); !ok {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if ins, ok = insMap[env.Zone]; !ok || len(ins) == 0 {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
for _, in := range ins {
|
||
|
|
for _, addr := range in.Addrs {
|
||
|
|
if strings.HasPrefix(addr, _schema) {
|
||
|
|
addr = strings.Replace(addr, _schema, "", -1)
|
||
|
|
nodes = append(nodes, addr)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
log.Info("tidb get %s instances(%v)", db.appid, nodes)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *DB) disc() (nodes []string) {
|
||
|
|
db.dis = discovery.Build(db.appid)
|
||
|
|
e := db.dis.Watch()
|
||
|
|
select {
|
||
|
|
case <-e:
|
||
|
|
nodes = db.nodeList()
|
||
|
|
case <-time.After(10 * time.Second):
|
||
|
|
panic("tidb init discovery err")
|
||
|
|
}
|
||
|
|
if len(nodes) == 0 {
|
||
|
|
panic(fmt.Sprintf("tidb %s no instance", db.appid))
|
||
|
|
}
|
||
|
|
go db.nodeproc(e)
|
||
|
|
log.Info("init tidb discvoery info successfully")
|
||
|
|
return
|
||
|
|
}
|