162 lines
3.6 KiB
Go
162 lines
3.6 KiB
Go
package consul
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"time"
|
|
|
|
consulapi "github.com/hashicorp/consul/api"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
const (
|
|
consulAddr string = "127.0.0.1:8500"
|
|
)
|
|
|
|
var _client *consulapi.Client
|
|
|
|
func getClient() *consulapi.Client {
|
|
if _client != nil {
|
|
return _client
|
|
}
|
|
|
|
// 创建连接consul服务配置
|
|
config := consulapi.DefaultConfig()
|
|
config.Address = consulAddr
|
|
client, err := consulapi.NewClient(config)
|
|
if err != nil {
|
|
log.WithFields(log.Fields{
|
|
"func": "getClient",
|
|
}).Errorf("NewClient: %v", err)
|
|
return nil
|
|
}
|
|
_client = client
|
|
return client
|
|
}
|
|
|
|
//Register 注册服务到consul
|
|
func Register(name string, addr string, port int, tags ...string) {
|
|
client := getClient()
|
|
if client == nil {
|
|
return
|
|
}
|
|
// 创建注册到consul的服务到
|
|
registration := new(consulapi.AgentServiceRegistration)
|
|
registration.ID = fmt.Sprintf("%s-%s:%d", name, addr, port)
|
|
registration.Name = name
|
|
registration.Port = port
|
|
registration.Tags = tags
|
|
registration.Address = addr
|
|
|
|
// 增加consul健康检查回调函数
|
|
check := new(consulapi.AgentServiceCheck)
|
|
check.GRPC = fmt.Sprintf("%s:%d", registration.Address, registration.Port)
|
|
check.Timeout = "5s"
|
|
check.Interval = "5s"
|
|
check.DeregisterCriticalServiceAfter = "30s" // 故障检查失败30s后 consul自动将注册服务删除
|
|
registration.Check = check
|
|
|
|
// 注册服务到consul
|
|
if err := client.Agent().ServiceRegister(registration); err != nil {
|
|
log.WithFields(log.Fields{
|
|
"func": "Register",
|
|
}).Errorf("ServiceRegister: %v", err)
|
|
}
|
|
}
|
|
|
|
//DeRegister 取消consul注册的服务
|
|
func DeRegister(name string, addr string, port int) {
|
|
client := getClient()
|
|
if client == nil {
|
|
return
|
|
}
|
|
client.Agent().ServiceDeregister(fmt.Sprintf("%s-%s:%d", name, addr, port))
|
|
}
|
|
|
|
//FindNode 查找节点
|
|
func FindNode(servicename string) (*consulapi.AgentService, error) {
|
|
client := getClient()
|
|
if client == nil {
|
|
return nil, nil
|
|
}
|
|
services, _, err := client.Health().Service(servicename, "", true, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
l := len(services)
|
|
if l == 0 {
|
|
return nil, nil
|
|
}
|
|
if l == 1 {
|
|
return services[0].Service, nil
|
|
}
|
|
|
|
r := rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
return services[r.Intn(l)%l].Service, nil
|
|
}
|
|
|
|
//FindServer 从consul中发现服务
|
|
func FindServer() {
|
|
client := getClient()
|
|
if client == nil {
|
|
return
|
|
}
|
|
|
|
// 获取所有service
|
|
services, _ := client.Agent().Services()
|
|
for _, value := range services {
|
|
fmt.Println(value.Address)
|
|
fmt.Println(value.Port)
|
|
}
|
|
|
|
fmt.Println("=================================")
|
|
// 获取指定service
|
|
service, _, err := client.Agent().Service("demo", nil)
|
|
if err == nil {
|
|
fmt.Println(service.Address)
|
|
fmt.Println(service.Port)
|
|
}
|
|
|
|
s, _, _ := client.Health().Service("demo", "demo", true, nil)
|
|
for _, service := range s {
|
|
fmt.Println("service.Service.Address:", service.Service.Address, "service.Service.Port:", service.Service.Port)
|
|
}
|
|
}
|
|
|
|
//CheckHeath 健康检查
|
|
func CheckHeath(serviceid string) {
|
|
client := getClient()
|
|
if client == nil {
|
|
return
|
|
}
|
|
// 健康检查
|
|
a, b, _ := client.Agent().AgentHealthServiceByID(serviceid)
|
|
fmt.Println(a)
|
|
fmt.Println(b)
|
|
}
|
|
|
|
//KVTest test
|
|
func KVTest() {
|
|
client := getClient()
|
|
if client == nil {
|
|
return
|
|
}
|
|
|
|
// KV, put值
|
|
values := "test"
|
|
key := "go-consul-test/127.0.0.1:8100"
|
|
client.KV().Put(&consulapi.KVPair{Key: key, Flags: 0, Value: []byte(values)}, nil)
|
|
|
|
// KV get值
|
|
data, _, _ := client.KV().Get(key, nil)
|
|
fmt.Println(string(data.Value))
|
|
|
|
// KV list
|
|
datas, _, _ := client.KV().List("go", nil)
|
|
for _, value := range datas {
|
|
fmt.Println(value)
|
|
}
|
|
keys, _, _ := client.KV().Keys("go", "", nil)
|
|
fmt.Println(keys)
|
|
}
|