This commit is contained in:
wangchuxiao 2023-06-01 18:36:20 +08:00
parent 228514b1bf
commit 7087bbe76d

View File

@ -18,15 +18,20 @@ func NewAuth(c discoveryregistry.SvcDiscoveryRegistry) *Auth {
panic(err) panic(err)
} }
log.ZInfo(context.Background(), "auth rpc conn", "conn", conn) log.ZInfo(context.Background(), "auth rpc conn", "conn", conn)
return &Auth{conn: conn} return &Auth{conn: conn, discov: c}
} }
type Auth struct { type Auth struct {
conn *grpc.ClientConn conn *grpc.ClientConn
discov discoveryregistry.SvcDiscoveryRegistry
} }
func (o *Auth) client(ctx context.Context) (auth.AuthClient, error) { func (o *Auth) client(ctx context.Context) (auth.AuthClient, error) {
return auth.NewAuthClient(o.conn), nil c, err := o.discov.GetConn(ctx, config.Config.RpcRegisterName.OpenImAuthName)
if err != nil {
return nil, err
}
return auth.NewAuthClient(c), nil
} }
func (o *Auth) UserRegister(c *gin.Context) { func (o *Auth) UserRegister(c *gin.Context) {