/
initialize.go
115 lines (99 loc) · 2.35 KB
/
initialize.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
package initz
import (
"github.com/baetyl/baetyl-go/v2/context"
"github.com/baetyl/baetyl-go/v2/errors"
"github.com/baetyl/baetyl-go/v2/log"
v2utils "github.com/baetyl/baetyl-go/v2/utils"
bh "github.com/timshannon/bolthold"
"github.com/baetyl/baetyl/v2/ami"
"github.com/baetyl/baetyl/v2/ami/kube"
"github.com/baetyl/baetyl/v2/config"
"github.com/baetyl/baetyl/v2/engine"
"github.com/baetyl/baetyl/v2/node"
"github.com/baetyl/baetyl/v2/plugin"
"github.com/baetyl/baetyl/v2/store"
"github.com/baetyl/baetyl/v2/sync"
"github.com/baetyl/baetyl/v2/utils"
)
type StartInitServiceFunc func()
type Initialize struct {
cfg config.Config
sto *bh.Store
nod node.Node
eng engine.Engine
syn sync.Sync
log *log.Logger
tomb v2utils.Tomb
}
// NewInitialize creates a new core
func NewInitialize(cfg config.Config) (*Initialize, error) {
initHooks()
// to activate if no node cert
if !v2utils.FileExists(cfg.Node.Cert) {
active, err := NewActivate(&cfg)
if err != nil {
return nil, errors.Trace(err)
}
active.Start()
active.WaitAndClose()
log.L().Info("init activates node success")
}
err := utils.ExtractNodeInfo(cfg.Node)
if err != nil {
return nil, errors.Trace(err)
}
init := &Initialize{
cfg: cfg,
log: log.With(log.Any("init", "sync")),
}
init.sto, err = store.NewBoltHold(cfg.Store.Path)
if err != nil {
return nil, errors.Trace(err)
}
init.nod, err = node.NewNode(init.sto)
if err != nil {
return nil, errors.Trace(err)
}
init.syn, err = sync.NewSync(cfg, init.sto, init.nod)
if err != nil {
return nil, errors.Trace(err)
}
init.eng, err = engine.NewEngine(cfg, init.sto, init.nod, init.syn, nil)
if err != nil {
return nil, errors.Trace(err)
}
init.eng.Start()
init.syn.Start()
return init, nil
}
func initHooks() {
ami.Hooks[kube.BaetylSetPodSpec] = kube.SetPodSpecFunc(kube.SetPodSpec)
}
func (init *Initialize) Close() {
if init.eng != nil {
init.eng.Close()
}
if init.sto != nil {
init.sto.Close()
}
if init.syn != nil {
init.syn.Close()
}
}
func StartInitService() {
context.Run(func(ctx context.Context) error {
var cfg config.Config
err := ctx.LoadCustomConfig(&cfg)
if err != nil {
return errors.Trace(err)
}
plugin.ConfFile = ctx.ConfFile()
init, err := NewInitialize(cfg)
if err != nil {
return errors.Trace(err)
}
defer init.Close()
ctx.Wait()
return nil
})
}