138 lines
3.1 KiB
Go
138 lines
3.1 KiB
Go
package controller
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"k8s.io/client-go/pkg/api"
|
|
"k8s.io/client-go/pkg/api/v1"
|
|
"k8s.io/client-go/pkg/runtime"
|
|
"k8s.io/client-go/pkg/watch"
|
|
|
|
"github.com/zalando-incubator/postgres-operator/pkg/spec"
|
|
"github.com/zalando-incubator/postgres-operator/pkg/util"
|
|
)
|
|
|
|
func (c *Controller) podListFunc(options api.ListOptions) (runtime.Object, error) {
|
|
var labelSelector string
|
|
var fieldSelector string
|
|
|
|
if options.LabelSelector != nil {
|
|
labelSelector = options.LabelSelector.String()
|
|
}
|
|
|
|
if options.FieldSelector != nil {
|
|
fieldSelector = options.FieldSelector.String()
|
|
}
|
|
opts := v1.ListOptions{
|
|
LabelSelector: labelSelector,
|
|
FieldSelector: fieldSelector,
|
|
Watch: options.Watch,
|
|
ResourceVersion: options.ResourceVersion,
|
|
TimeoutSeconds: options.TimeoutSeconds,
|
|
}
|
|
|
|
return c.KubeClient.CoreV1().Pods(c.opConfig.Namespace).List(opts)
|
|
}
|
|
|
|
func (c *Controller) podWatchFunc(options api.ListOptions) (watch.Interface, error) {
|
|
var labelSelector string
|
|
var fieldSelector string
|
|
|
|
if options.LabelSelector != nil {
|
|
labelSelector = options.LabelSelector.String()
|
|
}
|
|
|
|
if options.FieldSelector != nil {
|
|
fieldSelector = options.FieldSelector.String()
|
|
}
|
|
|
|
opts := v1.ListOptions{
|
|
LabelSelector: labelSelector,
|
|
FieldSelector: fieldSelector,
|
|
Watch: options.Watch,
|
|
ResourceVersion: options.ResourceVersion,
|
|
TimeoutSeconds: options.TimeoutSeconds,
|
|
}
|
|
|
|
return c.KubeClient.CoreV1Client.Pods(c.opConfig.Namespace).Watch(opts)
|
|
}
|
|
|
|
func (c *Controller) podAdd(obj interface{}) {
|
|
pod, ok := obj.(*v1.Pod)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
podEvent := spec.PodEvent{
|
|
ClusterName: c.podClusterName(pod),
|
|
PodName: util.NameFromMeta(pod.ObjectMeta),
|
|
CurPod: pod,
|
|
EventType: spec.EventAdd,
|
|
ResourceVersion: pod.ResourceVersion,
|
|
}
|
|
|
|
c.podCh <- podEvent
|
|
}
|
|
|
|
func (c *Controller) podUpdate(prev, cur interface{}) {
|
|
prevPod, ok := prev.(*v1.Pod)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
curPod, ok := cur.(*v1.Pod)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
podEvent := spec.PodEvent{
|
|
ClusterName: c.podClusterName(curPod),
|
|
PodName: util.NameFromMeta(curPod.ObjectMeta),
|
|
PrevPod: prevPod,
|
|
CurPod: curPod,
|
|
EventType: spec.EventUpdate,
|
|
ResourceVersion: curPod.ResourceVersion,
|
|
}
|
|
|
|
c.podCh <- podEvent
|
|
}
|
|
|
|
func (c *Controller) podDelete(obj interface{}) {
|
|
pod, ok := obj.(*v1.Pod)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
podEvent := spec.PodEvent{
|
|
ClusterName: c.podClusterName(pod),
|
|
PodName: util.NameFromMeta(pod.ObjectMeta),
|
|
CurPod: pod,
|
|
EventType: spec.EventDelete,
|
|
ResourceVersion: pod.ResourceVersion,
|
|
}
|
|
|
|
c.podCh <- podEvent
|
|
}
|
|
|
|
func (c *Controller) podEventsDispatcher(stopCh <-chan struct{}, wg *sync.WaitGroup) {
|
|
defer wg.Done()
|
|
wg.Add(1)
|
|
|
|
c.logger.Debugln("Watching all pod events")
|
|
for {
|
|
select {
|
|
case event := <-c.podCh:
|
|
c.clustersMu.RLock()
|
|
cluster, ok := c.clusters[event.ClusterName]
|
|
c.clustersMu.RUnlock()
|
|
|
|
if ok {
|
|
c.logger.Debugf("Sending %s event of pod '%s' to the '%s' cluster channel", event.EventType, event.PodName, event.ClusterName)
|
|
cluster.ReceivePodEvent(event)
|
|
}
|
|
case <-stopCh:
|
|
return
|
|
}
|
|
}
|
|
}
|