This repository has been archived by the owner on Nov 8, 2019. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 3
/
info.go
77 lines (65 loc) · 1.56 KB
/
info.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
package server
import (
"fmt"
simplejson "github.com/bitly/go-simplejson"
log "github.com/cihub/seelog"
"github.com/surgemq/message"
)
func (c *client) ProcessInfo(msg *message.PublishMessage) {
nc := c.nc
s := c.srv
if nc == nil {
return
}
log.Info("recv remoteInfo: ", string(msg.Payload()))
js, e := simplejson.NewJson(msg.Payload())
if e != nil {
log.Warn("parse info message err", e)
return
}
rid := js.Get("remoteID").MustString()
rurl := js.Get("url").MustString()
isForward := js.Get("isForward").MustBool()
if rid == "" {
log.Error("receive info message error with remoteID is null")
return
}
if rid == s.ID {
log.Info("recv self info")
if !isForward {
c.Close() //close connet self
}
return
}
exist := s.CheckRemoteExist(rid, rurl)
if !exist {
s.startGoRoutine(func() {
s.connectRouter(rurl, rid)
})
}
// log.Info("isforword: ", isForward)
if !isForward {
route := &Route{
remoteUrl: rurl,
remoteID: rid,
}
c.route = route
s.startGoRoutine(func() {
s.SendLocalSubsToRouter(c)
})
// log.Info("BroadcastInfoMessage starting... ")
infoMsg := NewInfo(rid, rurl, true)
s.BroadcastInfoMessage(rid, infoMsg)
}
return
}
func NewInfo(sid, url string, isforword bool) *message.PublishMessage {
infoMsg := message.NewPublishMessage()
infoMsg.SetTopic([]byte(BrokerInfoTopic))
info := fmt.Sprintf(`{"remoteID":"%s","url":"%s","isForward":%t}`, sid, url, isforword)
// log.Info("new info", string(info))
infoMsg.SetPayload([]byte(info))
infoMsg.SetQoS(0)
infoMsg.SetRetain(false)
return infoMsg
}