107 lines
2.2 KiB
Go
107 lines
2.2 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/micro/go-micro/v2/client"
|
|
"github.com/micro/go-micro/v2/config/source"
|
|
proto "github.com/micro/go-micro/v2/config/source/service/proto"
|
|
"github.com/micro/go-micro/v2/errors"
|
|
"github.com/micro/go-micro/v2/logger"
|
|
)
|
|
|
|
var (
|
|
DefaultName = "go.micro.config"
|
|
DefaultNamespace = "global"
|
|
DefaultPath = ""
|
|
)
|
|
|
|
type service struct {
|
|
serviceName string
|
|
namespace string
|
|
path string
|
|
opts source.Options
|
|
client proto.ConfigService
|
|
}
|
|
|
|
func (m *service) Read() (set *source.ChangeSet, err error) {
|
|
client := proto.NewConfigService(m.serviceName, m.opts.Client)
|
|
req, err := client.Read(context.Background(), &proto.ReadRequest{
|
|
Namespace: m.namespace,
|
|
Path: m.path,
|
|
})
|
|
if verr, ok := err.(*errors.Error); ok && verr.Code == http.StatusNotFound {
|
|
return &source.ChangeSet{Data: []byte{}}, nil
|
|
} else if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return toChangeSet(req.Change.ChangeSet), nil
|
|
}
|
|
|
|
func (m *service) Watch() (w source.Watcher, err error) {
|
|
client := proto.NewConfigService(m.serviceName, m.opts.Client)
|
|
stream, err := client.Watch(context.Background(), &proto.WatchRequest{
|
|
Namespace: m.namespace,
|
|
Path: m.path,
|
|
})
|
|
if err != nil {
|
|
if logger.V(logger.ErrorLevel, logger.DefaultLogger) {
|
|
logger.Error("watch err: ", err)
|
|
}
|
|
return
|
|
}
|
|
return newWatcher(stream)
|
|
}
|
|
|
|
// Write is unsupported
|
|
func (m *service) Write(cs *source.ChangeSet) error {
|
|
return nil
|
|
}
|
|
|
|
func (m *service) String() string {
|
|
return "service"
|
|
}
|
|
|
|
func NewSource(opts ...source.Option) source.Source {
|
|
var options source.Options
|
|
for _, o := range opts {
|
|
o(&options)
|
|
}
|
|
|
|
addr := DefaultName
|
|
namespace := DefaultNamespace
|
|
path := DefaultPath
|
|
|
|
if options.Context != nil {
|
|
a, ok := options.Context.Value(serviceNameKey{}).(string)
|
|
if ok {
|
|
addr = a
|
|
}
|
|
|
|
k, ok := options.Context.Value(namespaceKey{}).(string)
|
|
if ok {
|
|
namespace = k
|
|
}
|
|
|
|
p, ok := options.Context.Value(pathKey{}).(string)
|
|
if ok {
|
|
path = p
|
|
}
|
|
}
|
|
|
|
if options.Client == nil {
|
|
options.Client = client.DefaultClient
|
|
}
|
|
|
|
s := &service{
|
|
serviceName: addr,
|
|
opts: options,
|
|
namespace: namespace,
|
|
path: path,
|
|
}
|
|
|
|
return s
|
|
}
|