mirror of
https://github.com/fhmq/hmq.git
synced 2026-05-04 07:08:32 +00:00
108 lines
2.2 KiB
Go
108 lines
2.2 KiB
Go
/* Copyright (c) 2018, joy.zhou <chowyu08@gmail.com>
|
|
*/
|
|
package broker
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/tidwall/gjson"
|
|
|
|
"github.com/eclipse/paho.mqtt.golang/packets"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func (c *client) SendInfo() {
|
|
if c.status == Disconnected {
|
|
return
|
|
}
|
|
url := c.info.localIP + ":" + c.broker.config.RpcPort
|
|
|
|
infoMsg := NewInfo(c.broker.id, url, false)
|
|
err := c.WriterPacket(infoMsg)
|
|
if err != nil {
|
|
log.Error("send info message error, ", zap.Error(err))
|
|
return
|
|
}
|
|
}
|
|
|
|
func (c *client) StartPing() {
|
|
timeTicker := time.NewTicker(time.Second * 50)
|
|
ping := packets.NewControlPacket(packets.Pingreq).(*packets.PingreqPacket)
|
|
for {
|
|
select {
|
|
case <-timeTicker.C:
|
|
err := c.WriterPacket(ping)
|
|
if err != nil {
|
|
log.Error("ping error: ", zap.Error(err))
|
|
c.Close()
|
|
}
|
|
case <-c.ctx.Done():
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (c *client) SendConnect() {
|
|
|
|
if c.status != Connected {
|
|
return
|
|
}
|
|
m := packets.NewControlPacket(packets.Connect).(*packets.ConnectPacket)
|
|
|
|
m.CleanSession = true
|
|
m.ClientIdentifier = c.info.clientID
|
|
m.Keepalive = uint16(60)
|
|
err := c.WriterPacket(m)
|
|
if err != nil {
|
|
log.Error("send connect message error, ", zap.Error(err))
|
|
return
|
|
}
|
|
log.Info("send connect success")
|
|
}
|
|
|
|
func NewInfo(sid, url string, isforword bool) *packets.PublishPacket {
|
|
pub := packets.NewControlPacket(packets.Publish).(*packets.PublishPacket)
|
|
pub.Qos = 0
|
|
pub.TopicName = BrokerInfoTopic
|
|
pub.Retain = false
|
|
info := fmt.Sprintf(`{"brokerID":"%s","brokerUrl":"%s"}`, sid, url)
|
|
// log.Info("new info", string(info))
|
|
pub.Payload = []byte(info)
|
|
return pub
|
|
}
|
|
|
|
func (c *client) ProcessInfo(packet *packets.PublishPacket) {
|
|
nc := c.conn
|
|
b := c.broker
|
|
if nc == nil {
|
|
return
|
|
}
|
|
|
|
log.Info("recv remoteInfo: ", zap.String("payload", string(packet.Payload)))
|
|
|
|
routes := gjson.GetBytes(packet.Payload, "data").Map()
|
|
if len(routes) == 0 {
|
|
log.Error("receive nil info message ", zap.String("info", string(packet.Payload)))
|
|
return
|
|
}
|
|
|
|
b.nodes = routes
|
|
|
|
b.mu.Lock()
|
|
for rid, rurl := range routes {
|
|
if rid == b.id {
|
|
continue
|
|
}
|
|
|
|
url := rurl.String()
|
|
if url != "" {
|
|
if _, exist := b.rpcClient[rid]; !exist {
|
|
b.initRPCClient(rid, url)
|
|
}
|
|
}
|
|
|
|
}
|
|
b.mu.Unlock()
|
|
}
|