package com.medi.hk.conf;
import com.alibaba.fastjson.JSON;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessageProperties;
import org.springframework.amqp.support.converter.AbstractMessageConverter;
import org.springframework.amqp.support.converter.MessageConversionException;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.context.annotation.Bean;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Component;
@Component
@Slf4j
public class RabbitMqConfig {
    private final String mqConvertClassName = "className";
    /**
     * mq消息轉(zhuǎn)換
     * @return
     */
    @Bean
    public MessageConverter messageConverter() {
        return new AbstractMessageConverter() {
            /**
             * 接收消息時(shí)轉(zhuǎn)換
             * @param message
             * @return
             * @throws MessageConversionException
             */
            @Override
            public Object fromMessage(Message message) throws MessageConversionException {
                byte[] body = message.getBody();
                MessageProperties messageProperties = message.getMessageProperties();
                //需要轉(zhuǎn)為對(duì)應(yīng)類(lèi)的類(lèi)名
                String className = (String) messageProperties.getHeaders().get(mqConvertClassName);
                try {
                    Boolean aBoolean = (Boolean) messageProperties.getHeaders().get(isArray);
                    if (aBoolean != null && aBoolean) {
                        return JSONArray.parseArray(new String(body), Class.forName(className));
                    } else {
                        return JSON.parseObject(body, Class.forName(className));
                    }
                } catch (ClassNotFoundException e) {
                    log.error("mq轉(zhuǎn)換錯(cuò)誤: message ==> {} body ==> {}", JSON.toJSONString(message), new String(body), e);
                    throw new RuntimeException(e);
                }
            }
            /**
             * 發(fā)送消息時(shí)轉(zhuǎn)換
             * @param object
             * @param messageProperties
             * @return
             */
            @Override
            protected Message createMessage(Object object, MessageProperties messageProperties) {
                messageProperties.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE);
                //標(biāo)記時(shí)什么類(lèi)轉(zhuǎn)換過(guò)來(lái)的
                messageProperties.setHeader(mqConvertClassName, object.getClass().getName());
                //如果是數(shù)組，則設(shè)置數(shù)組成員的類(lèi)型（取第一個(gè)，只允許類(lèi)型相同的數(shù)組）
                if (object instanceof Collection) {
                    String className;
                    Collection coll = (Collection) object;
                    if (coll.size() > 0) {
                        Object next = coll.iterator().next();
                        className = next.getClass().getName();
                    } else {
                        className = Object.class.getName();
                    }
                    messageProperties.setHeader(mqConvertClassName, className);
                    messageProperties.setHeader(isArray, true);
                } else {
                    messageProperties.setHeader(mqConvertClassName, object.getClass().getName());
                }
                return new Message(JSON.toJSONBytes(object), messageProperties);
            }
        };
    }
}

ExchangeConfig.java
package com.space.rbq.order.config;
import org.springframework.amqp.core.DirectExchange;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
/**
 * 消息交換機(jī)配置  可以配置多個(gè)
 * @author zhuzhe
 * @date 2018/5/25 15:40
 * @email 1529949535@qq.com
 */
@Configuration
public class ExchangeConfig {
    /** 消息交換機(jī)1的名字*/
    public final String EXCHANGE_01 = "first_exchange";
    /**
     *   1.定義direct exchange，綁定first_exchange
     *   2.durable="true" 持久化交換機(jī)， rabbitmq重啟的時(shí)候不需要?jiǎng)?chuàng)建新的交換機(jī)
     *   3.direct交換器相對(duì)來(lái)說(shuō)比較簡(jiǎn)單，匹配規(guī)則為：如果路由鍵匹配，消息就被投送到相關(guān)的隊(duì)列
     *     fanout交換器中沒(méi)有路由鍵的概念，他會(huì)把消息發(fā)送到所有綁定在此交換器上面的隊(duì)列中。
     *     topic交換器你采用模糊匹配路由鍵的原則進(jìn)行轉(zhuǎn)發(fā)消息到隊(duì)列中
     */
    @Bean
    public DirectExchange directExchange(){
        DirectExchange directExchange = new DirectExchange(EXCHANGE_01,true,false);
        return directExchange;
    }
}
QueueConfig
package com.space.rbq.order.config;
import org.springframework.amqp.core.Queue;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import java.util.HashMap;
import java.util.Map;
/**
 * 隊(duì)列配置  可以配置多個(gè)隊(duì)列
 * @author zhuzhe
 * @date 2018/5/25 13:25
 * @email 1529949535@qq.com
 */
