Skip to content

Commit 39de499

Browse files
committed
Delete list_nodes1
1 parent a79a621 commit 39de499

File tree

2 files changed

+18
-25
lines changed

2 files changed

+18
-25
lines changed

deps/rabbitmq_peer_discovery_k8s/src/rabbit_peer_discovery_k8s.erl

Lines changed: 14 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -38,11 +38,16 @@ init() ->
3838
-spec list_nodes() -> {ok, {Nodes :: list(), NodeType :: rabbit_types:node_type()}} | {error, Reason :: string()}.
3939

4040
list_nodes() ->
41-
case list_nodes1() of
42-
{ok, Nodes} ->
41+
case make_request() of
42+
{ok, Response} ->
43+
Addresses = extract_node_list(Response),
44+
Nodes = lists:map(fun node_name/1, Addresses),
4345
{ok, {Nodes, disc}};
44-
{error, _} = Error ->
45-
Error
46+
{error, Reason} ->
47+
Details = io_lib:format("Failed to fetch a list of nodes from Kubernetes API: ~s", [Reason]),
48+
rabbit_log:error(Details),
49+
send_event("Warning", "Failed", Details),
50+
{error, Reason}
4651
end.
4752

4853
-spec supports_registration() -> boolean().
@@ -66,9 +71,9 @@ unregister() ->
6671
-spec lock(Node :: node()) -> {ok, {ResourceId :: string(), LockRequesterId :: node()}} | {error, Reason :: string()}.
6772

6873
lock(Node) ->
69-
%% call list_nodes1/0 externally such that meck can mock the function
70-
case ?MODULE:list_nodes1() of
71-
{ok, Nodes} ->
74+
%% call list_nodes/0 externally such that meck can mock the function
75+
case ?MODULE:list_nodes() of
76+
{ok, {Nodes, disc}} ->
7277
case lists:member(Node, Nodes) of
7378
true ->
7479
rabbit_log:info("Will try to lock connecting to nodes ~p", [Nodes]),
@@ -93,8 +98,8 @@ lock(Node) ->
9398
-spec unlock({ResourceId :: string(), LockRequesterId :: node()}) -> ok | {error, Reason :: string()}.
9499

95100
unlock(LockId) ->
96-
case ?MODULE:list_nodes1() of
97-
{ok, Nodes} ->
101+
case ?MODULE:list_nodes() of
102+
{ok, {Nodes, disc}} ->
98103
global:del_lock(LockId, Nodes),
99104
ok;
100105
{error, _} = Error ->
@@ -238,15 +243,3 @@ send_event(Type, Reason, Message) ->
238243
[{ssl, [{cacertfile, get_config_key(k8s_cert_path, M)}]}],
239244
Body
240245
).
241-
242-
list_nodes1() ->
243-
case make_request() of
244-
{ok, Response} ->
245-
Addresses = extract_node_list(Response),
246-
{ok, lists:map(fun node_name/1, Addresses)};
247-
{error, Reason} ->
248-
Details = io_lib:format("Failed to fetch a list of nodes from Kubernetes API: ~s", [Reason]),
249-
rabbit_log:error(Details),
250-
send_event("Warning", "Failed", Details),
251-
{error, Reason}
252-
end.

deps/rabbitmq_peer_discovery_k8s/test/rabbitmq_peer_discovery_k8s_SUITE.erl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,7 @@ event_v1_test(_Config) ->
145145

146146
lock_single_node(_Config) ->
147147
LocalNode = node(),
148-
meck:expect(rabbit_peer_discovery_k8s, list_nodes1, 0, {ok, [LocalNode]}),
148+
meck:expect(rabbit_peer_discovery_k8s, list_nodes, 0, {ok, {[LocalNode], disc}}),
149149

150150
{ok, LockId} = rabbit_peer_discovery_k8s:lock(LocalNode),
151151
?assertEqual(ok, rabbit_peer_discovery_k8s:unlock(LockId)).
@@ -154,7 +154,7 @@ lock_multiple_nodes(_Config) ->
154154
application:set_env(rabbit, cluster_formation, [{internal_lock_retries, 2}]),
155155
LocalNode = node(),
156156
OtherNode = other@host,
157-
meck:expect(rabbit_peer_discovery_k8s, list_nodes1, 0, {ok, [OtherNode, LocalNode]}),
157+
meck:expect(rabbit_peer_discovery_k8s, list_nodes, 0, {ok, {[OtherNode, LocalNode], disc}}),
158158

159159
{ok, {LockResourceId, OtherNode}} = rabbit_peer_discovery_k8s:lock(OtherNode),
160160
?assertEqual({error, "Acquiring lock taking too long, bailing out after 2 retries"}, rabbit_peer_discovery_k8s:lock(LocalNode)),
@@ -163,10 +163,10 @@ lock_multiple_nodes(_Config) ->
163163
?assertEqual(ok, rabbitmq_peer_discovery_k8s:unlock({LockResourceId, LocalNode})).
164164

165165
lock_local_node_not_discovered(_Config) ->
166-
meck:expect(rabbit_peer_discovery_k8s, list_nodes1, 0, {ok, [n1@host, n2@host]} ),
166+
meck:expect(rabbit_peer_discovery_k8s, list_nodes, 0, {ok, {[n1@host, n2@host], disc}} ),
167167
Expectation = {error, "Local node me@host is not part of discovered nodes [n1@host,n2@host]"},
168168
?assertEqual(Expectation, rabbit_peer_discovery_k8s:lock(me@host)).
169169

170170
lock_list_nodes_fails(_Config) ->
171-
meck:expect(rabbit_peer_discovery_k8s, list_nodes1, 0, {error, "K8s API unavailable"}),
171+
meck:expect(rabbit_peer_discovery_k8s, list_nodes, 0, {error, "K8s API unavailable"}),
172172
?assertEqual({error, "K8s API unavailable"}, rabbit_peer_discovery_k8s:lock(me@host)).

0 commit comments

Comments
 (0)