From 5f2adb42ed229c93e161567abea1681d0bb27944 Mon Sep 17 00:00:00 2001
From: JianBo He <heeejianbo@163.com>
Date: Sun, 26 Sep 2021 18:04:16 +0800
Subject: [PATCH] chore: export keepalive funcs for mqtt-sn

---
 src/emqx_channel.erl | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/emqx_channel.erl b/src/emqx_channel.erl
index 679d2c72a..e0789382a 100644
--- a/src/emqx_channel.erl
+++ b/src/emqx_channel.erl
@@ -49,7 +49,10 @@
         ]).
 
 %% Export for emqx_sn
--export([do_deliver/2]).
+-export([ do_deliver/2
+        , ensure_keepalive/2
+        , clear_keepalive/1
+        ]).
 
 %% Exports for CT
 -export([set_field/3]).
@@ -1512,6 +1515,15 @@ ensure_keepalive_timer(Interval, Channel = #channel{clientinfo = #{zone := Zone}
     Keepalive = emqx_keepalive:init(round(timer:seconds(Interval) * Backoff)),
     ensure_timer(alive_timer, Channel#channel{keepalive = Keepalive}).
 
+clear_keepalive(Channel = #channel{timers = Timers}) ->
+    case maps:get(alive_timer, Timers, undefined) of
+        undefined ->
+            Channel;
+        TRef ->
+            emqx_misc:cancel_timer(TRef),
+            Channel#channel{timers = maps:without([alive_timer], Timers)}
+    end.
+
 %%--------------------------------------------------------------------
 %% Maybe Resume Session