package serf
|
|
import (
|
"context"
|
|
"serfNode/config"
|
"serfNode/logger"
|
|
"basic.com/syncdb.git"
|
)
|
|
var agent *syncdb.Agent
|
var nodeRole string
|
|
// 初始化节点
|
func InitAgent(ctx context.Context, conf config.SerfConfig) error {
|
if agent != nil {
|
agent.Stop()
|
agent = nil
|
}
|
|
var err error
|
password := conf.AuthKey + "123456"
|
|
serfConfig := syncdb.DefaultConfig()
|
serfConfig.RPCAddr = "0.0.0.0:7373"
|
serfConfig.RPCPort = 7373
|
serfConfig.BindAddr = "0.0.0.0:5000"
|
serfConfig.Ctx = ctx
|
|
if conf.BindAddr != "" {
|
serfConfig.BindAddr = conf.BindAddr
|
}
|
|
if conf.RPCAddr != "" {
|
serfConfig.RPCAddr = conf.RPCAddr
|
}
|
|
if conf.RPCPort != 0 {
|
serfConfig.RPCPort = conf.RPCPort
|
}
|
|
agent, err = syncdb.InitNode(conf.ClusterName, password, conf.NodeName, "./serf-daemon.json", serfConfig)
|
if err != nil {
|
logger.Error(err)
|
return err
|
}
|
|
agent.RegisterHandleEventFunc(HandleSerfEvent)
|
|
//logger.Debug("agent init done")
|
|
return nil
|
}
|
|
// 加入集群
|
func JoinCluster(addr string) error {
|
return agent.JoinByNodeAddrs([]string{addr})
|
}
|
|
// 退出集群
|
func LeaveCluster() error {
|
return agent.Leave()
|
}
|
|
func ClusterState() map[string]map[string]string {
|
return agent.Stats()
|
}
|
|
func Members() []syncdb.NodeInfo {
|
return agent.GetNodes()
|
}
|
|
func Switch2Master() error {
|
nodeRole = "master"
|
SetTag("role", "master")
|
|
return nil
|
}
|
|
func Switch2Slave() error {
|
nodeRole = "slave"
|
SetTag("role", "slave")
|
|
return nil
|
}
|
|
func SetTag(key, value string) error {
|
// 有一些预置的tag
|
conf := agent.SerfConfig()
|
conf.Tags[key] = value
|
|
return agent.SetTags(conf.Tags)
|
}
|
|
func IsMaster() bool {
|
return nodeRole == "master"
|
}
|
