common/consul/consul.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() {
	consulCfg, err := g.Cfg().Get(context.Background(), "consul.address")
	if err != nil {
		panic(err)
	}
	consulAddr := consulCfg.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
}