'fhmq'

broker/info.go

@@ -0,0 +1,113 @@
+package broker
+
+import (
+	"fhmq/lib/message"
+	"fmt"
+	"time"
+
+	simplejson "github.com/bitly/go-simplejson"
+	log "github.com/cihub/seelog"
+)
+
+func (c *client) SendInfo() {
+	url := c.info.localIP + ":" + c.broker.config.Cluster.Port
+
+	infoMsg := NewInfo(c.broker.id, url, false)
+	err := c.writeMessage(infoMsg)
+	if err != nil {
+		log.Error("send info message error, ", err)
+		return
+	}
+	// log.Info("send info success")
+}
+
+func (c *client) StartPing() {
+	timeTicker := time.NewTicker(time.Second * 30)
+	ping := message.NewPingreqMessage()
+	for {
+		select {
+		case <-timeTicker.C:
+			err := c.writeMessage(ping)
+			if err != nil {
+				log.Error("ping error: ", err)
+			}
+		}
+	}
+}
+
+func (c *client) SendConnect() {
+
+	clientID := c.info.clientID
+	connMsg := message.NewConnectMessage()
+	connMsg.SetClientId(clientID)
+	connMsg.SetVersion(0x04)
+	err := c.writeMessage(connMsg)
+	if err != nil {
+		log.Error("send connect message error, ", err)
+		return
+	}
+	// log.Info("send connet success")
+}
+
+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
+}
+
+func (c *client) ProcessInfo(msg *message.PublishMessage) {
+	nc := c.conn
+	b := c.broker
+	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 == b.id {
+		if !isForward {
+			c.Close() //close connet self
+		}
+		return
+	}
+
+	exist := b.CheckRemoteExist(rid, rurl)
+	if !exist {
+		go b.connectRouter(rurl, rid)
+	}
+	// log.Info("isforword: ", isForward)
+	if !isForward {
+		route := &route{
+			remoteUrl: rurl,
+			remoteID:  rid,
+		}
+		c.route = route
+
+		go b.SendLocalSubsToRouter(c)
+		// log.Info("BroadcastInfoMessage starting... ")
+		infoMsg := NewInfo(rid, rurl, true)
+		b.BroadcastInfoMessage(rid, infoMsg)
+	}
+
+	return
+}