apps/mochi-mqtt
1
0
Fork 0
mirror of https://github.com/mochi-mqtt/server.git synced 2025-11-03 02:23:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Optionally drop a packet if the ErrRejectPacket error is returned from OnProcessMessage

Browse Source
This commit is contained in:
mochi
2022-03-31 17:53:13 +01:00
parent 7b9bc844c1
commit 56e8039093
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
server/server.go
View File

@@ -475,8 +475,16 @@ func (s *Server) processPublish(cl *clients.Client, pk packets.Packet) error {
// if an OnProcessMessage hook exists, potentially modify the packet.
if s.Events.OnProcessMessage != nil {
if pkx, err := s.Events.OnProcessMessage(cl.Info(), events.Packet(pk)); err == nil {
pk = packets.Packet(pkx)
pkx, err := s.Events.OnProcessMessage(cl.Info(), events.Packet(pk))
if err == nil {
pk = packets.Packet(pkx) // Only use the new package changes if there's no errors.
} else {
// If the ErrRejectPacket is return, abandon processing the packet.
if err == ErrRejectPacket {
return nil
}
s.Events.OnError(cl.Info(), err)
}
}