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
|
package edgecontroller
import (
"k8s.io/klog/v2"
"github.com/kubeedge/beehive/pkg/core"
"github.com/kubeedge/kubeedge/cloud/pkg/common/informers"
"github.com/kubeedge/kubeedge/cloud/pkg/common/modules"
"github.com/kubeedge/kubeedge/cloud/pkg/edgecontroller/controller"
"github.com/kubeedge/kubeedge/pkg/apis/componentconfig/cloudcore/v1alpha1"
)
// EdgeController use beehive context message layer
type EdgeController struct {
config v1alpha1.EdgeController
upstream *controller.UpstreamController
downstream *controller.DownstreamController
}
func newEdgeController(config *v1alpha1.EdgeController, tunnelPort int) *EdgeController {
ec := &EdgeController{config: *config}
if !ec.Enable() {
return ec
}
var err error
ec.upstream, err = controller.NewUpstreamController(config, informers.GetInformersManager().GetK8sInformerFactory())
if err != nil {
klog.Exitf("new upstream controller failed with error: %s", err)
}
ec.upstream.TunnelPort = tunnelPort
ec.downstream, err = controller.NewDownstreamController(config, informers.GetInformersManager().GetK8sInformerFactory(), informers.GetInformersManager(), informers.GetInformersManager().GetCRDInformerFactory())
if err != nil {
klog.Exitf("new downstream controller failed with error: %s", err)
}
return ec
}
func Register(ec *v1alpha1.EdgeController, commonConfig *v1alpha1.CommonConfig) {
core.Register(newEdgeController(ec, commonConfig.TunnelPort))
}
// Name of controller
func (ec *EdgeController) Name() string {
return modules.EdgeControllerModuleName
}
// Group of controller
func (ec *EdgeController) Group() string {
return modules.EdgeControllerGroupName
}
// Enable indicates whether enable this module
func (ec *EdgeController) Enable() bool {
return ec.config.Enable
}
// Start controller
func (ec *EdgeController) Start() {
if err := ec.upstream.Start(); err != nil {
klog.Exitf("start upstream failed with error: %s", err)
}
if err := ec.downstream.Start(); err != nil {
klog.Exitf("start downstream failed with error: %s", err)
}
}
|
