80 lines
3.5 KiB
Go
80 lines
3.5 KiB
Go
package discover
|
||
|
||
import (
|
||
"fmt"
|
||
|
||
"github.com/hashicorp/consul/api"
|
||
)
|
||
|
||
type ConsulConfig struct {
|
||
Client *api.Client `json:"client"` // consul client
|
||
ConsulAddress string `json:"consulAddress"` // consul 服务地址:IP+port
|
||
ServiceId string `json:"serviceId"` // 服务ID
|
||
ServiceName string `json:"serviceName"` // 服务名称
|
||
ServiceIP string `json:"serviceIP"` // 服务IP
|
||
ServicePort int `json:"servicePort"` // 服务端口
|
||
Tags []string `json:"tags"` // 服务标签列表
|
||
DeregisterCriticalServiceAfter int `json:"deregisterCriticalServiceAfter"` // 指定与服务关联的检查应在此时间之后注销
|
||
Interval int `json:"interval"` // 指定运行此检查的频率
|
||
Timeout int `json:"timeout"` // 在脚本、HTTP、TCP 或 gRPC 检查的情况下指定传出连接的超时时间
|
||
}
|
||
|
||
func NewConsulConfig(consulAddress string, serviceId string, serviceName string, serviceIP string, servicePort int, tags []string, deregisterCriticalServiceAfter int, interval int, timeout int) (*ConsulConfig, error) {
|
||
// 1.consul配置
|
||
config := api.DefaultConfig()
|
||
config.Address = consulAddress
|
||
// 2.client
|
||
client, err := api.NewClient(config)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return &ConsulConfig{
|
||
Client: client,
|
||
ConsulAddress: consulAddress,
|
||
ServiceId: serviceId,
|
||
ServiceName: serviceName,
|
||
ServiceIP: serviceIP,
|
||
ServicePort: servicePort,
|
||
Tags: tags,
|
||
DeregisterCriticalServiceAfter: deregisterCriticalServiceAfter,
|
||
Interval: interval,
|
||
Timeout: timeout,
|
||
}, nil
|
||
|
||
}
|
||
|
||
// ServiceRegister 服务注册
|
||
func (cf *ConsulConfig) ServiceRegister() (err error) {
|
||
// 注册器
|
||
reg := &api.AgentServiceRegistration{
|
||
ID: cf.ServiceId,
|
||
Name: cf.ServiceName,
|
||
Address: cf.ServiceIP,
|
||
Port: cf.ServicePort,
|
||
Tags: cf.Tags,
|
||
Check: &api.AgentServiceCheck{
|
||
Interval: fmt.Sprintf("%vs", cf.Interval), // 健康检查间隔
|
||
HTTP: fmt.Sprintf("http://%v:%v/health", cf.ServiceIP, cf.ServicePort), // HTTP 支持，执行健康检查的地址，service 会传到 Health.Check 函数中
|
||
Timeout: fmt.Sprintf("%vs", cf.Timeout), // 健康检查超时时间
|
||
DeregisterCriticalServiceAfter: fmt.Sprintf("%vs", cf.DeregisterCriticalServiceAfter), // 注销时间，相当于过期时间
|
||
Notes: "Consul check service health status.",
|
||
},
|
||
}
|
||
// 注册服务
|
||
err = cf.Client.Agent().ServiceRegister(reg)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// ServiceDeregister 服务注销
|
||
func (cf *ConsulConfig) ServiceDeregister() error {
|
||
return cf.Client.Agent().ServiceDeregister(cf.ServiceId)
|
||
}
|
||
|
||
// ServiceDiscover 服务发现
|
||
func (cf *ConsulConfig) ServiceDiscover(service string, tags []string, q *api.QueryOptions) ([]*api.CatalogService, *api.QueryMeta, error) {
|
||
return cf.Client.Catalog().ServiceMultipleTags(service, tags, q)
|
||
}
|