-
Notifications
You must be signed in to change notification settings - Fork 218
/
peer.go
87 lines (74 loc) · 2.12 KB
/
peer.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
/*
* Radon
*
* Copyright 2018 The Radon Authors.
* Code is licensed under the GPLv3.
*
*/
package v1
import (
"net/http"
"proxy"
"github.com/ant0ine/go-json-rest/rest"
"github.com/xelabs/go-mysqlstack/xlog"
)
type peerParams struct {
Address string `json:"address"`
}
// AddPeerHandler impl.
func AddPeerHandler(log *xlog.Log, proxy *proxy.Proxy) rest.HandlerFunc {
f := func(w rest.ResponseWriter, r *rest.Request) {
addPeerHandler(log, proxy, w, r)
}
return f
}
func addPeerHandler(log *xlog.Log, proxy *proxy.Proxy, w rest.ResponseWriter, r *rest.Request) {
syncer := proxy.Syncer()
p := peerParams{}
err := r.DecodeJsonPayload(&p)
if err != nil {
log.Error("api.v1.add.peer.error:%+v", err)
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
log.Warning("api.v1.add.peer[%+v].from[%v]", p, r.RemoteAddr)
if err := syncer.AddPeer(p.Address); err != nil {
log.Error("api.v1.add.peer[%+v].error:%+v", p, err)
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
}
// RemovePeerHandler impl.
func RemovePeerHandler(log *xlog.Log, proxy *proxy.Proxy) rest.HandlerFunc {
f := func(w rest.ResponseWriter, r *rest.Request) {
removePeerHandler(log, proxy, w, r)
}
return f
}
func removePeerHandler(log *xlog.Log, proxy *proxy.Proxy, w rest.ResponseWriter, r *rest.Request) {
syncer := proxy.Syncer()
p := peerParams{}
err := r.DecodeJsonPayload(&p)
if err != nil {
log.Error("api.v1.remove.peer.error:%+v", err)
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
log.Warning("api.v1.remove.peer[%+v].from[%v]", p, r.RemoteAddr)
if err := syncer.RemovePeer(p.Address); err != nil {
log.Error("api.v1.remove.peer[%+v].error:%+v", p, err)
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
}
// PeerzHandler impl.
func PeerzHandler(log *xlog.Log, proxy *proxy.Proxy) rest.HandlerFunc {
f := func(w rest.ResponseWriter, r *rest.Request) {
peerzHandler(log, proxy, w, r)
}
return f
}
func peerzHandler(log *xlog.Log, proxy *proxy.Proxy, w rest.ResponseWriter, r *rest.Request) {
syncer := proxy.Syncer()
w.WriteJson(syncer.Peers())
}