Add namespace prefix for wolfmq_workers_sup

erlangbureau · Dec 13, 2018 · 66ebeef · 66ebeef
1 parent cfe0c21
commit 66ebeef
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/src/wolfmq.erl b/src/wolfmq.erl
@@ -23,7 +23,7 @@ push({GroupId, _QId} = QueueId, Msg, Opts) when is_atom(GroupId) ->
     end,
     add_to_queue(QueueId, Msg);
 push(QueueId, Msg, Opts) ->
-    push({wolfmq_default_group, QueueId}, Msg, Opts).
+    push({default, QueueId}, Msg, Opts).
 
 size({GroupId, _QId} = QueueId) when is_atom(GroupId) ->
     case wolfmq_queues_catalog:is_existing(QueueId) of
@@ -35,7 +35,7 @@ size({GroupId, _QId} = QueueId) when is_atom(GroupId) ->
             0
     end;
 size(QueueId) ->
-    size({wolfmq_default_group, QueueId}).
+    size({default, QueueId}).
 
 %% internal
 add_to_queue(ExternalQueueId, Msgs) when is_list(Msgs) ->

diff --git a/src/wolfmq_workers_sup.erl b/src/wolfmq_workers_sup.erl
@@ -10,10 +10,12 @@
 
 %% API
 start_link(GroupId) ->
-    supervisor:start_link({local, GroupId}, ?MODULE, []).
+    Name = group_id_to_process_name(GroupId),
+    supervisor:start_link({local, Name}, ?MODULE, []).
 
 start_worker(GroupId, QueueId, Opts) ->
-    supervisor:start_child(GroupId, [[QueueId, Opts]]).
+    Name = group_id_to_process_name(GroupId),
+    supervisor:start_child(Name, [[QueueId, Opts]]).
 
 %% supervisor callbacks
 init([]) ->
@@ -26,3 +28,9 @@ init([]) ->
         modules     => [wolfmq_worker]
     },
     {ok, {{simple_one_for_one, 250, 5}, [Worker]}}.
+
+%% internal
+group_id_to_process_name(GroupId) ->
+    Id = atom_to_binary(GroupId, utf8),
+    Name = <<"wolfmq_group_sup_", Id/binary>>,
+    binary_to_atom(Name, utf8).