-
Notifications
You must be signed in to change notification settings - Fork 21
/
system_services.go
125 lines (111 loc) · 2.7 KB
/
system_services.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
package nodes
import (
"bytes"
"encoding/json"
"errors"
"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
teaconst "github.com/TeaOSLab/EdgeNode/internal/const"
"github.com/TeaOSLab/EdgeNode/internal/events"
"github.com/TeaOSLab/EdgeNode/internal/goman"
"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
"github.com/TeaOSLab/EdgeNode/internal/utils"
"github.com/iwind/TeaGo/maps"
"os"
"os/exec"
"runtime"
"time"
)
func init() {
var manager = NewSystemServiceManager()
events.On(events.EventReload, func() {
err := manager.Setup()
if err != nil {
remotelogs.Error("SYSTEM_SERVICE", "setup system services failed: "+err.Error())
}
})
}
// SystemServiceManager 系统服务管理
type SystemServiceManager struct {
}
func NewSystemServiceManager() *SystemServiceManager {
return &SystemServiceManager{}
}
func (this *SystemServiceManager) Setup() error {
if sharedNodeConfig == nil || !sharedNodeConfig.IsOn {
return nil
}
if len(sharedNodeConfig.SystemServices) == 0 {
return nil
}
systemdParams, ok := sharedNodeConfig.SystemServices[nodeconfigs.SystemServiceTypeSystemd]
if ok {
err := this.setupSystemd(systemdParams)
if err != nil {
return err
}
}
return nil
}
func (this *SystemServiceManager) setupSystemd(params maps.Map) error {
// 只有在Linux下运行
if runtime.GOOS != "linux" {
return nil
}
if params == nil {
params = maps.Map{}
}
data, err := json.Marshal(params)
if err != nil {
return err
}
config := &nodeconfigs.SystemdServiceConfig{}
err = json.Unmarshal(data, config)
if err != nil {
return err
}
// 检查当前的service
systemctl, err := exec.LookPath("systemctl")
if err != nil {
return err
}
if len(systemctl) == 0 {
return errors.New("can not find 'systemctl' on the system")
}
cmd := utils.NewCommandExecutor()
shortName := teaconst.SystemdServiceName
cmd.Add(systemctl, "is-enabled", shortName)
output, err := cmd.Run()
if err != nil {
return err
}
if config.IsOn {
exe, err := os.Executable()
if err != nil {
return err
}
// 启动Service
goman.New(func() {
time.Sleep(5 * time.Second)
_ = exec.Command(systemctl, "start", teaconst.SystemdServiceName).Start()
})
if output == "enabled" {
// 检查文件路径是否变化
data, err := os.ReadFile("/etc/systemd/system/" + teaconst.SystemdServiceName + ".service")
if err == nil && bytes.Index(data, []byte(exe)) > 0 {
return nil
}
}
manager := utils.NewServiceManager(shortName, teaconst.ProductName)
err = manager.Install(exe, []string{})
if err != nil {
return err
}
} else {
manager := utils.NewServiceManager(shortName, teaconst.ProductName)
err = manager.Uninstall()
if err != nil {
return err
}
}
return nil
}