1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
#
from qpid.tests.messaging.implementation import *
class Mgmt:
"""
Simple QMF management utility (qpidtoollibs uses
qpid.messaging.Message rather than swigged version)
"""
def __init__(self, conn):
self.conn = conn
self.sess = self.conn.session()
self.reply_to = "qmf.default.topic/direct.%s;{node:{type:topic}, link:{x-declare:{auto-delete:True,exclusive:True}}}" % \
str(uuid4())
self.reply_rx = self.sess.receiver(self.reply_to)
self.reply_rx.capacity = 10
self.tx = self.sess.sender("qmf.default.direct/broker")
self.next_correlator = 1
def close(self):
self.conn.close()
def list(self, class_name):
props = {'method' : 'request',
'qmf.opcode' : '_query_request',
'x-amqp-0-10.app-id' : 'qmf2'}
correlator = str(self.next_correlator)
self.next_correlator += 1
content = {'_what' : 'OBJECT',
'_schema_id' : {'_class_name' : class_name.lower()}}
message = Message(content, reply_to=self.reply_to, correlation_id=correlator,
properties=props, subject="broker")
self.tx.send(message)
response = self.reply_rx.fetch(10)
if response.properties['qmf.opcode'] != '_query_response':
raise Exception("bad response")
items = []
done = False
while not done:
for item in response.content:
items.append(item['_values'])
if 'partial' in response.properties:
response = self.reply_rx.fetch(10)
else:
done = True
self.sess.acknowledge()
return items
def do_qmf_method(self, method, arguments, addr="org.apache.qpid.broker:broker:amqp-broker", timeout=10):
props = {'method' : 'request',
'qmf.opcode' : '_method_request',
'x-amqp-0-10.app-id' : 'qmf2'}
correlator = str(self.next_correlator)
self.next_correlator += 1
content = {'_object_id' : {'_object_name' : addr},
'_method_name' : method,
'_arguments' : arguments}
message = Message(content, reply_to=self.reply_to, correlation_id=correlator,
properties=props, subject="broker")
self.tx.send(message)
response = self.reply_rx.fetch(timeout)
self.sess.acknowledge()
if response.properties['qmf.opcode'] == '_exception':
raise Exception("Exception from Agent: %r" % response.content['_values'])
if response.properties['qmf.opcode'] != '_method_response':
raise Exception("bad response: %r" % response.properties)
return response.content['_arguments']
def create(self, _type, name, properties={}):
return self.do_qmf_method('create', {'type': _type, 'name': name, 'properties': properties})
def delete(self, _type, name):
return self.do_qmf_method('delete', {'type': _type, 'name': name})
def reload_acl_file(self):
self.do_qmf_method('reloadACLFile', {}, "org.apache.qpid.acl:acl:org.apache.qpid.broker:broker:amqp-broker")
|
