如何在RabbitMQ 3.5.6的自定义交换中修改传递(消息)?

时间:2015-12-18 18:01:41

标签: erlang rabbitmq rabbitmq-exchange

我正在为RabbitMQ编写自定义交换,它必须从交换中获取参数并将它们放到消息头中。我对Erlang不太熟悉。

-module(rabbit_exchange_type_arguments_to_headers).


-include_lib("rabbit/include/rabbit.hrl").
-include_lib("rabbit/include/rabbit_framing.hrl").

-behaviour(rabbit_exchange_type).

%% API
-export([
  description/0,
  serialise_events/0,
  route/2,
  validate/1,
  validate_binding/2,
  create/2,
  delete/3,
  policy_changed/2,
  add_binding/3,
  remove_bindings/3,
  assert_args_equivalence/2
]).

-rabbit_boot_step(
  {?MODULE,
    [{description, "exchange type argument-to-header"},
      {mfa,         {rabbit_registry, register, [exchange, <<"argument">>, ?MODULE]}},
      {cleanup, {rabbit_registry, unregister, [exchange, <<"argumente">>]}},
      {requires,    rabbit_registry},
      {enables,     kernel_ready}]
  }
).


description() ->
  [{name, <<"argument">>},
   {description, <<"Adds exchange argumets to message headers">>}].


serialise_events() -> false.

% This is classic fanout routing
route(#exchange{name = Name, arguments = Arguments},
    #delivery{message = #basic_message{content = #content{properties = #'P_basic'{headers = MessageHeaders}}}} = Delivery) ->

  Headers = make_headers(MessageHeaders, Arguments),

  NewDelivery = Delivery#delivery{message = #basic_message{content = #content{properties = #'P_basic'{headers = Headers}}}},

  % Do some thing here to send NewDelivery instead of old one

  rabbit_router:match_routing_key(Name, ['_']).


make_headers(undefined, undefined) -> [];
make_headers(undefined, Arguments) -> Arguments;
make_headers(Headers, undefined) -> Headers;
make_headers(Headers, Arguments) -> lists:append(Headers, Arguments).

validate_binding(_X, _B) -> ok.


validate(_Exchange) -> ok.
create(_Tx, _X) -> ok.
deladd_binding(_Tx, _X, _B) -> ok.
remove_bindings(_Tx, _X, _Bs) -> ok.
assert_args_equivalence(Exchange, Args) ->
  rabbit_exchange:assert_args_equivalence(Exchange, Args).
ete(_Tx, _X, _Bs) -> ok.
policy_changed(_X1, _X2) -> ok.

我在rout方法中进行了新的Delivery,但据我所知,我无法在Erlang中修改原始变量,因此Rabbit继续使用原始变量。

是否有某种方法可以修改交换内部的消息,或者说让兔子继续使用新的消息。

2 个答案:

答案 0 :(得分:0)

您可以修改源代码,一切皆有可能。

答案 1 :(得分:0)

我已经解决了这个问题。 如果有人看到我的存储库https://github.com/thecederick/rabbitmq-arguments-to-headers-exchange。 有源代码和工件可供使用。