package org.lkl.mq.rabbitmq.test;

import java.io.IOException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
import com.rabbitmq.client.ConsumerCancelledException;
import com.rabbitmq.client.QueueingConsumer;
import com.rabbitmq.client.QueueingConsumer.Delivery;
import com.rabbitmq.client.ShutdownSignalException;

/**
 * 客戶端01
 * 
 * @author liaokailin
 * @version $Id: Receive01.java, v 0.1 2015年11月01日 下午3:47:58 liaokailin Exp $
 */
public class Receive01 {
    public static void main(String[] args) throws IOException, TimeoutException, ShutdownSignalException,
                                          ConsumerCancelledException, InterruptedException {
        ConnectionFactory facotry = new ConnectionFactory();
        facotry.setUsername("test");
        facotry.setPassword("test");
        facotry.setVirtualHost("test");
        facotry.setHost("localhost");

        Connection conn = facotry.newConnection(); //獲取一個連結
        //通過Channel進行通訊
        Channel channel = conn.createChannel();
        int prefetchCount = 1;
        channel.basicQos(prefetchCount); //保證公平分發

        boolean durable = true;
        //宣告交換機
        channel.exchangeDeclare(Send.EXCHANGE_NAME, "direct", durable); //按照routingKey過濾
        //宣告佇列
        String queueName = channel.queueDeclare("queue-01", true, true, false, null).getQueue();
        //將佇列和交換機繫結
        String routingKey = "lkl-0";
        //佇列可以多次繫結，繫結不同的交換機或者路由key
        channel.queueBind(queueName, Send.EXCHANGE_NAME, routingKey);

        //建立消費者
        QueueingConsumer consumer = new QueueingConsumer(channel);
        
        //將消費者和佇列關聯
        channel.basicConsume(queueName, false, consumer); // 設定為false表面手動確認訊息消費

        //獲取訊息

        System.out.println(" Wait message ....");
        while (true) {
            Delivery delivery = consumer.nextDelivery();
            String msg = new String(delivery.getBody());
            String key = delivery.getEnvelope().getRoutingKey();

            System.out.println("  Received '" + key + "':'" + msg + "'");
            System.out.println(" Handle message");
            TimeUnit.SECONDS.sleep(3); //mock handle message
            channel.basicAck(delivery.getEnvelope().getDeliveryTag(), false); //確定該訊息已成功消費
        }

    }
}