collab/hmq
3
0
Fork 0
mirror of https://github.com/fhmq/hmq.git synced 2026-04-26 11:38:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

config

Browse Source
This commit is contained in:
zhouyuyan
2017-08-25 15:28:41 +08:00
parent 0bc226d071
commit 0ea7da1dc0
8 changed files with 91 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
broker/client.go
View File

@@ -7,7 +7,7 @@ import (
"strings"
"sync"
"github.com/prometheus/common/log"
log "github.com/cihub/seelog"
)
type client struct {
@@ -130,6 +130,15 @@ func (c *client) ProcessPublish(buf []byte) {
}
c.ProcessPublishMessage(buf, msg)
if msg.Retain() {
if b := c.broker; b != nil {
err := b.rl.Insert(msg.Topic(), buf)
if err != nil {
log.Error("Insert Retain Message error: ", err)
}
}
}
}
func (c *client) ProcessPublishMessage(buf []byte, msg *message.PublishMessage) {
@@ -283,18 +292,17 @@ func (c *client) ProcessSubscribe(buf []byte) {
// srv.BroadcastSubscribeMessage(buf)
// })
// }
//process retain message
// for _, t := range topics {
// srv.startGoRoutine(func() {
// bufs := srv.rl.Match(t)
// for _, buf := range bufs {
// log.Info("process retain message: ", string(buf))
// if buf != nil && string(buf) != "" {
// c.writeBuffer(buf)
// }
// }
// })
// }
for _, t := range topics {
bufs := srv.rl.Match(t)
for _, buf := range bufs {
log.Info("process retain message: ", string(buf))
if buf != nil && string(buf) != "" {
c.writeBuffer(buf)
}
}
}
}
func (c *client) ProcessUnSubscribe(buf []byte) {
@@ -366,16 +374,11 @@ func (c *client) Close() {
srv := c.broker
subs := c.subs
if srv != nil {
// srv.removeClient(c)
for _, sub := range subs {
// log.Info("remove Sub")
err := srv.sl.Remove(sub)
if err != nil {
log.Error("closed client but remove sublist error, ", err)
}
// if c.typ == CLIENT {
// srv.BroadcastUnSubscribe(sub)
// }
}
}
if c.conn != nil {