mirror of
https://github.com/opencloud-eu/opencloud.git
synced 2026-05-18 13:35:37 -04:00
4f26424db6
* enhancement: use reva client pool selectors register mock service to registry and pass tests * enhancement: bump reva * Fix a couple of linter issues --------- Co-authored-by: Ralf Haferkamp <rhaferkamp@owncloud.com>
84 lines
2.1 KiB
Go
84 lines
2.1 KiB
Go
package registry
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
rRegistry "github.com/cs3org/reva/v2/pkg/registry"
|
|
consulr "github.com/go-micro/plugins/v4/registry/consul"
|
|
etcdr "github.com/go-micro/plugins/v4/registry/etcd"
|
|
kubernetesr "github.com/go-micro/plugins/v4/registry/kubernetes"
|
|
mdnsr "github.com/go-micro/plugins/v4/registry/mdns"
|
|
memr "github.com/go-micro/plugins/v4/registry/memory"
|
|
natsr "github.com/go-micro/plugins/v4/registry/nats"
|
|
mRegistry "go-micro.dev/v4/registry"
|
|
"go-micro.dev/v4/registry/cache"
|
|
)
|
|
|
|
const (
|
|
registryEnv = "MICRO_REGISTRY"
|
|
registryAddressEnv = "MICRO_REGISTRY_ADDRESS"
|
|
)
|
|
|
|
var (
|
|
once sync.Once
|
|
regPlugin string
|
|
reg mRegistry.Registry
|
|
)
|
|
|
|
func Configure(plugin string) {
|
|
if reg == nil {
|
|
regPlugin = plugin
|
|
}
|
|
}
|
|
|
|
// GetRegistry returns a configured micro registry based on Micro env vars.
|
|
// It defaults to mDNS, so mind that systems with mDNS disabled by default (i.e SUSE) will have a hard time
|
|
// and it needs to explicitly use etcd. Os awareness for providing a working registry out of the box should be done.
|
|
func GetRegistry() mRegistry.Registry {
|
|
once.Do(func() {
|
|
addresses := strings.Split(os.Getenv(registryAddressEnv), ",")
|
|
// prefer env of setting from Configure()
|
|
plugin := os.Getenv(registryEnv)
|
|
if plugin == "" {
|
|
plugin = regPlugin
|
|
}
|
|
|
|
switch plugin {
|
|
case "nats":
|
|
reg = natsr.NewRegistry(
|
|
mRegistry.Addrs(addresses...),
|
|
)
|
|
case "kubernetes":
|
|
reg = kubernetesr.NewRegistry(
|
|
mRegistry.Addrs(addresses...),
|
|
)
|
|
case "etcd":
|
|
reg = etcdr.NewRegistry(
|
|
mRegistry.Addrs(addresses...),
|
|
)
|
|
case "consul":
|
|
reg = consulr.NewRegistry(
|
|
mRegistry.Addrs(addresses...),
|
|
)
|
|
case "memory":
|
|
reg = memr.NewRegistry()
|
|
default:
|
|
reg = mdnsr.NewRegistry()
|
|
}
|
|
|
|
// No cache needed for in-memory registry
|
|
if plugin != "memory" {
|
|
reg = cache.New(reg, cache.WithTTL(30*time.Second))
|
|
}
|
|
|
|
// fixme: lazy initialization of reva registry, needs refactor to a explicit call per service
|
|
_ = rRegistry.Init(reg)
|
|
})
|
|
// always use cached registry to prevent registry
|
|
// lookup for every request
|
|
return reg
|
|
}
|