This commit is contained in:
withchao 2023-03-28 14:40:51 +08:00
parent 330e8d0f05
commit 3c1b0e3994
2 changed files with 9 additions and 2 deletions

View File

@ -1,9 +1,11 @@
package main package main
import ( import (
"context"
"github.com/OpenIMSDK/Open-IM-Server/internal/rpc/third" "github.com/OpenIMSDK/Open-IM-Server/internal/rpc/third"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/cmd" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/cmd"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
) )
func main() { func main() {
@ -13,7 +15,9 @@ func main() {
if err := rpcCmd.Exec(); err != nil { if err := rpcCmd.Exec(); err != nil {
panic(err.Error()) panic(err.Error())
} }
if err := rpcCmd.StartSvr(config.Config.RpcRegisterName.OpenImThirdName, third.Start); err != nil { name := config.Config.RpcRegisterName.OpenImThirdName
log.ZInfo(context.Background(), "StartRpc", "name", name)
if err := rpcCmd.StartSvr(name, third.Start); err != nil {
panic(err.Error()) panic(err.Error())
} }
} }

View File

@ -4,6 +4,7 @@ import (
"context" "context"
"github.com/OpenIMSDK/Open-IM-Server/pkg/a2r" "github.com/OpenIMSDK/Open-IM-Server/pkg/a2r"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
"github.com/OpenIMSDK/Open-IM-Server/pkg/discoveryregistry" "github.com/OpenIMSDK/Open-IM-Server/pkg/discoveryregistry"
"github.com/OpenIMSDK/Open-IM-Server/pkg/errs" "github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/third" "github.com/OpenIMSDK/Open-IM-Server/pkg/proto/third"
@ -22,7 +23,9 @@ type Third struct {
} }
func (o *Third) client() (third.ThirdClient, error) { func (o *Third) client() (third.ThirdClient, error) {
conn, err := o.c.GetConn(config.Config.RpcRegisterName.OpenImThirdName) name := config.Config.RpcRegisterName.OpenImThirdName
log.ZInfo(context.Background(), "GetThirdClient", "name", name)
conn, err := o.c.GetConn(name)
if err != nil { if err != nil {
return nil, err return nil, err
} }