yuanbiao
/
emqx
1
0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add hook message.acked

This commit is contained in:
terry-xiaoyu 2019-08-09 19:12:56 +08:00
parent a93f8c6788
commit a19777c2e6
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/emqx_session.erl
View File

@ -378,6 +378,7 @@ do_publish(PacketId, Msg = #message{timestamp = Ts},
puback(PacketId, Session = #session{inflight = Inflight}) -> puback(PacketId, Session = #session{inflight = Inflight}) ->
case emqx_inflight:lookup(PacketId, Inflight) of case emqx_inflight:lookup(PacketId, Inflight) of
{value, {Msg, _Ts}} when is_record(Msg, message) -> {value, {Msg, _Ts}} when is_record(Msg, message) ->
ok = emqx_hooks:run('message.acked', [Msg]),
Inflight1 = emqx_inflight:delete(PacketId, Inflight), Inflight1 = emqx_inflight:delete(PacketId, Inflight),
dequeue(Session#session{inflight = Inflight1}); dequeue(Session#session{inflight = Inflight1});
{value, {_OtherPub, _Ts}} -> {value, {_OtherPub, _Ts}} ->
@ -398,6 +399,7 @@ puback(PacketId, Session = #session{inflight = Inflight}) ->
pubrec(PacketId, Session = #session{inflight = Inflight}) -> pubrec(PacketId, Session = #session{inflight = Inflight}) ->
case emqx_inflight:lookup(PacketId, Inflight) of case emqx_inflight:lookup(PacketId, Inflight) of
{value, {Msg, _Ts}} when is_record(Msg, message) -> {value, {Msg, _Ts}} when is_record(Msg, message) ->
ok = emqx_hooks:run('message.acked', [Msg]),
Inflight1 = emqx_inflight:update(PacketId, {pubrel, os:timestamp()}, Inflight), Inflight1 = emqx_inflight:update(PacketId, {pubrel, os:timestamp()}, Inflight),
{ok, Session#session{inflight = Inflight1}}; {ok, Session#session{inflight = Inflight1}};
{value, {pubrel, _Ts}} -> {value, {pubrel, _Ts}} ->