34 lines
966 B
Go
34 lines
966 B
Go
package consul
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"github.com/gogf/gf/contrib/registry/consul/v2"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/net/gsel"
|
|
"github.com/gogf/gf/v2/net/gsvc"
|
|
"github.com/gogf/gf/v2/util/grand"
|
|
)
|
|
|
|
func init() {
|
|
consulAddr := g.Cfg().MustGet(context.Background(), "consul.address").String()
|
|
registry, err := consul.New(consul.WithAddress(consulAddr))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
gsvc.SetRegistry(registry)
|
|
gsel.SetBuilder(gsel.NewBuilderRoundRobin())
|
|
}
|
|
func GetInstanceAddr(ctx context.Context, name string) (addr string, err error) {
|
|
watch, err := gsvc.GetRegistry().Watch(ctx, name)
|
|
service, err := watch.Proceed()
|
|
if err != nil || service == nil {
|
|
err = errors.New("获取customerService服务实例失败！")
|
|
return
|
|
}
|
|
maxService := grand.N(0, len(service)-1)
|
|
maxEndpoints := grand.N(0, len(service[maxService].GetEndpoints())-1)
|
|
addr = service[maxService].GetEndpoints()[maxEndpoints].String()
|
|
return
|
|
}
|