refactor(core): select discovery backend from configured type

pull/3772/head
dsx137 2 months ago
parent a2553f2028
commit d11ea11552

3
.gitignore vendored

@ -388,3 +388,6 @@ Sessionx.vim
# End of https://www.toptal.com/developers/gitignore/api/go,git,vim,tags,test,emacs,backup,jetbrains # End of https://www.toptal.com/developers/gitignore/api/go,git,vim,tags,test,emacs,backup,jetbrains
.idea .idea
dist/ dist/
.omo
.sisyphus

@ -15,6 +15,7 @@
package discoveryregister package discoveryregister
import ( import (
"strings"
"time" "time"
"google.golang.org/grpc" "google.golang.org/grpc"
@ -39,16 +40,14 @@ func NewDiscoveryRegister(discovery *config.Discovery, watchNames []string) (dis
} }
switch discoveryType { switch discoveryType {
case config.KUBERNETES: case config.KUBERNETES:
namespace := discovery.Kubernetes.Namespace for i := range watchNames {
if namespace == "" { watchNames[i] = strings.Split(watchNames[i], ":")[0]
namespace = "default"
} }
return kubernetes.NewConnManager( return kubernetes.NewConnManager(discovery.Kubernetes.Namespace, watchNames,
namespace,
watchNames,
grpc.WithDefaultCallOptions( grpc.WithDefaultCallOptions(
grpc.MaxCallSendMsgSize(20*1024*1024), grpc.MaxCallSendMsgSize(1024*1024*20),
), ),
grpc.WithDefaultServiceConfig(`{"loadBalancingPolicy":"round_robin"}`),
) )
case config.ETCD: case config.ETCD:
return etcd.NewSvcDiscoveryRegistry( return etcd.NewSvcDiscoveryRegistry(

Loading…
Cancel
Save