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 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112
|
## 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.Queues.Commands.ShrinkCommand do
alias RabbitMQ.CLI.Core.DocGuide
import RabbitMQ.CLI.Core.DataCoercion
@behaviour RabbitMQ.CLI.CommandBehaviour
def switches() do
[errors_only: :boolean]
end
def merge_defaults(args, opts) do
{args, Map.merge(%{errors_only: false}, opts)}
end
use RabbitMQ.CLI.Core.AcceptsOnePositionalArgument
use RabbitMQ.CLI.Core.RequiresRabbitAppRunning
def run([node], %{node: node_name, errors_only: errs}) do
case :rabbit_misc.rpc_call(node_name, :rabbit_quorum_queue, :shrink_all, [to_atom(node)]) do
{:error, _} = error ->
error
{:badrpc, _} = error ->
error
results when errs ->
for {{:resource, vhost, _kind, name}, {:error, _, _} = res} <- results,
do: [
{:vhost, vhost},
{:name, name},
{:size, format_size(res)},
{:result, format_result(res)}
]
results ->
for {{:resource, vhost, _kind, name}, res} <- results,
do: [
{:vhost, vhost},
{:name, name},
{:size, format_size(res)},
{:result, format_result(res)}
]
end
end
use RabbitMQ.CLI.DefaultOutput
def formatter(), do: RabbitMQ.CLI.Formatters.Table
def banner([node], _) do
"Shrinking quorum queues on #{node}..."
end
def usage, do: "shrink <node> [--errors-only]"
def usage_additional() do
[
["<node>", "node name to remove replicas from"],
["--errors-only", "only list queues which reported an error"]
]
end
def usage_doc_guides() do
[
DocGuide.quorum_queues()
]
end
def help_section, do: :cluster_management
def description,
do: "Shrinks quorum queue clusters by removing any members (replicas) on the given node."
#
# Implementation
#
defp format_size({:ok, size}) do
size
end
defp format_size({:error, _size, :timeout}) do
# the actual size is uncertain here
"?"
end
defp format_size({:error, size, _}) do
size
end
defp format_result({:ok, _size}) do
"ok"
end
defp format_result({:error, _size, :last_node}) do
"error: the last node cannot be removed"
end
defp format_result({:error, _size, :timeout}) do
"error: the operation timed out and may not have been completed"
end
defp format_result({:error, _size, err}) do
to_string(:io_lib.format("error: ~W", [err, 10]))
end
end
|