feat: refactor.

This commit is contained in:
tx7do
2024-05-06 08:23:05 +08:00
parent 8e16b4309c
commit 601a7e5545
8 changed files with 63 additions and 1063 deletions

View File

@@ -17,10 +17,6 @@ import (
etcdKratos "github.com/go-kratos/kratos/contrib/config/etcd/v2"
etcdClient "go.etcd.io/etcd/client/v3"
// consul
consulKratos "github.com/go-kratos/kratos/contrib/config/consul/v2"
consulApi "github.com/hashicorp/consul/api"
// nacos
nacosKratos "github.com/go-kratos/kratos/contrib/config/nacos/v2"
nacosClients "github.com/nacos-group/nacos-sdk-go/clients"
@@ -34,6 +30,8 @@ import (
k8sKratos "github.com/go-kratos/kratos/contrib/config/kubernetes/v2"
k8sUtil "k8s.io/client-go/util/homedir"
"github.com/tx7do/kratos-bootstrap/config/consul"
conf "github.com/tx7do/kratos-bootstrap/api/gen/go/conf/v1"
)
@@ -212,7 +210,7 @@ func NewRemoteConfigSource(c *conf.RemoteConfig) config.Source {
case ConfigTypeNacos:
return NewNacosConfigSource(c)
case ConfigTypeConsul:
return NewConsulConfigSource(c)
return consul.NewConfigSource(c)
case ConfigTypeEtcd:
return NewEtcdConfigSource(c)
case ConfigTypeApollo:
@@ -292,27 +290,6 @@ func NewEtcdConfigSource(c *conf.RemoteConfig) config.Source {
return source
}
// NewConsulConfigSource 创建一个远程配置源 - Consul
func NewConsulConfigSource(c *conf.RemoteConfig) config.Source {
cfg := consulApi.DefaultConfig()
cfg.Address = c.Consul.Address
cfg.Scheme = c.Consul.Scheme
cli, err := consulApi.NewClient(cfg)
if err != nil {
log.Fatal(err)
}
source, err := consulKratos.New(cli,
consulKratos.WithPath(getConfigKey(c.Consul.Key, true)),
)
if err != nil {
log.Fatal(err)
}
return source
}
// NewApolloConfigSource 创建一个远程配置源 - Apollo
func NewApolloConfigSource(c *conf.RemoteConfig) config.Source {
source := apolloKratos.NewSource(