apps/mochi-mqtt
1
0
Fork 0
mirror of https://github.com/mochi-mqtt/server.git synced 2025-10-30 03:01:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Client Tests

Browse Source
This commit is contained in:
Mochi
2019-10-07 10:19:20 +01:00
parent e7f1576cc1
commit 9b37eece4e
3 changed files with 191 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mqtt.go
View File

@@ -3,6 +3,7 @@ package mqtt
import (
"bufio"
"errors"
"log"
"net"
"github.com/mochi-co/mqtt/auth"
@@ -261,6 +262,7 @@ func (s *Server) processPacket(cl *client, pk packets.Packet) error {
})
if err != nil {
s.closeClient(cl, true)
return err
}
case *packets.PublishPacket:
@@ -295,6 +297,7 @@ func (s *Server) processPacket(cl *client, pk packets.Packet) error {
err := s.writeClient(client, out)
if err != nil {
s.closeClient(client, true)
return err
}
// If QoS byte is set, save as message to inflight index so we
@@ -318,6 +321,7 @@ func (s *Server) processPacket(cl *client, pk packets.Packet) error {
err := s.writeClient(cl, out)
if err != nil {
s.closeClient(cl, true)
return err
}
cl.inFlight.set(out.PacketID, out)
@@ -333,6 +337,7 @@ func (s *Server) processPacket(cl *client, pk packets.Packet) error {
err := s.writeClient(cl, out)
if err != nil {
s.closeClient(cl, true)
return err
}
cl.inFlight.delete(msg.PacketID)
@@ -358,8 +363,10 @@ func (s *Server) processPacket(cl *client, pk packets.Packet) error {
PacketID: msg.PacketID,
ReturnCodes: retCodes,
})
if err != nil {
s.closeClient(cl, true)
return err
}
// Publish out any retained messages matching the subscription filter.
@@ -368,7 +375,9 @@ func (s *Server) processPacket(cl *client, pk packets.Packet) error {
for _, pkv := range messages {
err := s.writeClient(cl, pkv)
if err != nil {
log.Println("write err B", err)
s.closeClient(cl, true)
return err
}
}
}
@@ -385,6 +394,7 @@ func (s *Server) processPacket(cl *client, pk packets.Packet) error {
})
if err != nil {
s.closeClient(cl, true)
return err
}
}