import time
import stomp
import json
from config import *
from engine.common import case_log


def default_call_back(data):
    print("******************Receive New Data******************")
    print data
    print("****************************************************")


class MyListener(stomp.ConnectionListener):
    def __init__(self, callback):
        self.callback = callback

    def on_error(self, headers, message):
        print('received an error %s' % message)

    def on_message(self, headers, message):
        print('received a headers %s' % headers)
        print('received a message %s' % message)
        # key = headers.get('ack', None)
        self.callback(headers, message)


class MessSendOrRecv(object):
    def __init__(self, ip='127.0.0.1', port=61613, user=None, pwd=None, callback=None):
        print("stomp init")
        case_log.info("stomp init.[{0}]:[{1}]".format(ip,port))
        self.conn = stomp.Connection12([(ip, port)])
        call_back = callback if callback else default_call_back
        self.conn.set_listener(name='logicServerQueue',
                               lstnr=MyListener(callback=call_back))
        self.conn.start()
        self.conn.connect(wait=True)
        print("ip: {0}, port: {1} ,connected!".format(ip, port))
        case_log.info("ip: {0}, port: {1} ,connected!".format(ip,port))

    def send_message(self, destination="QLOUD_TEST_ENGINE",
                     message=b"test", headers={'send_id':'123456', 'content_type':'text/plain'}):
        """
        发布信息到事业总线,发送消息到testQueue队列，指定consumerId='88.3@6006
        :param destination: 接收消息的队列名称
        :param message: 消息内容
        :return: 
        """
        # print("start to send message...")
        case_log.info("Start to send message......")
        self.conn.send(
            body=json.dumps(message),
            destination=destination,
            headers=headers,
        )

    def recv_message(self, destination="QLOUD_TEST_ENGINE", headers=None,
                     subscription_id=None):
        """
        从testQueue队列中接收消息，用selector过滤，只接收consumerId = '88.3@6006'的消息
        :param destination: 消息发送到的队列名称
        :param subscription_id: 消息订阅者的ID
        :return: None
        """
        case_log.info("Listen Topic:"+destination)
        self.conn.subscribe(destination=destination,
                             headers=headers,
                            id=subscription_id,
                            ack='auto')

    def ack_message(self, key=None, transaction=None, receipt=None):
        self.conn.ack(id=key, transaction=transaction, receipt=receipt)
        case_log.info("Send ACK Message, keyID:{0}".format(key))

    def disconnect(self):
        time.sleep(1.5)
        self.conn.disconnect()

    @staticmethod
    def run_forever():
        print "run forever"
        while True:
            time.sleep(1)


if __name__ == '__main__':
    stompmess = MessSendOrRecv(ip="127.0.0.1", port=30211)
    headers = {'send_id':123456}
    msg = {"order":"1", ""}
    stompmess.recv_message(destination='TEST_ENGINE', subscription_id=9527, headers=headers)
    stompmess.send_message(destination='TEST_ENGINE', message=msg, headers={'send_id':123, 'content-text':'application/json'})
    while True:
        time.sleep(1)

    stompmess.run_forever()
    # stompmess.disconnect()