コード例 #1
0
ファイル: qdweb.py プロジェクト: fabric8io/qpid-dispatch
class BusManager:
    def __init__(self):
        pass

    def SetHost(self, host, router):
        self.M = Messenger()
        self.M.start()
        self.M.timeout = 3
        self.M.route("amqp:/*", "amqp://%s/$1" % host)
        if router:
            self.address = "amqp:/_topo/0/%s/$management" % router
        else:
            self.address = "amqp:/_local/$management"
        self.subscription = self.M.subscribe("amqp:/#")
        self.reply = self.subscription.address

    def Disconnect(self):
        self.M.stop()

    def GetObject(self, cls):
        request = Message()
        response = Message()

        request.address = self.address
        request.reply_to = self.reply
        request.correlation_id = 1
        request.properties = {u'operation':u'GET', u'type':cls}
        request.body = {'attributeNames': []}

        self.M.put(request)
        self.M.send()
        self.M.recv()
        self.M.get(response)

        if response.properties['status-code'] != 200:
            raise Exception("Agent reports: %d %s" % (response.properties['status-code'], response.properties['status-description']))

        entities = []
        results = response.body
        for e in results:
            entities.append(AmqpEntity(e))

        return entities
コード例 #2
0
ファイル: adapter.py プロジェクト: ted-ross/ocean-weasel
class Adapter(object):
    def __init__(self):
        self.db  = MongoClient()[DATABASE]
        self.mng = Messenger()
        self.mng.route("amqp:/*", "amqp://%s/$1" % AMQP_HOST)
        self.mng.start()
        self.mng.subscribe(SERVICE)

    def run(self):
        while (True):
            if self.mng.incoming < 1:
                self.mng.recv(1)
            if self.mng.incoming > 0:
                request = Message()
                self.mng.get(request)
                if request.reply_to:
                    response = Message()
                    response.address = request.reply_to
                    response.correlation_id = request.correlation_id
                    response.properties = {}
                    response.properties['result'], response.body = self.process(request.properties, request.body)
                    self.mng.put(response)
                    self.mng.send()

    def process(self, properties, body):
        if 'collection' not in properties:
            return 'Error: collection not specified', None
        if 'command' not in properties:
            return 'Error: command not specified', None
        collection = properties['collection']
        command    = properties['command']
        col = self.db[collection]
        if command == 'find':
            if body == None:
                cursor = col.find()
            elif body.__class__ == dict:
                cursor = col.find(body)
            elif body.__class__ == list:
                cursor = col.find(body[0], body[1])
            answer = []
            for i in range(cursor.count()):
                answer.append(cursor[i])
        return 'OK', answer
    def _do_dump(self, args=None):

        if not (self.issuer and self.key and self.namespace and self.queuename):
            print "Service bus is not configured"
            return
            
        messenger = Messenger()

        address_syntax = "amqps://{issuer}:{key}@{namespace}.servicebus.windows.net/{queuename}"
        address = address_syntax.format(issuer = self.issuer,
                                        key = urllib.quote(self.key, ""),
                                        namespace = self.namespace,
                                        queuename = self.queuename)
        censored_address = address_syntax.format(issuer = self.issuer,
                                                 key = "XXX",
                                                 namespace = self.namespace,
                                                 queuename = self.queuename)

        print "Subscribing to %s" % censored_address
        messenger.subscribe(address)

        # https://github.com/Azure/azure-service-bus-samples/blob/master/proton-c-queues-and-topics/receiver.c
        messenger.incoming_window = 10

        messenger.start()
        while True:
            messenger.recv()
            while messenger.incoming:
                message = Message()
                tracker = messenger.get(message)
                if args == "verbose":
                    print "-" * 32
                    pprint.pprint(message.address)
                    pprint.pprint(message.delivery_count)
                    pprint.pprint(message.annotations)
                    pprint.pprint(message.properties)
                    pprint.pprint(message.body)
                else:
                    print message.delivery_count, message.properties
                messenger.accept(tracker)
        messenger.stop()