73 lines
1.4 KiB
Go
73 lines
1.4 KiB
Go
|
package kubernetes
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/api"
|
||
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/proxy/config"
|
||
|
|
"github.com/GoogleCloudPlatform/kubernetes/pkg/util"
|
||
|
|
)
|
||
|
|
|
||
|
|
type watcher struct {
|
||
|
|
registry *kregistry
|
||
|
|
}
|
||
|
|
|
||
|
|
func (k *watcher) OnUpdate(services []api.Service) {
|
||
|
|
fmt.Println("got update")
|
||
|
|
activeServices := util.StringSet{}
|
||
|
|
for _, svc := range services {
|
||
|
|
fmt.Printf("%#v\n", svc.ObjectMeta)
|
||
|
|
name, exists := svc.ObjectMeta.Labels["name"]
|
||
|
|
if !exists {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
activeServices.Insert(name)
|
||
|
|
serviceIP := net.ParseIP(svc.Spec.PortalIP)
|
||
|
|
|
||
|
|
ks := &service{
|
||
|
|
name: name,
|
||
|
|
nodes: []*node{
|
||
|
|
&node{
|
||
|
|
address: serviceIP.String(),
|
||
|
|
port: svc.Spec.Ports[0].Port,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
k.registry.mtx.Lock()
|
||
|
|
k.registry.services[name] = ks
|
||
|
|
k.registry.mtx.Unlock()
|
||
|
|
}
|
||
|
|
|
||
|
|
k.registry.mtx.Lock()
|
||
|
|
defer k.registry.mtx.Unlock()
|
||
|
|
for name, _ := range k.registry.services {
|
||
|
|
if !activeServices.Has(name) {
|
||
|
|
delete(k.registry.services, name)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func newWatcher(kr *kregistry) *watcher {
|
||
|
|
serviceConfig := config.NewServiceConfig()
|
||
|
|
endpointsConfig := config.NewEndpointsConfig()
|
||
|
|
|
||
|
|
config.NewSourceAPI(
|
||
|
|
kr.client.Services(api.NamespaceAll),
|
||
|
|
kr.client.Endpoints(api.NamespaceAll),
|
||
|
|
time.Second*10,
|
||
|
|
serviceConfig.Channel("api"),
|
||
|
|
endpointsConfig.Channel("api"),
|
||
|
|
)
|
||
|
|
|
||
|
|
ks := &watcher{
|
||
|
|
registry: kr,
|
||
|
|
}
|
||
|
|
|
||
|
|
serviceConfig.RegisterHandler(ks)
|
||
|
|
return ks
|
||
|
|
}
|