RabbitMQ使用

/**
 * RabbitMQ配置，主要是配置队列，如果提前存在该队列，可以省略本配置类
 *
 */
@Slf4j
@Configuration
public class RabbitMqConfiguration {

	@Bean
	public RabbitTemplate rabbitTemplate(CachingConnectionFactory connectionFactory) {
		connectionFactory.setPublisherConfirms(true);
		connectionFactory.setPublisherReturns(true);
		RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
		rabbitTemplate.setMandatory(true);
		rabbitTemplate.setConfirmCallback((correlationData, ack, cause) -> log.info("消息发送成功:correlationData({}),ack({}),cause({})", correlationData, ack, cause));
		rabbitTemplate.setReturnCallback((message, replyCode, replyText, exchange, routingKey) -> log.info("消息丢失:exchange({}),route({}),replyCode({}),replyText({}),message:{}", exchange, routingKey, replyCode, replyText, message));
		return rabbitTemplate;
	}

	/**
	 * 直接模式队列1
	 */
	@Bean
	public Queue directOneQueue() {
		return new Queue(RabbitConstant.DIRECT_MODE_QUEUE_ONE);
	}

	/**
	 * 队列2
	 */
	@Bean
	public Queue queueTwo() {
		return new Queue(RabbitConstant.QUEUE_TWO);
	}

	/**
	 * 队列3
	 */
	@Bean
	public Queue queueThree() {
		return new Queue(RabbitConstant.QUEUE_THREE);
	}
	/**
	 * 队列4
	 */
	@Bean
	public Queue queueFour() {
		return new Queue(RabbitConstant.QUEUE_FOUR);
	}
	/**
	 * 队列5
	 */
	@Bean
	public Queue queueFive() {
		return new Queue(RabbitConstant.QUEUE_FIVE);
	}

	/**
	 * 分列模式队列
	 */
	@Bean
	public FanoutExchange fanoutExchange() {
		return new FanoutExchange(RabbitConstant.FANOUT_MODE_QUEUE);
	}

	/**
	 * 分列模式绑定队列1
	 *
	 * @param directOneQueue 绑定队列1
	 * @param fanoutExchange 分列模式交换器
	 */
	@Bean
	public Binding fanoutBinding1(Queue directOneQueue, FanoutExchange fanoutExchange) {
		return BindingBuilder.bind(directOneQueue).to(fanoutExchange);
	}

	/**
	 * 分列模式绑定队列2
	 *
	 * @param queueTwo       绑定队列2
	 * @param fanoutExchange 分列模式交换器
	 */
	@Bean
	public Binding fanoutBinding2(Queue queueTwo, FanoutExchange fanoutExchange) {
		return BindingBuilder.bind(queueTwo).to(fanoutExchange);
	}

	/**
	 * 主题模式队列
	 * <li>路由格式必须以 . 分隔，比如 user.email 或者 user.aaa.email</li>
	 * <li>通配符 * ，代表一个占位符，或者说一个单词，比如路由为 user.*，那么 user.email 可以匹配，但是 user.aaa.email 就匹配不了</li>
	 * <li>通配符 # ，代表一个或多个占位符，或者说一个或多个单词，比如路由为 user.#，那么 user.email 可以匹配，user.aaa.email 也可以匹配</li>
	 */
	@Bean
	public TopicExchange topicExchange() {
		return new TopicExchange(RabbitConstant.TOPIC_MODE_QUEUE);
	}


	/**
	 * 主题模式绑定分列模式
	 *
	 * @param fanoutExchange 分列模式交换器
	 * @param topicExchange  主题模式交换器
	 */
	@Bean
	public Binding topicBinding1(FanoutExchange fanoutExchange, TopicExchange topicExchange) {
		return BindingBuilder.bind(fanoutExchange).to(topicExchange).with(RabbitConstant.TOPIC_ROUTING_KEY_ONE);
	}

	/**
	 * 主题模式绑定队列2
	 *
	 * @param queueTwo      队列2
	 * @param topicExchange 主题模式交换器
	 */
	@Bean
	public Binding topicBinding2(Queue queueTwo, TopicExchange topicExchange) {
		return BindingBuilder.bind(queueTwo).to(topicExchange).with(RabbitConstant.TOPIC_ROUTING_KEY_TWO);
	}

	/**
	 * 主题模式绑定队列3
	 *
	 * @param queueThree    队列3
	 * @param topicExchange 主题模式交换器
	 */
	@Bean
	public Binding topicBinding3(Queue queueThree, TopicExchange topicExchange) {
		return BindingBuilder.bind(queueThree).to(topicExchange).with(RabbitConstant.TOPIC_ROUTING_KEY_THREE);
	}

	/**
	 * 延迟队列
	 */
	@Bean
	public Queue delayQueue() {
		return new Queue(RabbitConstant.DELAY_QUEUE, true);
	}

	/**
	 * 延迟队列交换器, x-delayed-type 和 x-delayed-message 固定
	 */
	@Bean
	public CustomExchange delayExchange() {
		Map<String, Object> args = Maps.newHashMap();
		args.put("x-delayed-type", "direct");
		return new CustomExchange(RabbitConstant.DELAY_MODE_QUEUE, "x-delayed-message", true, false, args);
	}

	/**
	 * 延迟队列绑定自定义交换器
	 *
	 * @param delayQueue    队列
	 * @param delayExchange 延迟交换器
	 */
	@Bean
	public Binding delayBinding(Queue delayQueue, CustomExchange delayExchange) {
		return BindingBuilder.bind(delayQueue).to(delayExchange).with(RabbitConstant.DELAY_QUEUE).noargs();
	}

}

/**
 * 延迟队列处理器
 *
 */
@Slf4j
@Component
public class DelayQueueHandler {

	@RabbitListener(queues = RabbitConstant.DELAY_QUEUE)
	@RabbitHandler
	public void directHandlerManualAck(MessageStruct messageStruct, Message message, Channel channel) {
		//  如果手动ACK,消息会被监听消费,但是消息在队列中依旧存在,如果 未配置 acknowledge-mode 默认是会在消费完毕后自动ACK掉
		final long deliveryTag = message.getMessageProperties().getDeliveryTag();
		try {
			log.info("延迟队列，手动ACK，接收消息：{}", JsonUtil.toJson(messageStruct));
			// 通知 MQ 消息已被成功消费,可以ACK了
			channel.basicAck(deliveryTag, false);
		} catch (IOException e) {
			try {
				// 处理失败,重新压入MQ
				channel.basicRecover();
			} catch (IOException e1) {
				e1.printStackTrace();
			}
		}
	}
}

/**
 * RabbitMQ常量池
 *
 * @author yangkai.shen
 */
public interface RabbitConstant {
	/**
	 * 直接模式1
	 *
	 */
	String DIRECT_MODE_QUEUE_ONE = "queue.direct.1";

	/**
	 * 分列模式
	 */
	String FANOUT_MODE_QUEUE = "fanout.mode";

	/**
	 * 主题模式
	 */
	String TOPIC_MODE_QUEUE = "topic.mode";

	/**
	 * 路由1
	 */
	String TOPIC_ROUTING_KEY_ONE = "queue.#";

	/**
	 * 路由2
	 */
	String TOPIC_ROUTING_KEY_TWO = "*.queue";

	/**
	 * 路由3
	 */
	String TOPIC_ROUTING_KEY_THREE = "3.queue";

	/**
	 * 延迟队列
	 */
	String DELAY_QUEUE = "delay.queue";

	/**
	 * 延迟队列交换器
	 */
	String DELAY_MODE_QUEUE = "delay.mode";
}