/
api_rpc.go
145 lines (124 loc) · 3.44 KB
/
api_rpc.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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
package riff
import (
"fmt"
"io/ioutil"
"os"
"strings"
"github.com/teatak/riff/api"
"github.com/teatak/riff/common"
"gopkg.in/yaml.v2"
)
type Query struct{}
func (q *Query) SnapShot(_ struct{}, snap *string) error {
*snap = server.SnapShot
server.Logger.Printf(infoServerPrefix+"client get snapshot: %s", *snap)
return nil
}
func (q *Query) Nodes(_ struct{}, nodes *api.Nodes) error {
server.Logger.Printf(infoServerPrefix + "client get nodes")
*nodes = server.api.Nodes()
return nil
}
func (q *Query) Node(p api.ParamNode, node *api.Node) error {
server.Logger.Printf(infoServerPrefix+"client get node %s", p.Name)
n := server.api.Node(p.Name)
if n == nil {
return fmt.Errorf("node %s not found", p.Name)
}
*node = *n
return nil
}
func (q *Query) Services(_ struct{}, services *api.Services) error {
server.Logger.Printf(infoServerPrefix + "client get services")
*services = server.api.Services()
return nil
}
func (q *Query) Service(p api.ParamService, service *api.Service) error {
server.Logger.Printf(infoServerPrefix+"client get service %s", p.Name)
s := server.api.Service(p.Name, p.State)
if s == nil {
return fmt.Errorf("service %s not found", p.Name)
}
*service = *s
return nil
}
type Mutation struct{}
func (m *Mutation) Reload(_ struct{}, result *bool) error {
server.initServices()
*result = true
return nil
}
func (m *Mutation) RegisteService(config string, result *bool) error {
serviceConfig := &Config{}
if err := yaml.Unmarshal([]byte(config), serviceConfig); err != nil {
return fmt.Errorf("config file error")
}
//write to config file
file := common.BinDir + "/config/" + serviceConfig.Name + ".yml"
_ = ioutil.WriteFile(file, []byte(config), 0666)
s := server.Self.LoadService(serviceConfig.Name)
server.Self.Services[s.Name] = s
server.Self.Shutter()
server.Shutter()
*result = true
server.Logger.Printf(infoServerPrefix + "client add new service")
return nil
}
func (m *Mutation) UnregisteService(name string, result *bool) error {
*result = true
if server.Self.Services[name] != nil {
if server.Self.Services[name].State == api.StateAlive {
if err := server.Self.Services[name].Stop(); err != nil {
server.Logger.Printf(errorServicePrefix+"client stop service %s error", name)
}
}
}
delete(server.Self.Services, name)
file := common.BinDir + "/config/" + name + ".yml"
run := common.BinDir + "/run/" + name + ".ver"
update := common.BinDir + "/update/" + name
_ = os.Remove(file)
_ = os.Remove(run)
_ = os.RemoveAll(update)
server.Logger.Printf(infoServerPrefix+"client remove service %s", name)
return nil
}
func (m *Mutation) Service(p api.ParamServiceMutation, reply *bool) (err error) {
server.Logger.Printf(infoServerPrefix+"client %s service %s", strings.ToLower(p.Cmd.Name()), p.Name)
s := server.Self.Services[p.Name]
if s == nil {
*reply = false
err = fmt.Errorf("service %s not found", p.Name)
return
}
switch p.Cmd {
case api.CmdStart:
_ = s.Start()
case api.CmdStop:
_ = s.Stop()
case api.CmdRestart:
_ = s.Restart()
}
*reply = true
return nil
}
type Riff struct{}
// push request a digest
func (r *Riff) Request(snap string, digests *[]*Digest) error {
if snap == server.SnapShot {
*digests = nil
} else {
//build digest
*digests = server.MakeDigest()
}
return nil
}
// push changes
func (r *Riff) PushDiff(diff []*Node, remoteDiff *[]*Node) error {
if len(diff) == 0 {
*remoteDiff = nil
} else {
*remoteDiff = server.MergeDiff(diff)
}
return nil
}