mirror of
https://github.com/openimsdk/open-im-server.git
synced 2026-05-03 08:35:59 +08:00
Error code standardization
This commit is contained in:
@@ -0,0 +1,46 @@
|
||||
package rpc_server
|
||||
|
||||
import (
|
||||
"Open_IM/internal/common/network"
|
||||
"Open_IM/pkg/common/constant"
|
||||
"Open_IM/pkg/common/log"
|
||||
discoveryRegistry "Open_IM/pkg/discovery_registry"
|
||||
"github.com/OpenIMSDK/openKeeper"
|
||||
"net"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type RpcServer struct {
|
||||
Port int
|
||||
RegisterName string
|
||||
RegisterCenter discoveryRegistry.SvcDiscoveryRegistry
|
||||
}
|
||||
|
||||
func NewRpcServer(registerIPInConfig string, port int, registerName string, zkServers []string, zkRoot string) (*RpcServer, error) {
|
||||
log.NewPrivateLog(constant.LogFileName)
|
||||
s := &RpcServer{
|
||||
Port: port,
|
||||
RegisterName: registerName,
|
||||
}
|
||||
|
||||
zkClient, err := openKeeper.NewClient(zkServers, zkRoot, 10, "", "")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
registerIP, err := network.GetRpcRegisterIP(registerIPInConfig)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
err = zkClient.Register(s.RegisterName, registerIP, s.Port)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
s.RegisterCenter = zkClient
|
||||
return s, nil
|
||||
}
|
||||
|
||||
func GetTcpListen(listenIPInConfig string, port int) (net.Listener, string, error) {
|
||||
address := network.GetListenIP(listenIPInConfig) + ":" + strconv.Itoa(port)
|
||||
listener, err := net.Listen("tcp", address)
|
||||
return listener, address, err
|
||||
}
|
||||
Reference in New Issue
Block a user