4107733453
* PoC: memory registry using maps instead of slice madness * Updated proto and handlers. Fixed tests across codebase. * Implemented ttl pruning for memory registry * Added extensive memory registry tests * Squased a bunch of bugs * Proto indent; memory.Registry.String() returns "memory" * Write a test to prove memory registry TTLs are busted Signed-off-by: Erik Hollensbe <github@hollensbe.org> * Additional memory testing and fixups: * DefaultTTL removed * When TTL == 0, it is automatically removed from expiry conditions * Additional improvements to new tests Signed-off-by: Erik Hollensbe <github@hollensbe.org>
41 lines
897 B
Go
41 lines
897 B
Go
package memory
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/micro/go-micro/registry"
|
|
)
|
|
|
|
type servicesKey struct{}
|
|
|
|
func getServiceRecords(ctx context.Context) map[string]map[string]*record {
|
|
memServices, ok := ctx.Value(servicesKey{}).(map[string][]*registry.Service)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
|
|
services := make(map[string]map[string]*record)
|
|
|
|
for name, svc := range memServices {
|
|
if _, ok := services[name]; !ok {
|
|
services[name] = make(map[string]*record)
|
|
}
|
|
// go through every version of the service
|
|
for _, s := range svc {
|
|
services[s.Name][s.Version] = serviceToRecord(s, 0)
|
|
}
|
|
}
|
|
|
|
return services
|
|
}
|
|
|
|
// Services is an option that preloads service data
|
|
func Services(s map[string][]*registry.Service) registry.Option {
|
|
return func(o *registry.Options) {
|
|
if o.Context == nil {
|
|
o.Context = context.Background()
|
|
}
|
|
o.Context = context.WithValue(o.Context, servicesKey{}, s)
|
|
}
|
|
}
|