1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
## This Source Code Form is subject to the terms of the Mozilla Public
## License, v. 2.0. If a copy of the MPL was not distributed with this
## file, You can obtain one at https://mozilla.org/MPL/2.0/.
##
## Copyright (c) 2007-2023 Broadcom. All Rights Reserved. The term “Broadcom” refers to Broadcom Inc. and/or its subsidiaries. All rights reserved.
defmodule RabbitMQ.CLI.Streams.Commands.RestartStreamCommand do
alias RabbitMQ.CLI.Core.DocGuide
import RabbitMQ.CLI.Core.DataCoercion
@behaviour RabbitMQ.CLI.CommandBehaviour
def merge_defaults(args, opts) do
{args, Map.merge(%{vhost: "/"}, opts)}
end
def switches() do
[preferred_leader_node: :string]
end
use RabbitMQ.CLI.Core.AcceptsOnePositionalArgument
use RabbitMQ.CLI.Core.RequiresRabbitAppRunning
def run([name], %{vhost: vhost, node: node_name} = switches) do
preferred = Map.get(switches, :preferred_leader_node, :undefined)
options = %{preferred_leader_node: to_atom(preferred)}
case :rabbit_misc.rpc_call(node_name, :rabbit_stream_queue, :restart_stream, [
vhost,
name,
options
]) do
{:error, :classic_queue_not_supported} ->
{:error, "Cannot restart a classic queue"}
{:error, :quorum_queue_not_supported} ->
{:error, "Cannot restart a quorum queue"}
{:ok, LeaderNode} ->
LeaderNode
other ->
other
end
end
use RabbitMQ.CLI.DefaultOutput
def usage, do: "restart_stream [--vhost <vhost>] <stream> [--preferred-leader-node <node>]"
def usage_additional do
[
["<stream>", "stream name"],
["--preferred-leader-node", "preferred leader node"]
]
end
def usage_doc_guides() do
[
DocGuide.streams()
]
end
def help_section, do: :replication
def description, do: "Restarts a stream."
def banner([name], _) do
[
"Restarting stream #{name}..."
]
end
end
|