@Configuration.java
public class QueueConfig {
    /*對(duì)列名稱(chēng)*/
    public static final String QUEUE_NAME1 = "first-queue";
    public static final String QUEUE_NAME2 = "second-queue";
    public static final String QUEUE_NAME3 = "third-queue";
    @Bean
    public Queue firstQueue() {
        /**
         durable="true" 持久化消息隊(duì)列 ， rabbitmq重啟的時(shí)候不需要?jiǎng)?chuàng)建新的隊(duì)列
         auto-delete 表示消息隊(duì)列沒(méi)有在使用時(shí)將被自動(dòng)刪除 默認(rèn)是false
         exclusive  表示該消息隊(duì)列是否只在當(dāng)前connection生效,默認(rèn)是false
         */
        return new Queue(QUEUE_NAME1,true,false,false);
    }
    @Bean
    public Queue secondQueue() {
        return new Queue(QUEUE_NAME2,true,false,false);
    }
    @Bean
    public Queue thirdQueue() {
        // 配置 自動(dòng)刪除
        Map<String, Object> arguments = new HashMap<>();
        arguments.put("x-message-ttl", 60000);//60秒自動(dòng)刪除
        return new Queue(QUEUE_NAME3,true,false,true,arguments);
    }
}
RabbitMqConfig.java
package com.space.rbq.order.config;
import com.space.rbq.order.mqcallback.MsgSendConfirmCallBack;
import com.space.rbq.order.mqcallback.MsgSendReturnCallback;
import org.springframework.amqp.core.Binding;
import org.springframework.amqp.core.BindingBuilder;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import javax.annotation.PostConstruct;
/**
 * RabbitMq配置
 * @author zhuzhe
 * @date 2018/5/25 13:37
 * @email 1529949535@qq.com
 */
