-
Notifications
You must be signed in to change notification settings - Fork 438
/
provider-network.go
129 lines (111 loc) · 3.3 KB
/
provider-network.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
package controller
import (
"context"
"fmt"
corev1 "k8s.io/api/core/v1"
k8serrors "k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/labels"
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
"k8s.io/client-go/tools/cache"
"k8s.io/klog/v2"
kubeovnv1 "github.com/kubeovn/kube-ovn/pkg/apis/kubeovn/v1"
"github.com/kubeovn/kube-ovn/pkg/util"
)
func (c *Controller) enqueueUpdateProviderNetwork(_, obj interface{}) {
if !c.isLeader() {
return
}
key, err := cache.MetaNamespaceKeyFunc(obj)
if err != nil {
utilruntime.HandleError(err)
return
}
klog.V(3).Infof("enqueue update provider network %s", key)
c.updateProviderNetworkQueue.Add(key)
}
func (c *Controller) runUpdateProviderNetworkWorker() {
for c.processNextUpdateProviderNetworkWorkItem() {
}
}
func (c *Controller) processNextUpdateProviderNetworkWorkItem() bool {
obj, shutdown := c.updateProviderNetworkQueue.Get()
if shutdown {
return false
}
err := func(obj interface{}) error {
defer c.updateProviderNetworkQueue.Done(obj)
var key string
var ok bool
if key, ok = obj.(string); !ok {
c.updateProviderNetworkQueue.Forget(obj)
utilruntime.HandleError(fmt.Errorf("expected string in workqueue but got %#v", obj))
return nil
}
if err := c.handleUpdateProviderNetwork(key); err != nil {
c.updateProviderNetworkQueue.AddRateLimited(key)
return fmt.Errorf("error syncing '%s': %s, requeuing", key, err.Error())
}
c.updateProviderNetworkQueue.Forget(obj)
return nil
}(obj)
if err != nil {
utilruntime.HandleError(err)
}
return true
}
func (c *Controller) handleUpdateProviderNetwork(key string) error {
pn, err := c.providerNetworksLister.Get(key)
if err != nil {
if k8serrors.IsNotFound(err) {
return nil
}
return err
}
nodes, err := c.nodesLister.List(labels.Everything())
if err != nil {
klog.Errorf("failed to list nodes: %v", err)
return err
}
if providerNetworkIsReady(pn, nodes) != pn.Status.Ready {
newPn := pn.DeepCopy()
newPn.Status.Ready = !pn.Status.Ready
_, err = c.config.KubeOvnClient.KubeovnV1().ProviderNetworks().UpdateStatus(context.Background(), newPn, metav1.UpdateOptions{})
if err != nil {
klog.Errorf("failed to update status of provider network %s: %v", pn.Name, err)
return err
}
}
return nil
}
func (c *Controller) resyncProviderNetworkStatus() {
nodeList, err := c.nodesLister.List(labels.Everything())
if err != nil {
klog.Errorf("failed to get nodes %v", err)
return
}
pnList, err := c.providerNetworksLister.List(labels.Everything())
if err != nil {
klog.Errorf("failed to get provider networks %v", err)
return
}
for _, pn := range pnList {
if providerNetworkIsReady(pn, nodeList) != pn.Status.Ready {
newPn := pn.DeepCopy()
newPn.Status.Ready = !newPn.Status.Ready
_, err = c.config.KubeOvnClient.KubeovnV1().ProviderNetworks().UpdateStatus(context.Background(), newPn, metav1.UpdateOptions{})
if err != nil {
klog.Errorf("failed to update status of provider network %s: %v", pn.Name, err)
}
}
}
}
func providerNetworkIsReady(pn *kubeovnv1.ProviderNetwork, nodes []*corev1.Node) bool {
for _, node := range nodes {
if !util.ContainsString(pn.Spec.ExcludeNodes, node.Name) &&
!util.ContainsString(pn.Status.ReadyNodes, node.Name) {
return false
}
}
return true
}