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)
|
|||
|
|
}
|