@Configuration
public class RabbitMqConfig {
    /**
     * key: queue在該direct-exchange中的key值，當(dāng)消息發(fā)送給direct-exchange中指定key為設(shè)置值時(shí)，
     * 消息將會(huì)轉(zhuǎn)發(fā)給queue參數(shù)指定的消息隊(duì)列
     */
    /** 隊(duì)列key1*/
    public static final String ROUTING_KEY_1 = "queue_one_key1";
    public static final String ROUTING_KEY_2 = "queue_one_key2";
    @Autowired
    private QueueConfig queueConfig;
    @Autowired
    private ExchangeConfig exchangeConfig;
    @Autowired
    private ConnectionFactory connectionFactory;
//    @Autowired
//    RabbitTemplate rabbitTemplate;
//    @PostConstruct
//    public void init() {
//        rabbitTemplate.setConfirmCallback(msgSendConfirmCallBack());
//        rabbitTemplate.setReturnCallback(msgSendReturnCallback());
//    }
    /**
     * 將消息隊(duì)列1和交換機(jī)1進(jìn)行綁定,指定隊(duì)列key1
     */
    @Bean
    public Binding binding_one() {
        return BindingBuilder.bind(queueConfig.firstQueue()).to(exchangeConfig.directExchange()).with(RabbitMqConfig.ROUTING_KEY_1);
    }
    /**
     * 將消息隊(duì)列2和交換機(jī)1進(jìn)行綁定,指定隊(duì)列key2
     */
    @Bean
    public Binding binding_two() {
        return BindingBuilder.bind(queueConfig.secondQueue()).to(exchangeConfig.directExchange()).with(RabbitMqConfig.ROUTING_KEY_2);
    }
    /**
     * 定義rabbit template用于數(shù)據(jù)的接收和發(fā)送
     * 可以設(shè)置消息確認(rèn)機(jī)制和回調(diào)
     * @return
     */
    @Bean
    public RabbitTemplate rabbitTemplate() {
        RabbitTemplate template = new RabbitTemplate(connectionFactory);
        // template.setMessageConverter(); 可以自定義消息轉(zhuǎn)換器  默認(rèn)使用的JDK的，所以消息對(duì)象需要實(shí)現(xiàn)Serializable
        // template.setMessageConverter(new Jackson2JsonMessageConverter());
        /**若使用confirm-callback或return-callback，
         * 必須要配置publisherConfirms或publisherReturns為true
         * 每個(gè)rabbitTemplate只能有一個(gè)confirm-callback和return-callback
         */
        template.setConfirmCallback(msgSendConfirmCallBack());
        /**
         * 使用return-callback時(shí)必須設(shè)置mandatory為true，或者在配置中設(shè)置mandatory-expression的值為true，
         * 可針對(duì)每次請(qǐng)求的消息去確定'mandatory'的boolean值，
         * 只能在提供'return -callback'時(shí)使用，與mandatory互斥
         */
        template.setReturnCallback(msgSendReturnCallback());
        template.setMandatory(true);
        return template;
    }
    /**
     * 關(guān)于 msgSendConfirmCallBack 和 msgSendReturnCallback 的回調(diào)說(shuō)明：
     * 1.如果消息沒(méi)有到exchange,則confirm回調(diào),ack=false
     * 2.如果消息到達(dá)exchange,則confirm回調(diào),ack=true
     * 3.exchange到queue成功,則不回調(diào)return
     * 4.exchange到queue失敗,則回調(diào)return(需設(shè)置mandatory=true,否則不回調(diào),消息就丟了)
     */
    /**
     * 消息確認(rèn)機(jī)制
     * Confirms給客戶(hù)端一種輕量級(jí)的方式，能夠跟蹤哪些消息被broker處理，
     * 哪些可能因?yàn)閎roker宕掉或者網(wǎng)絡(luò)失敗的情況而重新發(fā)布。
     * 確認(rèn)并且保證消息被送達(dá)，提供了兩種方式：發(fā)布確認(rèn)和事務(wù)。(兩者不可同時(shí)使用)
     * 在channel為事務(wù)時(shí)，不可引入確認(rèn)模式；同樣channel為確認(rèn)模式下，不可使用事務(wù)。
     * @return
     */
    @Bean
    public MsgSendConfirmCallBack msgSendConfirmCallBack(){
        return new MsgSendConfirmCallBack();
    }
    @Bean
    public MsgSendReturnCallback msgSendReturnCallback(){
        return new MsgSendReturnCallback();
    }
}

 /**
     * queues  指定從哪個(gè)隊(duì)列（queue）訂閱消息
     * @param message
     * @param channel
     */
    @RabbitListener(queues = {QUEUE_NAME1})
    public void handleMessage(Message message,Channel channel) throws IOException {
        try {
            // 處理消息
            System.out.println("OrderConsumer {} handleMessage :"+message);
            // 執(zhí)行減庫(kù)存操作
//            storeService.update(new Gson().fromJson(new String(message.getBody()),Order.class));
           /**
            * 第一個(gè)參數(shù) deliveryTag：就是接受的消息的deliveryTag,可以通過(guò)msg.getMessageProperties().getDeliveryTag()獲得
            * 第二個(gè)參數(shù) multiple：如果為true，確認(rèn)之前接受到的消息；如果為false，只確認(rèn)當(dāng)前消息。
            * 如果為true就表示連續(xù)取得多條消息才發(fā)會(huì)確認(rèn)，和計(jì)算機(jī)網(wǎng)絡(luò)的中tcp協(xié)議接受分組的累積確認(rèn)十分相似，
            * 能夠提高效率。
            *
            * 同樣的，如果要nack或者拒絕消息（reject）的時(shí)候，
            * 也是調(diào)用channel里面的basicXXX方法就可以了（要指定tagId）。
            *
            * 注意：如果拋異常或nack（并且requeue為true），消息會(huì)重新入隊(duì)列，
            * 并且會(huì)造成消費(fèi)者不斷從隊(duì)列中讀取同一條消息的假象。
            */
            // 確認(rèn)消息
            // 如果 channel.basicAck   channel.basicNack  channel.basicReject 這三個(gè)方法都不執(zhí)行，消息也會(huì)被確認(rèn) 【這個(gè)其實(shí)并沒(méi)有在官方看到，不過(guò)自己測(cè)試的確是這樣哈】
            // 所以，正常情況下一般不需要執(zhí)行 channel.basicAck
            // channel.basicAck(message.getMessageProperties().getDeliveryTag(), true);
        }catch (Exception e){
            log.error("OrderConsumer  handleMessage {} , error:",message,e);
            // 處理消息失敗，將消息重新放回隊(duì)列
            channel.basicNack(message.getMessageProperties().getDeliveryTag(), false,true);
        }
    }

RabbitListener(bindings = @QueueBinding(
            value = @Queue(value = QUEUE_NAME1, durable = "true", autoDelete = "false"),
            exchange = @Exchange(value = "first_exchange", type = ExchangeTypes.DIRECT),
            key = "queue_one_key1"))