1
0
mirror of https://github.com/go-micro/go-micro.git synced 2024-12-12 08:23:58 +02:00
go-micro/selector/filter.go

74 lines
1.6 KiB
Go
Raw Normal View History

2016-04-23 21:57:46 +02:00
package selector
import (
"github.com/micro/go-micro/registry"
)
2016-04-23 22:24:07 +02:00
// FilterEndpoint is an endpoint based Select Filter which will
// only return services with the endpoint specified.
func FilterEndpoint(name string) Filter {
return func(old []*registry.Service) []*registry.Service {
var services []*registry.Service
for _, service := range old {
for _, ep := range service.Endpoints {
if ep.Name == name {
services = append(services, service)
break
}
}
}
return services
}
}
2016-04-23 22:09:48 +02:00
// FilterLabel is a label based Select Filter which will
2016-04-23 21:57:46 +02:00
// only return services with the label specified.
2016-04-23 22:09:48 +02:00
func FilterLabel(key, val string) Filter {
2016-04-23 21:57:46 +02:00
return func(old []*registry.Service) []*registry.Service {
var services []*registry.Service
for _, service := range old {
serv := new(registry.Service)
var nodes []*registry.Node
for _, node := range service.Nodes {
if node.Metadata == nil {
continue
}
if node.Metadata[key] == val {
nodes = append(nodes, node)
}
}
// only add service if there's some nodes
if len(nodes) > 0 {
// copy
*serv = *service
serv.Nodes = nodes
services = append(services, serv)
}
}
return services
}
}
2016-04-23 22:09:48 +02:00
// FilterVersion is a version based Select Filter which will
2016-04-23 21:57:46 +02:00
// only return services with the version specified.
2016-04-23 22:09:48 +02:00
func FilterVersion(version string) Filter {
2016-04-23 21:57:46 +02:00
return func(old []*registry.Service) []*registry.Service {
var services []*registry.Service
for _, service := range old {
if service.Version == version {
services = append(services, service)
}
}
return services
}
}