edge/pkg/devicetwin/devicetwin.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

package devicetwin

import (
	"k8s.io/klog/v2"

	"github.com/kubeedge/beehive/pkg/core"
	"github.com/kubeedge/kubeedge/edge/pkg/common/modules"
	deviceconfig "github.com/kubeedge/kubeedge/edge/pkg/devicetwin/config"
	"github.com/kubeedge/kubeedge/edge/pkg/devicetwin/dtclient"
	"github.com/kubeedge/kubeedge/edge/pkg/devicetwin/dtcontext"
	"github.com/kubeedge/kubeedge/edge/pkg/devicetwin/dtmodule"
	"github.com/kubeedge/kubeedge/pkg/apis/componentconfig/edgecore/v1alpha2"
)

// DeviceTwin the module
type DeviceTwin struct {
	HeartBeatToModule map[string]chan interface{}
	DTContexts        *dtcontext.DTContext
	DTModules         map[string]dtmodule.DTModule
	enable            bool
}

var _ core.Module = (*DeviceTwin)(nil)

func newDeviceTwin(enable bool) *DeviceTwin {
	return &DeviceTwin{
		HeartBeatToModule: make(map[string]chan interface{}),
		DTModules:         make(map[string]dtmodule.DTModule),
		enable:            enable,
	}
}

// Register register devicetwin
func Register(deviceTwin *v1alpha2.DeviceTwin, nodeName string) {
	deviceconfig.InitConfigure(deviceTwin, nodeName)
	dt := newDeviceTwin(deviceTwin.Enable)
	dtclient.InitDBTable(dt)
	core.Register(dt)
}

// Name get name of the module
func (dt *DeviceTwin) Name() string {
	return modules.DeviceTwinModuleName
}

// Group get group of the module
func (dt *DeviceTwin) Group() string {
	return modules.TwinGroup
}

// Enable indicates whether this module is enabled
func (dt *DeviceTwin) Enable() bool {
	return dt.enable
}

// Start run the module
func (dt *DeviceTwin) Start() {
	dtContexts, _ := dtcontext.InitDTContext()
	dt.DTContexts = dtContexts
	err := SyncSqlite(dt.DTContexts)
	if err != nil {
		klog.Errorf("Start DeviceTwin Failed, Sync Sqlite error:%v", err)
		return
	}
	dt.runDeviceTwin()
}