syn.erl 7.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191
  1. %% ==========================================================================================================
  2. %% Syn - A global Process Registry and Process Group manager.
  3. %%
  4. %% The MIT License (MIT)
  5. %%
  6. %% Copyright (c) 2015-2019 Roberto Ostinelli <roberto@ostinelli.net> and Neato Robotics, Inc.
  7. %%
  8. %% Permission is hereby granted, free of charge, to any person obtaining a copy
  9. %% of this software and associated documentation files (the "Software"), to deal
  10. %% in the Software without restriction, including without limitation the rights
  11. %% to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  12. %% copies of the Software, and to permit persons to whom the Software is
  13. %% furnished to do so, subject to the following conditions:
  14. %%
  15. %% The above copyright notice and this permission notice shall be included in
  16. %% all copies or substantial portions of the Software.
  17. %%
  18. %% THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19. %% IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20. %% FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  21. %% AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22. %% LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  23. %% OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  24. %% THE SOFTWARE.
  25. %% ==========================================================================================================
  26. -module(syn).
  27. %% API
  28. -export([start/0, stop/0]).
  29. -export([register/2, register/3]).
  30. -export([reregister/2, reregister/3]).
  31. -export([unregister/1]).
  32. -export([whereis/1, whereis/2]).
  33. -export([registry_count/0, registry_count/1]).
  34. -export([join/2, join/3]).
  35. -export([leave/2]).
  36. -export([get_members/1, get_members/2]).
  37. -export([member/2]).
  38. -export([get_local_members/1, get_local_members/2]).
  39. -export([local_member/2]).
  40. -export([publish/2]).
  41. -export([publish_to_local/2]).
  42. -export([multi_call/2, multi_call/3, multi_call_reply/2]).
  43. -export([sync_from_node/2]).
  44. %% gen_server via interface
  45. -export([register_name/2, unregister_name/1, whereis_name/1, send/2]).
  46. %% ===================================================================
  47. %% API
  48. %% ===================================================================
  49. -spec start() -> ok.
  50. start() ->
  51. {ok, _} = application:ensure_all_started(syn),
  52. ok.
  53. -spec stop() -> ok | {error, Reason :: any()}.
  54. stop() ->
  55. application:stop(syn).
  56. %% ----- \/ registry -------------------------------------------------
  57. -spec register(Name :: any(), Pid :: pid()) -> ok | {error, Reason :: any()}.
  58. register(Name, Pid) ->
  59. syn_registry:register(Name, Pid).
  60. -spec register(Name :: any(), Pid :: pid(), Meta :: any()) -> ok | {error, Reason :: any()}.
  61. register(Name, Pid, Meta) ->
  62. syn_registry:register(Name, Pid, Meta).
  63. -spec reregister(Name :: any(), Pid :: pid()) -> ok | {error, Reason :: any()}.
  64. reregister(Name, Pid) ->
  65. syn_registry:reregister(Name, Pid).
  66. -spec reregister(Name :: any(), Pid :: pid(), Meta :: any()) -> ok | {error, Reason :: any()}.
  67. reregister(Name, Pid, Meta) ->
  68. syn_registry:reregister(Name, Pid, Meta).
  69. -spec unregister(Name :: any()) -> ok | {error, Reason :: any()}.
  70. unregister(Name) ->
  71. syn_registry:unregister(Name).
  72. -spec whereis(Name :: any()) -> pid() | undefined.
  73. whereis(Name) ->
  74. syn_registry:whereis(Name).
  75. -spec whereis(Name :: any(), with_meta) -> {pid(), Meta :: any()} | undefined.
  76. whereis(Name, with_meta) ->
  77. syn_registry:whereis(Name, with_meta).
  78. -spec registry_count() -> non_neg_integer().
  79. registry_count() ->
  80. syn_registry:count().
  81. -spec registry_count(Node :: atom()) -> non_neg_integer().
  82. registry_count(Node) ->
  83. syn_registry:count(Node).
  84. %% ----- \/ gen_server via module interface --------------------------
  85. -spec register_name(Name :: any(), Pid :: pid()) -> yes | no.
  86. register_name(Name, Pid) ->
  87. case syn_registry:register(Name, Pid) of
  88. ok -> yes;
  89. _ -> no
  90. end.
  91. -spec unregister_name(Name :: any()) -> any().
  92. unregister_name(Name) ->
  93. case syn_registry:unregister(Name) of
  94. ok -> Name;
  95. _ -> nil
  96. end.
  97. -spec whereis_name(Name :: any()) -> pid() | undefined.
  98. whereis_name(Name) ->
  99. syn_registry:whereis(Name).
  100. -spec send(Name :: any(), Message :: any()) -> pid().
  101. send(Name, Message) ->
  102. case whereis_name(Name) of
  103. undefined ->
  104. {badarg, {Name, Message}};
  105. Pid ->
  106. Pid ! Message,
  107. Pid
  108. end.
  109. %% ----- \/ groups ---------------------------------------------------
  110. -spec join(GroupName :: any(), Pid :: pid()) -> ok.
  111. join(GroupName, Pid) ->
  112. syn_groups:join(GroupName, Pid).
  113. -spec join(GroupName :: any(), Pid :: pid(), Meta :: any()) -> ok.
  114. join(GroupName, Pid, Meta) ->
  115. syn_groups:join(GroupName, Pid, Meta).
  116. -spec leave(GroupName :: any(), Pid :: pid()) -> ok | {error, Reason :: any()}.
  117. leave(GroupName, Pid) ->
  118. syn_groups:leave(GroupName, Pid).
  119. -spec get_members(GroupName :: any()) -> [pid()].
  120. get_members(GroupName) ->
  121. syn_groups:get_members(GroupName).
  122. -spec get_members(GroupName :: any(), with_meta) -> [{pid(), Meta :: any()}].
  123. get_members(GroupName, with_meta) ->
  124. syn_groups:get_members(GroupName, with_meta).
  125. -spec member(GroupName :: any(), Pid :: pid()) -> boolean().
  126. member(GroupName, Pid) ->
  127. syn_groups:member(GroupName, Pid).
  128. -spec get_local_members(GroupName :: any()) -> [pid()].
  129. get_local_members(GroupName) ->
  130. syn_groups:get_local_members(GroupName).
  131. -spec get_local_members(GroupName :: any(), with_meta) -> [{pid(), Meta :: any()}].
  132. get_local_members(GroupName, with_meta) ->
  133. syn_groups:get_local_members(GroupName, with_meta).
  134. -spec local_member(GroupName :: any(), Pid :: pid()) -> boolean().
  135. local_member(GroupName, Pid) ->
  136. syn_groups:local_member(GroupName, Pid).
  137. -spec publish(GroupName :: any(), Message :: any()) -> {ok, RecipientCount :: non_neg_integer()}.
  138. publish(GroupName, Message) ->
  139. syn_groups:publish(GroupName, Message).
  140. -spec publish_to_local(GroupName :: any(), Message :: any()) -> {ok, RecipientCount :: non_neg_integer()}.
  141. publish_to_local(GroupName, Message) ->
  142. syn_groups:publish_to_local(GroupName, Message).
  143. -spec multi_call(GroupName :: any(), Message :: any()) ->
  144. {[{pid(), Reply :: any()}], [BadPid :: pid()]}.
  145. multi_call(GroupName, Message) ->
  146. syn_groups:multi_call(GroupName, Message).
  147. -spec multi_call(GroupName :: any(), Message :: any(), Timeout :: non_neg_integer()) ->
  148. {[{pid(), Reply :: any()}], [BadPid :: pid()]}.
  149. multi_call(GroupName, Message, Timeout) ->
  150. syn_groups:multi_call(GroupName, Message, Timeout).
  151. -spec multi_call_reply(CallerPid :: pid(), Reply :: any()) -> {syn_multi_call_reply, pid(), Reply :: any()}.
  152. multi_call_reply(CallerPid, Reply) ->
  153. syn_groups:multi_call_reply(CallerPid, Reply).
  154. %% ----- \/ anti entropy ---------------------------------------------
  155. -spec sync_from_node(registry | groups, RemoteNode :: node()) -> ok | {error, Reason :: any()}.
  156. sync_from_node(registry, RemoteNode) ->
  157. syn_registry:sync_from_node(RemoteNode);
  158. sync_from_node(groups, RemoteNode) ->
  159. syn_groups:sync_from_node(RemoteNode).