You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.1 KiB
47 lines
1.1 KiB
3 years ago
|
package apollo
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/go-kratos/kratos/v2/config"
|
||
|
|
|
||
|
|
"github.com/apolloconfig/agollo/v4"
|
||
|
|
)
|
||
|
|
|
||
|
|
type apollo struct {
|
||
|
|
client *agollo.Client
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewSource start with config file in ENV
|
||
|
|
// Linux/Mac export AGOLLO_CONF=/a/conf.properties
|
||
|
|
// Windows set AGOLLO_CONF=c:/a/conf.properties
|
||
|
|
// more detail:https://github.com/apolloconfig/agollo/wiki/%E4%BD%BF%E7%94%A8%E6%8C%87%E5%8D%97#1312%E7%8E%AF%E5%A2%83%E5%8F%98%E9%87%8F%E6%8C%87%E5%AE%9A%E9%85%8D%E7%BD%AE%E6%96%87%E4%BB%B6
|
||
|
|
func NewSource() config.Source {
|
||
|
|
client, err := agollo.Start()
|
||
|
|
if err != nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return &apollo{client}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *apollo) load() []*config.KeyValue {
|
||
|
|
kv := make([]*config.KeyValue,e.client.GetDefaultConfigCache().EntryCount())
|
||
|
|
e.client.GetDefaultConfigCache().Range(func(key, value interface{}) bool {
|
||
|
|
kv = append(kv, &config.KeyValue{
|
||
|
|
Key: key.(string),
|
||
|
|
Value: []byte(value.(string)),
|
||
|
|
})
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
return kv
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *apollo) Load() (kv []*config.KeyValue, err error) {
|
||
|
|
return e.load(), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *apollo) Watch() (config.Watcher, error) {
|
||
|
|
w, err := NewWatcher(e)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return w, nil
|
||
|
|
}
|