blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-06 04:15:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

conf

Browse Source
This commit is contained in:
wangchuxiao 2023-02-28 15:20:57 +08:00
parent 0547f25ecb
commit 4ac9943565
3 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pkg/common/config/config.go
View File

@ -22,6 +22,7 @@ const (
NotificationFileName = "notification.yaml"
ENV = "CONFIG_NAME"
DefaultPath = "../config/"
ConfKey = "conf"
)
var Config config
@ -525,8 +526,16 @@ func (c *config) initConfig(config interface{}, configName, configPath string) e
return c.unmarshalConfig(config, configPath)
}
func (c *config) Register(registry discoveryregistry.SvcDiscoveryRegistry) error {
registry
func (c *config) RegisterConf2Registry(registry discoveryregistry.SvcDiscoveryRegistry) error {
bytes, err := yaml.Marshal(Config)
if err != nil {
return err
}
return registry.RegisterConf2Registry(ConfKey, bytes)
}
func (c *config) GetConfFromRegistry(registry discoveryregistry.SvcDiscoveryRegistry) ([]byte, error) {
return registry.GetConfFromRegistry(ConfKey)
}
func InitConfig() error {

4
pkg/discoveryregistry/discovery_register.go
View File

@ -9,5 +9,7 @@ type SvcDiscoveryRegistry interface {
UnRegister() error
GetConns(serviceName string, opts ...grpc.DialOption) ([]*grpc.ClientConn, error)
GetConn(serviceName string, opts ...grpc.DialOption) (*grpc.ClientConn, error)
Re
RegisterConf2Registry(key string, conf []byte) error
GetConfFromRegistry(key string) ([]byte, error)
}

6
pkg/utils/lock.go Normal file
View File

@ -0,0 +1,6 @@
package utils
type DistributedLock interface {
Lock()
UnLock()
}