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
|
%% 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) 2011-2020 VMware, Inc. or its affiliates. All rights reserved.
%%
-module(rabbit_mgmt_wm_exchange_publish).
-export([init/2, resource_exists/2, is_authorized/2,
allowed_methods/2, content_types_provided/2, accept_content/2,
content_types_accepted/2]).
-export([variances/2]).
-include_lib("rabbitmq_management_agent/include/rabbit_mgmt_records.hrl").
-include_lib("amqp_client/include/amqp_client.hrl").
%%--------------------------------------------------------------------
init(Req, _State) ->
{cowboy_rest, rabbit_mgmt_headers:set_common_permission_headers(Req, ?MODULE), #context{}}.
variances(Req, Context) ->
{[<<"accept-encoding">>, <<"origin">>], Req, Context}.
allowed_methods(ReqData, Context) ->
{[<<"POST">>, <<"OPTIONS">>], ReqData, Context}.
content_types_provided(ReqData, Context) ->
{rabbit_mgmt_util:responder_map(to_json), ReqData, Context}.
resource_exists(ReqData, Context) ->
{case rabbit_mgmt_wm_exchange:exchange(ReqData) of
not_found -> false;
_ -> true
end, ReqData, Context}.
content_types_accepted(ReqData, Context) ->
{[{'*', accept_content}], ReqData, Context}.
accept_content(ReqData, Context) ->
rabbit_mgmt_util:post_respond(do_it(ReqData, Context)).
do_it(ReqData0, Context) ->
VHost = rabbit_mgmt_util:vhost(ReqData0),
X = rabbit_mgmt_util:id(exchange, ReqData0),
rabbit_mgmt_util:with_decode(
[routing_key, properties, payload, payload_encoding], ReqData0, Context,
fun ([RoutingKey, Props0, Payload0, Enc], _, ReqData) when is_binary(Payload0) ->
rabbit_mgmt_util:with_channel(
VHost, ReqData, Context,
fun (Ch) ->
MRef = erlang:monitor(process, Ch),
amqp_channel:register_confirm_handler(Ch, self()),
amqp_channel:register_return_handler(Ch, self()),
amqp_channel:call(Ch, #'confirm.select'{}),
Props = rabbit_mgmt_format:to_basic_properties(Props0),
Payload = decode(Payload0, Enc),
amqp_channel:cast(Ch, #'basic.publish'{
exchange = X,
routing_key = RoutingKey,
mandatory = true},
#amqp_msg{props = Props,
payload = Payload}),
receive
{#'basic.return'{}, _} ->
receive
#'basic.ack'{} -> ok
end,
good(MRef, false, ReqData, Context);
#'basic.ack'{} ->
good(MRef, true, ReqData, Context);
#'basic.nack'{} ->
erlang:demonitor(MRef),
bad(rejected, ReqData, Context);
{'DOWN', _, _, _, Err} ->
bad(Err, ReqData, Context)
end
end);
([_RoutingKey, _Props, _Payload, _Enc], _, _ReqData) ->
throw({error, payload_not_string})
end).
good(MRef, Routed, ReqData, Context) ->
erlang:demonitor(MRef),
rabbit_mgmt_util:reply([{routed, Routed}], ReqData, Context).
bad({shutdown, {connection_closing,
{server_initiated_close, Code, Reason}}}, ReqData, Context) ->
rabbit_mgmt_util:bad_request_exception(Code, Reason, ReqData, Context);
bad({shutdown, {server_initiated_close, Code, Reason}}, ReqData, Context) ->
rabbit_mgmt_util:bad_request_exception(Code, Reason, ReqData, Context);
bad(rejected, ReqData, Context) ->
Msg = "Unable to publish message. Check queue limits.",
rabbit_mgmt_util:bad_request_exception(rejected, Msg, ReqData, Context).
is_authorized(ReqData, Context) ->
rabbit_mgmt_util:is_authorized_vhost(ReqData, Context).
%%--------------------------------------------------------------------
decode(Payload, <<"string">>) -> Payload;
decode(Payload, <<"base64">>) -> rabbit_mgmt_util:b64decode_or_throw(Payload);
decode(_Payload, Enc) -> throw({error, {unsupported_encoding, Enc}}).
|
