Java MessagePostProcessor类代码示例

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
@Override
public <T> Future<T> send(final T payload, final Address destination) throws MessagingException {
    Preconditions.checkNotNull(payload, "payload cannot be null!");
    Preconditions.checkNotNull(destination, "destination cannot be null!");

    try {
        UniqueId messageId = uniqueIdFactory.createUniqueId();
        UniqueId correlationId = uniqueIdFactory.createUniqueId();

        Future<T> responseFuture = responseListener.awaitResponse(correlationId);
        MessagePostProcessor postProcessor = postProcessorFactory.createPostProcessor(messageId, correlationId);

        amqpTemplate.convertAndSend(destination.getExchangeName(), destination.getRoutingKey(), payload, postProcessor);
        return responseFuture;
    } catch (Exception e) {
        logger.warn("exception when sending message", e);
        throw new MessagingException(e);
    }
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
public void sendMessage(final String serviceName, final String serviceMethodName,final String correlationId, Object request) {
	logger.info("sendMessage [this.{}, serviceMethodName:{} serviceName:{} correlationId: {}]", this.getClass(), serviceMethodName, serviceName, correlationId);
	rabbitTemplate.setConfirmCallback(this);
	rabbitTemplate.setCorrelationKey(correlationId);
	rabbitTemplate.convertAndSend(routingkey, request, new MessagePostProcessor() {            
    	@Override
        public Message postProcessMessage(Message message) throws AmqpException {
            message.getMessageProperties().setAppId(appId);
            message.getMessageProperties().setTimestamp(new Date());
            message.getMessageProperties().setMessageId(UUID.randomUUID().toString());
            message.getMessageProperties().setCorrelationId(correlationId.getBytes());
            message.getMessageProperties().setHeader("ServiceMethodName", serviceMethodName);
            message.getMessageProperties().setHeader("ServiceName", serviceName);
            return message;
        }
    }, new CorrelationData(correlationId));
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
@Around(
    "execution(* org.springframework.amqp.core.AmqpTemplate.convertAndSend(Object,org.springframework.amqp.core.MessagePostProcessor))")
public void executeAroundConvertAndSendOneArgWithProcessor(ProceedingJoinPoint call)
    throws Throwable {
  final ArgumentDiscover argumentDiscover = ArgumentDiscover.from(call);
  final MessagePostProcessor argPostProcessor = argumentDiscover.messagePostProcessor;
  final boolean byPass = argPostProcessor instanceof SpanManagerMessagePostProcessor;
  if (byPass) {
    call.proceed(call.getArgs());
  } else {
    executeConvertAndSendWithoutPostProcessor(call, argumentDiscover);
  }
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
@Around(
    "execution(* org.springframework.amqp.core.AmqpTemplate.convertAndSend(String,Object,org.springframework.amqp.core.MessagePostProcessor))")
public void executeAroundConvertAndSendTwoArgsWithPostProcessor(ProceedingJoinPoint call)
    throws Throwable {
  final ArgumentDiscover argumentDiscover = ArgumentDiscover.from(call);
  final MessagePostProcessor argPostProcessor = argumentDiscover.messagePostProcessor;
  final boolean byPass = argPostProcessor instanceof SpanManagerMessagePostProcessor;
  if (byPass) {
    call.proceed(call.getArgs());
  } else {
    executeConvertAndSendWithoutPostProcessor(call, argumentDiscover);
  }
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
@Around(
    "execution(* org.springframework.amqp.core.AmqpTemplate.convertAndSend(String,String,Object,org.springframework.amqp.core.MessagePostProcessor))")
public void executeAroundConvertAndSendThreeArgsWithPostProcessor(ProceedingJoinPoint call)
    throws Throwable {
  final ArgumentDiscover discoverArguments = ArgumentDiscover.from(call);
  final MessagePostProcessor argPostProcessor = discoverArguments.messagePostProcessor;
  final boolean byPass = argPostProcessor instanceof SpanManagerMessagePostProcessor;
  if (byPass) {
    call.proceed(call.getArgs());
  } else {
    executeConvertAndSendWithoutPostProcessor(call, discoverArguments);
  }
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
private void executeConvertAndSendWithoutPostProcessor(
    ProceedingJoinPoint call, ArgumentDiscover argumentDiscover) throws Throwable {
  final Object[] args = argumentDiscover.args;
  final MessagePostProcessor argPostProcessor = argumentDiscover.messagePostProcessor;

  boolean executed;

  try {
    if (argPostProcessor != null) {
      final String exchange = argumentDiscover.exchangeOfSender;
      final String routingKey = argumentDiscover.routingKey;
      final SpanManagerMessagePostProcessor postProcessor =
          new SpanManagerMessagePostProcessor(spanManager, buildSpanName(exchange, routingKey));
      final MessagePostProcessor overwritePostProcessor =
          new CompositeMessagePostProcessor(argPostProcessor, postProcessor);

      args[args.length - 1] = overwritePostProcessor;
      call.proceed(args);
      executed = true;
    } else {
      executed = changeExecutionOfMethodToUsePostProcessor(call, argumentDiscover);
    }
  } catch (Exception e) {
    spanManager.afterSend(e);
    throw e;
  }
  if (executed) {
    spanManager.afterSend(null);
  } else {
    executeWithoutPostProcessor(call, argumentDiscover);
  }
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
@Override
public Message postProcessMessage(Message message) throws AmqpException {
  Message localMessage = message;
  for (MessagePostProcessor postProcessor : postProcessorList) {
    localMessage = postProcessor.postProcessMessage(localMessage);
  }
  return localMessage;
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
@Override
public void convertAndSend(Object message, MessagePostProcessor messagePostProcessor)
    throws AmqpException {
  messagePostProcessor.postProcessMessage(convertMessageIfNecessary(message));
  mockManager.throwExceptionIfConfigured();
  logger.debug("convertAndSend: {} {}", message, messagePostProcessor);
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
@Override
public Object convertSendAndReceive(Object message, MessagePostProcessor messagePostProcessor)
    throws AmqpException {
  mockManager.throwExceptionIfConfigured();
  logger.debug("convertSendAndReceive: {} {}", message, messagePostProcessor);
  return message;
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
@Test
public void test() throws Exception {
	Advice[] adviceChain = TestUtils.getPropertyValue(this.container, "adviceChain", Advice[].class);
	assertEquals(1, adviceChain.length);
	RetryTemplate retryTemplate = TestUtils.getPropertyValue(adviceChain[0], "retryOperations",
			RetryTemplate.class);
	assertEquals(5, TestUtils.getPropertyValue(retryTemplate, "retryPolicy.maxAttempts"));
	assertEquals(123L, TestUtils.getPropertyValue(retryTemplate, "backOffPolicy.initialInterval"));
	assertEquals(345L, TestUtils.getPropertyValue(retryTemplate, "backOffPolicy.maxInterval"));
	assertEquals(1.5, TestUtils.getPropertyValue(retryTemplate, "backOffPolicy.multiplier"));
	assertEquals("scsapp-testq", this.container.getQueueNames()[0]);
	assertFalse(TestUtils.getPropertyValue(this.container, "defaultRequeueRejected", Boolean.class));
	assertEquals(2, TestUtils.getPropertyValue(this.container, "concurrentConsumers"));
	assertEquals(3, TestUtils.getPropertyValue(this.container, "maxConcurrentConsumers"));
	assertEquals(AcknowledgeMode.NONE, TestUtils.getPropertyValue(this.container, "acknowledgeMode"));
	assertEquals(10, TestUtils.getPropertyValue(this.container, "prefetchCount"));
	assertEquals(5, TestUtils.getPropertyValue(this.container, "txSize"));

	this.rabbitTemplate.convertAndSend("", "scsapp-testq", "foo", new MessagePostProcessor() {

		@Override
		public org.springframework.amqp.core.Message postProcessMessage(
				org.springframework.amqp.core.Message message) throws AmqpException {
			message.getMessageProperties().getHeaders().put("bar", "baz");
			return message;
		}

	});
	Message<?> out = this.messageCollector.forChannel(this.channels.output()).poll(10, TimeUnit.SECONDS);
	assertNotNull(out);
	assertEquals("foo", out.getPayload());
	assertEquals("baz", out.getHeaders().get("bar"));
	assertNull(out.getHeaders().get(AmqpHeaders.DELIVERY_MODE));
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
@Test
public void test() throws Exception {
	Advice[] adviceChain = TestUtils.getPropertyValue(this.container, "adviceChain", Advice[].class);
	assertEquals(1, adviceChain.length);
	RetryTemplate retryTemplate = TestUtils.getPropertyValue(adviceChain[0], "retryOperations",
			RetryTemplate.class);
	assertEquals(5, TestUtils.getPropertyValue(retryTemplate, "retryPolicy.maxAttempts"));
	assertEquals(123L, TestUtils.getPropertyValue(retryTemplate, "backOffPolicy.initialInterval"));
	assertEquals(345L, TestUtils.getPropertyValue(retryTemplate, "backOffPolicy.maxInterval"));
	assertEquals(1.5, TestUtils.getPropertyValue(retryTemplate, "backOffPolicy.multiplier"));
	assertEquals("scsapp-testq", this.container.getQueueNames()[0]);
	assertFalse(TestUtils.getPropertyValue(this.container, "defaultRequeueRejected", Boolean.class));
	assertEquals(2, TestUtils.getPropertyValue(this.container, "concurrentConsumers"));
	assertEquals(3, TestUtils.getPropertyValue(this.container, "maxConcurrentConsumers"));
	assertEquals(AcknowledgeMode.NONE, TestUtils.getPropertyValue(this.container, "acknowledgeMode"));
	assertEquals(10, TestUtils.getPropertyValue(this.container, "prefetchCount"));
	assertEquals(5, TestUtils.getPropertyValue(this.container, "txSize"));

	this.rabbitTemplate.convertAndSend("", "scsapp-testq", "foo", new MessagePostProcessor() {

		@Override
		public org.springframework.amqp.core.Message postProcessMessage(
				org.springframework.amqp.core.Message message) throws AmqpException {
			message.getMessageProperties().getHeaders().put("bar", "baz");
			return message;
		}

	});
	Message<?> out = this.messageCollector.forChannel(this.channels.output()).poll(10,  TimeUnit.SECONDS);
	assertNotNull(out);
	assertEquals("foo", out.getPayload());
	assertEquals("baz", out.getHeaders().get("bar"));
	assertNull(out.getHeaders().get(AmqpHeaders.DELIVERY_MODE));
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
public void startCalification(SubmissionJudge submit, final int priority) {
    SubmissionJudgeToSubmissionDTOAdapter submission = new SubmissionJudgeToSubmissionDTOAdapter(submit);
    submitTemplate.convertAndSend(submit.getLang(), submission, new MessagePostProcessor() {
        @Override
        public Message postProcessMessage(Message message) throws AmqpException {
            message.getMessageProperties().setPriority(priority);
            return message;
        }
    });
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
@RequestMapping(value = "/process", method = RequestMethod.POST)
ResponseEntity<Map<?, ?>> process(OAuthAuthenticationToken t, @RequestBody String query) {
    logger.debug("the principal is " + t.getName());

    SecurityContext securityContext = ForceSecurityContextHolder.get();

    String accessToken = securityContext.getSessionId();
    String endpoint = securityContext.getEndPointHost();
    String uuid = UUID.randomUUID().toString() + System.currentTimeMillis() + "";

    final Map<String, String> stringStringMap = new HashMap<>();
    stringStringMap.put("batchId", uuid);
    stringStringMap.put("accessToken", accessToken);
    stringStringMap.put("apiEndpoint", endpoint);
    stringStringMap.put("query", query);

    String batchId = (String) this.rabbitTemplate.convertSendAndReceive(
            (Object) query, new MessagePostProcessor() {
        @Override
        public Message postProcessMessage(Message message) throws AmqpException {
            for (String h : stringStringMap.keySet()) {
                message.getMessageProperties().setHeader(h, stringStringMap.get(h));
            }
            return message;
        }
    });
    // the client will know to update the results page once this REST endpoint returns.
    // todo could we refactor this so that client is notified of the updated state by a websocket? \
    // todo could we maybe refactor this to use async servlets?

    log("received batchId: " + batchId);

    return new ResponseEntity<Map<?, ?>>(stringStringMap, HttpStatus.OK);
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
private ArgumentDiscover(ProceedingJoinPoint call) {
  args = call.getArgs();
  target = call.getTarget();

  String exchangeFromArgs = null;
  final String methodName = call.getSignature().getName();
  final int argsLength = args.length;
  if (CONVERT_AND_SEND.equals(methodName)) {
    if (args[argsLength - 1] instanceof MessagePostProcessor) {
      if (argsLength > 3) {
        exchangeFromArgs = (String) args[0];
        routingKey = (String) args[1];
      } else if (argsLength > 2) {
        routingKey = (String) args[0];
      }
    } else {
      if (argsLength > 2) {
        exchangeFromArgs = (String) args[0];
        routingKey = (String) args[1];
      } else if (argsLength > 1) {
        routingKey = (String) args[0];
      }
    }
  } else {
    if (argsLength > 1) {
      if (args[0] instanceof String && args[1] instanceof String) {
        exchangeFromArgs = (String) args[0];
        routingKey = (String) args[1];
      } else {
        routingKey = (String) args[0];
      }
    }
  }

  if (exchangeFromArgs == null) {
    exchangeOfSender = getExchangeFromRabbitTemplate(target);
  } else {
    exchangeOfSender = exchangeFromArgs;
  }

  for (Object arg : args) {
    if (arg instanceof MessagePostProcessor) {
      messagePostProcessor = (MessagePostProcessor) arg;
    } else if (arg instanceof Message) {
      message = (Message) arg;
    }
  }
}
 

import org.springframework.amqp.core.MessagePostProcessor; //导入依赖的package包/类
public CompositeMessagePostProcessor(MessagePostProcessor... postProcessors) {
  postProcessorList = new ArrayList<>();
  for (MessagePostProcessor processor : postProcessors) {
    postProcessorList.add(processor);
  }
}