2019-05-31 00:11:13 +02:00
|
|
|
package config
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"sync"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/micro/go-micro/config/loader"
|
|
|
|
"github.com/micro/go-micro/config/loader/memory"
|
|
|
|
"github.com/micro/go-micro/config/reader"
|
|
|
|
"github.com/micro/go-micro/config/reader/json"
|
|
|
|
"github.com/micro/go-micro/config/source"
|
|
|
|
)
|
|
|
|
|
|
|
|
type config struct {
|
|
|
|
exit chan bool
|
|
|
|
opts Options
|
|
|
|
|
|
|
|
sync.RWMutex
|
|
|
|
// the current snapshot
|
|
|
|
snap *loader.Snapshot
|
|
|
|
// the current values
|
|
|
|
vals reader.Values
|
|
|
|
}
|
|
|
|
|
|
|
|
type watcher struct {
|
|
|
|
lw loader.Watcher
|
|
|
|
rd reader.Reader
|
|
|
|
path []string
|
|
|
|
value reader.Value
|
|
|
|
}
|
|
|
|
|
2020-01-19 10:31:02 +02:00
|
|
|
func newConfig(opts ...Option) (Config, error) {
|
2019-05-31 00:11:13 +02:00
|
|
|
options := Options{
|
|
|
|
Loader: memory.NewLoader(),
|
|
|
|
Reader: json.NewReader(),
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, o := range opts {
|
|
|
|
o(&options)
|
|
|
|
}
|
|
|
|
|
2020-01-19 10:31:02 +02:00
|
|
|
if err := options.Loader.Load(options.Source...); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2020-01-19 10:42:05 +02:00
|
|
|
snap, err := options.Loader.Snapshot()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
vals, err := options.Reader.Values(snap.ChangeSet)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2019-05-31 00:11:13 +02:00
|
|
|
|
|
|
|
c := &config{
|
|
|
|
exit: make(chan bool),
|
|
|
|
opts: options,
|
|
|
|
snap: snap,
|
|
|
|
vals: vals,
|
|
|
|
}
|
|
|
|
|
|
|
|
go c.run()
|
|
|
|
|
2020-01-19 10:31:02 +02:00
|
|
|
return c, nil
|
2019-05-31 00:11:13 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func (c *config) run() {
|
|
|
|
watch := func(w loader.Watcher) error {
|
|
|
|
for {
|
|
|
|
// get changeset
|
|
|
|
snap, err := w.Next()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
c.Lock()
|
|
|
|
|
|
|
|
// save
|
|
|
|
c.snap = snap
|
|
|
|
|
|
|
|
// set values
|
|
|
|
c.vals, _ = c.opts.Reader.Values(snap.ChangeSet)
|
|
|
|
|
|
|
|
c.Unlock()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for {
|
|
|
|
w, err := c.opts.Loader.Watch()
|
|
|
|
if err != nil {
|
|
|
|
time.Sleep(time.Second)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
done := make(chan bool)
|
|
|
|
|
|
|
|
// the stop watch func
|
|
|
|
go func() {
|
|
|
|
select {
|
|
|
|
case <-done:
|
|
|
|
case <-c.exit:
|
|
|
|
}
|
|
|
|
w.Stop()
|
|
|
|
}()
|
|
|
|
|
|
|
|
// block watch
|
|
|
|
if err := watch(w); err != nil {
|
|
|
|
// do something better
|
|
|
|
time.Sleep(time.Second)
|
|
|
|
}
|
|
|
|
|
|
|
|
// close done chan
|
|
|
|
close(done)
|
|
|
|
|
|
|
|
// if the config is closed exit
|
|
|
|
select {
|
|
|
|
case <-c.exit:
|
|
|
|
return
|
|
|
|
default:
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *config) Map() map[string]interface{} {
|
|
|
|
c.RLock()
|
|
|
|
defer c.RUnlock()
|
|
|
|
return c.vals.Map()
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *config) Scan(v interface{}) error {
|
|
|
|
c.RLock()
|
|
|
|
defer c.RUnlock()
|
|
|
|
return c.vals.Scan(v)
|
|
|
|
}
|
|
|
|
|
|
|
|
// sync loads all the sources, calls the parser and updates the config
|
|
|
|
func (c *config) Sync() error {
|
|
|
|
if err := c.opts.Loader.Sync(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
snap, err := c.opts.Loader.Snapshot()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
c.Lock()
|
|
|
|
defer c.Unlock()
|
|
|
|
|
|
|
|
c.snap = snap
|
|
|
|
vals, err := c.opts.Reader.Values(snap.ChangeSet)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
c.vals = vals
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *config) Close() error {
|
|
|
|
select {
|
|
|
|
case <-c.exit:
|
|
|
|
return nil
|
|
|
|
default:
|
|
|
|
close(c.exit)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *config) Get(path ...string) reader.Value {
|
|
|
|
c.RLock()
|
|
|
|
defer c.RUnlock()
|
|
|
|
|
|
|
|
// did sync actually work?
|
|
|
|
if c.vals != nil {
|
|
|
|
return c.vals.Get(path...)
|
|
|
|
}
|
|
|
|
|
|
|
|
// no value
|
|
|
|
return newValue()
|
|
|
|
}
|
|
|
|
|
2020-01-11 22:50:09 +02:00
|
|
|
func (c *config) Set(val interface{}, path ...string) {
|
|
|
|
c.Lock()
|
|
|
|
defer c.Unlock()
|
|
|
|
|
|
|
|
if c.vals != nil {
|
|
|
|
c.vals.Set(val, path...)
|
|
|
|
}
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *config) Del(path ...string) {
|
|
|
|
c.Lock()
|
|
|
|
defer c.Unlock()
|
|
|
|
|
|
|
|
if c.vals != nil {
|
|
|
|
c.vals.Del(path...)
|
|
|
|
}
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2019-05-31 00:11:13 +02:00
|
|
|
func (c *config) Bytes() []byte {
|
|
|
|
c.RLock()
|
|
|
|
defer c.RUnlock()
|
|
|
|
|
|
|
|
if c.vals == nil {
|
|
|
|
return []byte{}
|
|
|
|
}
|
|
|
|
|
|
|
|
return c.vals.Bytes()
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *config) Load(sources ...source.Source) error {
|
|
|
|
if err := c.opts.Loader.Load(sources...); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
snap, err := c.opts.Loader.Snapshot()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
c.Lock()
|
|
|
|
defer c.Unlock()
|
|
|
|
|
|
|
|
c.snap = snap
|
|
|
|
vals, err := c.opts.Reader.Values(snap.ChangeSet)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
c.vals = vals
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *config) Watch(path ...string) (Watcher, error) {
|
|
|
|
value := c.Get(path...)
|
|
|
|
|
|
|
|
w, err := c.opts.Loader.Watch(path...)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return &watcher{
|
|
|
|
lw: w,
|
|
|
|
rd: c.opts.Reader,
|
|
|
|
path: path,
|
|
|
|
value: value,
|
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *config) String() string {
|
|
|
|
return "config"
|
|
|
|
}
|
|
|
|
|
|
|
|
func (w *watcher) Next() (reader.Value, error) {
|
|
|
|
for {
|
|
|
|
s, err := w.lw.Next()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
// only process changes
|
|
|
|
if bytes.Equal(w.value.Bytes(), s.ChangeSet.Data) {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
v, err := w.rd.Values(s.ChangeSet)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
w.value = v.Get()
|
|
|
|
return w.value, nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (w *watcher) Stop() error {
|
|
|
|
return w.lw.Stop()
|
|
|
|
}
|