swagger聚合封装

This commit is contained in:
2025-11-28 08:52:22 +08:00
parent d8410fab37
commit 067c16b7ec
3 changed files with 121 additions and 2 deletions

View File

@@ -2,10 +2,12 @@ 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() {
@@ -21,3 +23,15 @@ func init() {
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
}