/
service.go
108 lines (94 loc) · 2.26 KB
/
service.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
package service
import (
"fmt"
"github.com/aiyijing/smart-gateway/internal/util"
"github.com/docker/docker/client"
"path"
)
const (
// Service Status
Running = "Running"
Stop = "Stop"
Unknown = "Unknown"
)
type Service interface {
Stop() error
Start() error
Restart() error
Describe() *Desc
}
type Desc struct {
Name string
Volume string
Config string
}
func (d *Desc) Equal(desc *Desc) bool {
return d.Name == desc.Name &&
d.Config == desc.Config &&
d.Volume == desc.Volume
}
type Manager struct {
service map[string]Service
docker *client.Client
}
func NewManager() *Manager {
return &Manager{service: map[string]Service{}}
}
func (m *Manager) AddService(name string, svc Service) error {
des := svc.Describe()
for _, v := range m.service {
if des.Equal(v.Describe()) {
return fmt.Errorf("service %s has exists", des.Name)
}
}
m.service[name] = svc
return nil
}
func (m *Manager) StartService(name string) error {
svc, ok := m.service[name]
if !ok {
return fmt.Errorf("service %s not exists", name)
}
return svc.Start()
}
func (m *Manager) StopService(name string) error {
svc, ok := m.service[name]
if !ok {
return fmt.Errorf("service %s not exists", name)
}
return svc.Stop()
}
func (m *Manager) RestartService(name string) error {
svc, ok := m.service[name]
if !ok {
return fmt.Errorf("service %s not exists", name)
}
return svc.Restart()
}
func (m *Manager) UpdateConfig(name string, data string) error {
svc, ok := m.service[name]
if !ok {
return fmt.Errorf("service %s not exists", name)
}
return m.updateConfig(data, svc.Describe())
}
func (m *Manager) GetCurrentConfig(name string) (string, error) {
svc, ok := m.service[name]
if !ok {
return "", fmt.Errorf("service %s not exists", name)
}
return m.getCurrentConfig(svc.Describe())
}
func (m *Manager) getCurrentConfig(desc *Desc) (string, error) {
mount := "/var/lib/docker/volumes/"
p := path.Join(mount, desc.Volume, "_data")
filePath := path.Join(p, desc.Config)
data, err := util.ReadAll(filePath)
return string(data), err
}
func (m *Manager) updateConfig(data string, desc *Desc) error {
mount := "/var/lib/docker/volumes/"
p := path.Join(mount, desc.Volume, "_data")
filePath := path.Join(p, desc.Config)
return util.WriteOnCreate(filePath, []byte(data))
}