mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-12-08 21:47:17 +08:00
abf8e37e78
* Code adaptation k8s: service discovery and registration adaptation, configuration adaptation * Initial submission of the help charts script for openim API * change the help charts script * change the help charts script * change helm chart codes * change dockerfiles script * change chart script:add configmap mounts * change chart script:change repository * change chart script:msggateway add one service * change config.yaml * roll back some config values * change chart script:change Ingress rule with a rewrite annotation * add mysql charts scrible * change chart script:add mysql.config.yaml * add nfs provisioner charts * change chart script:add nfs.config.yaml * add ingress-nginx charts * change chart script:add ingress-nginx.config.yaml * add redis &mongodb charts * add kafka&minio charts * change chart script:change redis.values.yaml * change chart script:add redis.config.yaml * change chart script:change redis.config.yaml * change chart script:change mongodb.value.yaml * change chart script:change mongodb.value.yaml * change chart script:add mongodb.config.yaml * change chart script:change minio.values.yaml * change chart script:add minio.config.yaml * change chart script:change kafka.values.yaml * change chart script:add kafka.config.yaml * change chart script:change services.config.yaml * bug fix:Delete websocket's Port restrictions * bug fix:change port value * change chart script:Submit a stable version script * fix bug:Implement option interface * fix bug:change K8sDR.Register * change config.yaml * change chats script:minio service add ingress * change chats script:minio service add ingress * change chats script:kafka.replicaCount=3& change minio.api ingress * delete change chats script * change config.yaml * change openim.yaml --------- Co-authored-by: lin.huang <lin.huang@apulis.com>
89 lines
2.5 KiB
Go
89 lines
2.5 KiB
Go
package discovery_register
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/OpenIMSDK/tools/discoveryregistry"
|
|
openkeeper "github.com/OpenIMSDK/tools/discoveryregistry/zookeeper"
|
|
"github.com/OpenIMSDK/tools/log"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/config"
|
|
"google.golang.org/grpc"
|
|
"time"
|
|
)
|
|
|
|
func NewDiscoveryRegister(envType string) (discoveryregistry.SvcDiscoveryRegistry, error) {
|
|
var client discoveryregistry.SvcDiscoveryRegistry
|
|
var err error
|
|
switch envType {
|
|
case "zookeeper":
|
|
client, err = openkeeper.NewClient(config.Config.Zookeeper.ZkAddr, config.Config.Zookeeper.Schema,
|
|
openkeeper.WithFreq(time.Hour), openkeeper.WithUserNameAndPassword(
|
|
config.Config.Zookeeper.Username,
|
|
config.Config.Zookeeper.Password,
|
|
), openkeeper.WithRoundRobin(), openkeeper.WithTimeout(10), openkeeper.WithLogger(log.NewZkLogger()))
|
|
case "k8s":
|
|
client, err = NewK8sDiscoveryRegister()
|
|
default:
|
|
client = nil
|
|
err = errors.New("envType not correct")
|
|
}
|
|
return client, err
|
|
}
|
|
|
|
type K8sDR struct {
|
|
options []grpc.DialOption
|
|
rpcRegisterAddr string
|
|
}
|
|
|
|
func NewK8sDiscoveryRegister() (discoveryregistry.SvcDiscoveryRegistry, error) {
|
|
return &K8sDR{}, nil
|
|
}
|
|
|
|
func (cli *K8sDR) Register(serviceName, host string, port int, opts ...grpc.DialOption) error {
|
|
cli.rpcRegisterAddr = serviceName
|
|
return nil
|
|
}
|
|
func (cli *K8sDR) UnRegister() error {
|
|
|
|
return nil
|
|
}
|
|
func (cli *K8sDR) CreateRpcRootNodes(serviceNames []string) error {
|
|
|
|
return nil
|
|
}
|
|
func (cli *K8sDR) RegisterConf2Registry(key string, conf []byte) error {
|
|
|
|
return nil
|
|
}
|
|
|
|
func (cli *K8sDR) GetConfFromRegistry(key string) ([]byte, error) {
|
|
|
|
return nil, nil
|
|
}
|
|
func (cli *K8sDR) GetConns(ctx context.Context, serviceName string, opts ...grpc.DialOption) ([]*grpc.ClientConn, error) {
|
|
|
|
conn, err := grpc.DialContext(ctx, serviceName, append(cli.options, opts...)...)
|
|
return []*grpc.ClientConn{conn}, err
|
|
}
|
|
func (cli *K8sDR) GetConn(ctx context.Context, serviceName string, opts ...grpc.DialOption) (*grpc.ClientConn, error) {
|
|
|
|
return grpc.DialContext(ctx, serviceName, append(cli.options, opts...)...)
|
|
}
|
|
func (cli *K8sDR) GetSelfConnTarget() string {
|
|
|
|
return cli.rpcRegisterAddr
|
|
}
|
|
func (cli *K8sDR) AddOption(opts ...grpc.DialOption) {
|
|
cli.options = append(cli.options, opts...)
|
|
}
|
|
func (cli *K8sDR) CloseConn(conn *grpc.ClientConn) {
|
|
conn.Close()
|
|
}
|
|
|
|
// do not use this method for call rpc
|
|
func (cli *K8sDR) GetClientLocalConns() map[string][]*grpc.ClientConn {
|
|
fmt.Println("should not call this function!!!!!!!!!!!!!!!!!!!!!!!!!")
|
|
return nil
|
|
}
|