From ca0fb214a7095b3ed0b2613fe162d5a175d23e54 Mon Sep 17 00:00:00 2001
From: k32 <10274441+k32@users.noreply.github.com>
Date: Mon, 16 Aug 2021 15:42:20 +0200
Subject: [PATCH] feat(emqx-sn): Add tables to the SN shard

---
 apps/emqx_gateway/src/mqttsn/emqx_sn_registry.erl | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/apps/emqx_gateway/src/mqttsn/emqx_sn_registry.erl b/apps/emqx_gateway/src/mqttsn/emqx_sn_registry.erl
index 1249831cc..c29f290b8 100644
--- a/apps/emqx_gateway/src/mqttsn/emqx_sn_registry.erl
+++ b/apps/emqx_gateway/src/mqttsn/emqx_sn_registry.erl
@@ -149,9 +149,11 @@ init([InstaId, PredefTopics]) ->
                 {ram_copies, [node()]},
                 {record_name, emqx_sn_registry},
                 {attributes, record_info(fields, emqx_sn_registry)},
-                {storage_properties, [{ets, [{read_concurrency, true}]}]}
+                {storage_properties, [{ets, [{read_concurrency, true}]}]},
+                {rlog_shard, ?SN_SHARD}
                ]),
     ok = ekka_mnesia:copy_table(Tab, ram_copies),
+    ok = ekka_rlog:wait_for_shards([?SN_SHARD], infinity),
     % FIXME:
     %ok = ekka_rlog:wait_for_shards([?CM_SHARD], infinity),
     MaxPredefId = lists:foldl(