import com.aliyun.openservices.lmq.example.util.ConnectionOptionWrapper;
import com.aliyun.openservices.ons.api.Message;
import com.aliyun.openservices.ons.api.ONSFactory;
import com.aliyun.openservices.ons.api.Producer;
import com.aliyun.openservices.ons.api.PropertyKeyConst;
import com.aliyun.openservices.ons.api.SendResult;
import java.util.Properties;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import org.eclipse.paho.client.mqttv3.IMqttDeliveryToken;
import org.eclipse.paho.client.mqttv3.MqttCallbackExtended;
import org.eclipse.paho.client.mqttv3.MqttClient;
import org.eclipse.paho.client.mqttv3.MqttException;
import org.eclipse.paho.client.mqttv3.MqttMessage;
import org.eclipse.paho.client.mqttv3.persist.MemoryPersistence;

public class RocketMQSendMessageToMQ4IoT {
    public static void main(String[] args) throws Exception {
        /**
         * 初始化云消息隊列 RocketMQ 版發(fā)送客戶端，實際業(yè)務中一般部署在服務端應用中。
         */
        Properties properties = new Properties();
        /**
         * 設置云消息隊列 RocketMQ 版Group ID，在云消息隊列 RocketMQ 版控制臺創(chuàng)建。
         */
        properties.setProperty(PropertyKeyConst.GROUP_ID, "GID-XXXXX");
        /**
         * AccessKey ID，阿里云身份驗證，在阿里云RAM控制臺創(chuàng)建。
         * 阿里云賬號AccessKey擁有所有API的訪問權限，建議您使用RAM用戶進行API訪問或日常運維。
         * 強烈建議不要把AccessKey ID和AccessKey Secret保存到工程代碼里，否則可能導致AccessKey泄露，威脅您賬號下所有資源的安全。
         * 本示例以將AccessKey 和 AccessKeySecret 保存在環(huán)境變量為例說明。
         */
        properties.put(PropertyKeyConst.AccessKey, System.getenv("MQTT_AK_ENV"));
        /**
         * AccessKey Secret，阿里云身份驗證，在阿里云RAM控制臺創(chuàng)建。僅在簽名鑒權模式下需要設置。
         */
        properties.put(PropertyKeyConst.SecretKey, System.getenv("MQTT_SK_ENV"));
        /**
         * 設置TCP接入點，該接入點為云消息隊列 RocketMQ 版實例的接入點。進入云消息隊列 RocketMQ 版控制臺實例詳情頁面獲取。
         */
        properties.put(PropertyKeyConst.NAMESRV_ADDR, "XXXX");
        /**
         * 設置云消息隊列 RocketMQ 版的Topic，在云消息隊列 RocketMQ 版控制臺創(chuàng)建。
         * 云消息隊列 RocketMQ 版和云消息隊列 MQTT 版配合使用時，RocketMQ客戶端僅操作一級Topic。
         */
        final String parentTopic = "XXXXX";
        Producer producer = ONSFactory.createProducer(properties);
        producer.start();
        //////////////////////////////////////////////////////////////////////////////////////////////////////////////////
        /**
         * 初始化云消息隊列 MQTT 版接收客戶端，實際業(yè)務中云消息隊列 MQTT 版一般部署在移動終端環(huán)境。
         */

        /**
         * 您在控制臺創(chuàng)建的云消息隊列 MQTT 版的實例ID。
         */
        String instanceId = "XXXXX";
        /**
         * 設置接入點，進入云消息隊列 MQTT 版控制臺實例詳情頁面獲取。
         */
        String endPoint = "XXXXXX.mqtt.aliyuncs.com";
        /**
         * AccessKey ID，阿里云身份驗證，在阿里云RAM控制臺創(chuàng)建。
         * 阿里云賬號AccessKey擁有所有API的訪問權限，建議您使用RAM用戶進行API訪問或日常運維。
         * 強烈建議不要把AccessKey ID和AccessKey Secret保存到工程代碼里，否則可能導致AccessKey泄露，威脅您賬號下所有資源的安全。
         * 本示例以將AccessKey 和 AccessKeySecret 保存在環(huán)境變量為例說明。
         */
        String accessKey = System.getenv("MQTT_AK_ENV");
        /**
         * AccessKey Secret，阿里云身份驗證，在阿里云RAM控制臺創(chuàng)建。僅在簽名鑒權模式下需要設置。
         */
        String secretKey = System.getenv("MQTT_SK_ENV");
        /**
         * MQTT客戶端ID，由業(yè)務系統(tǒng)分配，需要保證每個TCP連接都不一樣，保證全局唯一，如果不同的客戶端對象（TCP連接）使用了相同的clientId會導致連接異常斷開。
         * clientId由兩部分組成，格式為GroupID@@@DeviceID，其中GroupID在云消息隊列 MQTT 版控制臺創(chuàng)建，DeviceID由業(yè)務方自己設置，clientId總長度不得超過64個字符。
         */
        String clientId = "GID_XXXX@@@XXXXX";
        /**
         * 云消息隊列 MQTT 版支持子級Topic，用來做自定義的過濾，此處為示例，可以填寫任何字符串。
         * 需要注意的是，完整的Topic長度不得超過128個字符。
         */
        final String subTopic = "/testMq4Iot";
        final String mq4IotTopic = parentTopic + subTopic;
        /**
         * QoS參數(shù)代表傳輸質量，可選0，1，2。詳細信息，請參見名詞解釋。
         */
        final int qosLevel = 0;
        ConnectionOptionWrapper connectionOptionWrapper = new ConnectionOptionWrapper(instanceId, accessKey, secretKey, clientId);
        final MemoryPersistence memoryPersistence = new MemoryPersistence();
        /**
         * 客戶端協(xié)議和端口。客戶端使用的協(xié)議和端口必須匹配，如果是SSL加密則設置ssl://endpoint:8883。
         */
        final MqttClient mqttClient = new MqttClient("tcp://" + endPoint + ":1883", clientId, memoryPersistence);
        /**
         * 設置客戶端發(fā)送超時時間，防止無限阻塞。
         */
        mqttClient.setTimeToWait(5000);
        final ExecutorService executorService = new ThreadPoolExecutor(1, 1, 0, TimeUnit.MILLISECONDS,
            new LinkedBlockingQueue<Runnable>());
        mqttClient.setCallback(new MqttCallbackExtended() {
            @Override
            public void connectComplete(boolean reconnect, String serverURI) {
                /**
                 * 客戶端連接成功后就需要盡快訂閱需要的Topic。
                 */
                System.out.println("connect success");
                executorService.submit(new Runnable() {
                    @Override
                    public void run() {
                        try {
                            final String topicFilter[] = {mq4IotTopic};
                            final int[] qos = {qosLevel};
                            mqttClient.subscribe(topicFilter, qos);
                        } catch (MqttException e) {
                            e.printStackTrace();
                        }
                    }
                });
            }

            @Override
            public void connectionLost(Throwable throwable) {
                throwable.printStackTrace();
            }

            @Override
            public void messageArrived(String s, MqttMessage mqttMessage) throws Exception {
                /**
                 * 消費消息的回調接口，需要確保該接口不拋異常，該接口運行返回即代表消息消費成功。
                 * 消費消息需要保證在規(guī)定時間內完成，如果消費耗時超過服務端約定的超時時間，對于可靠傳輸?shù)哪Ｊ剑斩丝赡軙卦囃扑停瑯I(yè)務需要做好冪等去重處理。
                 */
                System.out.println(
                    "receive msg from topic " + s + " , body is " + new String(mqttMessage.getPayload()));
            }

            @Override
            public void deliveryComplete(IMqttDeliveryToken iMqttDeliveryToken) {
                System.out.println("send msg succeed topic is : " + iMqttDeliveryToken.getTopics()[0]);
            }
        });
        mqttClient.connect(connectionOptionWrapper.getMqttConnectOptions());
        for (int i = 0; i < 10; i++) {
            /**
             * 使用RocketMQ客戶端發(fā)消息給MQTT客戶端時，Topic指定為一級父Topic，Tag指定為MQ2MQTT。
             */
            Message msg = new Message(parentTopic, "MQ2MQTT", "hello mq send mqtt msg".getBytes());
            /**
             * 使用RocketMQ客戶端發(fā)消息給MQTT客戶端時，可以通過MqttSecondTopic屬性設置MQTT的子級Topic屬性。
             */
            msg.putUserProperties(PropertyKeyConst.MqttSecondTopic, subTopic);
            SendResult result = producer.send(msg);
            System.out.println(result);
//            /**
//             * 發(fā)送P2P消息，設置子級Topic。
//             */
//            msg.putUserProperties(PropertyKeyConst.MqttSecondTopic, "/p2p/" + clientId);
//            result = producer.send(msg);
//            System.out.println(result);
        }
        Thread.sleep(Long.MAX_VALUE);

    }

}