138 lines
2.6 KiB
Go
138 lines
2.6 KiB
Go
package registry
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"sync"
|
|
|
|
k8s "github.com/GoogleCloudPlatform/kubernetes/pkg/client"
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/labels"
|
|
)
|
|
|
|
type KubernetesRegistry struct {
|
|
Client *k8s.Client
|
|
Namespace string
|
|
|
|
mtx sync.RWMutex
|
|
services map[string]Service
|
|
}
|
|
|
|
func (c *KubernetesRegistry) Watch() {
|
|
NewKubernetesWatcher(c)
|
|
}
|
|
|
|
func (c *KubernetesRegistry) Deregister(s Service) error {
|
|
return nil
|
|
}
|
|
|
|
func (c *KubernetesRegistry) Register(s Service) error {
|
|
return nil
|
|
}
|
|
|
|
func (c *KubernetesRegistry) GetService(name string) (Service, error) {
|
|
c.mtx.RLock()
|
|
service, ok := c.services[name]
|
|
c.mtx.RUnlock()
|
|
|
|
if ok {
|
|
return service, nil
|
|
}
|
|
|
|
selector := labels.SelectorFromSet(labels.Set{"name": name})
|
|
|
|
services, err := c.Client.Services(c.Namespace).List(selector)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(services.Items) == 0 {
|
|
return nil, fmt.Errorf("Service not found")
|
|
}
|
|
|
|
ks := &KubernetesService{ServiceName: name}
|
|
for _, item := range services.Items {
|
|
ks.ServiceNodes = append(ks.ServiceNodes, &KubernetesNode{
|
|
NodeAddress: item.Spec.PortalIP,
|
|
NodePort: item.Spec.Ports[0].Port,
|
|
})
|
|
}
|
|
|
|
return ks, nil
|
|
}
|
|
|
|
func (c *KubernetesRegistry) ListServices() ([]Service, error) {
|
|
c.mtx.RLock()
|
|
serviceMap := c.services
|
|
c.mtx.RUnlock()
|
|
|
|
var services []Service
|
|
|
|
if len(serviceMap) > 0 {
|
|
for _, service := range serviceMap {
|
|
services = append(services, service)
|
|
}
|
|
return services, nil
|
|
}
|
|
|
|
rsp, err := c.Client.Services(c.Namespace).List(labels.Everything())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, service := range rsp.Items {
|
|
if len(service.ObjectMeta.Labels["name"]) == 0 {
|
|
continue
|
|
}
|
|
|
|
services = append(services, &KubernetesService{
|
|
ServiceName: service.ObjectMeta.Labels["name"],
|
|
})
|
|
}
|
|
|
|
return services, nil
|
|
}
|
|
|
|
func (c *KubernetesRegistry) NewService(name string, nodes ...Node) Service {
|
|
var snodes []*KubernetesNode
|
|
|
|
for _, node := range nodes {
|
|
if n, ok := node.(*KubernetesNode); ok {
|
|
snodes = append(snodes, n)
|
|
}
|
|
}
|
|
|
|
return &KubernetesService{
|
|
ServiceName: name,
|
|
ServiceNodes: snodes,
|
|
}
|
|
}
|
|
|
|
func (c *KubernetesRegistry) NewNode(id, address string, port int) Node {
|
|
return &KubernetesNode{
|
|
NodeId: id,
|
|
NodeAddress: address,
|
|
NodePort: port,
|
|
}
|
|
}
|
|
|
|
func NewKubernetesRegistry(addrs []string, opts ...Options) Registry {
|
|
host := "http://" + os.Getenv("KUBERNETES_RO_SERVICE_HOST") + ":" + os.Getenv("KUBERNETES_RO_SERVICE_PORT")
|
|
if len(addrs) > 0 {
|
|
host = addrs[0]
|
|
}
|
|
|
|
client, _ := k8s.New(&k8s.Config{
|
|
Host: host,
|
|
})
|
|
|
|
kr := &KubernetesRegistry{
|
|
Client: client,
|
|
Namespace: "default",
|
|
services: make(map[string]Service),
|
|
}
|
|
|
|
kr.Watch()
|
|
|
|
return kr
|
|
}
|