class _agentApp(Thread): def __init__(self, name, broker_url, heartbeat): Thread.__init__(self) self.notifier = _testNotifier() self.broker_url = broker_url self.agent = Agent(name, _notifier=self.notifier, heartbeat_interval=heartbeat) # Dynamically construct a management database _schema = SchemaObjectClass(_classId=SchemaClassId( "MyPackage", "MyClass"), _desc="A test data schema", _object_id_names=["index1", "index2"]) # add properties _schema.add_property("index1", SchemaProperty(qmfTypes.TYPE_UINT8)) _schema.add_property("index2", SchemaProperty(qmfTypes.TYPE_LSTR)) # these two properties are statistics _schema.add_property("query_count", SchemaProperty(qmfTypes.TYPE_UINT32)) _schema.add_property("method_call_count", SchemaProperty(qmfTypes.TYPE_UINT32)) # These two properties can be set via the method call _schema.add_property("set_string", SchemaProperty(qmfTypes.TYPE_LSTR)) _schema.add_property("set_int", SchemaProperty(qmfTypes.TYPE_UINT32)) # add method _meth = SchemaMethod(_desc="Method to set string and int in object.") _meth.add_argument("arg_int", SchemaProperty(qmfTypes.TYPE_UINT32)) _meth.add_argument("arg_str", SchemaProperty(qmfTypes.TYPE_LSTR)) _schema.add_method("set_meth", _meth) # Add schema to Agent self.agent.register_object_class(_schema) # instantiate managed data objects matching the schema _obj1 = QmfAgentData(self.agent, _schema=_schema, _values={ "index1": 100, "index2": "a name" }) _obj1.set_value("set_string", "UNSET") _obj1.set_value("set_int", 0) _obj1.set_value("query_count", 0) _obj1.set_value("method_call_count", 0) self.agent.add_object(_obj1) self.agent.add_object( QmfAgentData(self.agent, _schema=_schema, _values={ "index1": 99, "index2": "another name", "set_string": "UNSET", "set_int": 0, "query_count": 0, "method_call_count": 0 })) # add an "unstructured" object to the Agent _obj2 = QmfAgentData(self.agent, _object_id="01545") _obj2.set_value("field1", "a value") _obj2.set_value("field2", 2) _obj2.set_value("field3", {"a": 1, "map": 2, "value": 3}) _obj2.set_value("field4", ["a", "list", "value"]) self.agent.add_object(_obj2) self.running = False self.ready = Event() def start_app(self): self.running = True self.start() self.ready.wait(10) if not self.ready.is_set(): raise Exception("Agent failed to connect to broker.") def stop_app(self): self.running = False # wake main thread self.notifier.indication() # hmmm... collide with daemon??? self.join(10) if self.isAlive(): raise Exception("AGENT DID NOT TERMINATE AS EXPECTED!!!") def run(self): # broker_url = "user/passwd@hostname:port" self.conn = qpid.messaging.Connection(self.broker_url) self.conn.open() self.agent.set_connection(self.conn) self.ready.set() # Agent application main processing loop while self.running: self.notifier.wait_for_work(None) wi = self.agent.get_next_workitem(timeout=0) while wi is not None: if wi.get_type() == WorkItem.METHOD_CALL: mc = wi.get_params() if not isinstance(mc, MethodCallParams): raise Exception("Unexpected method call parameters") if mc.get_name() == "set_meth": obj = self.agent.get_object(mc.get_object_id(), mc.get_schema_id()) if obj is None: error_info = QmfData.create( { "code": -2, "description": "Bad Object Id." }, _object_id="_error") self.agent.method_response(wi.get_handle(), _error=error_info) else: obj.inc_value("method_call_count") if "arg_int" in mc.get_args(): obj.set_value("set_int", mc.get_args()["arg_int"]) if "arg_str" in mc.get_args(): obj.set_value("set_string", mc.get_args()["arg_str"]) self.agent.method_response(wi.get_handle(), {"code": 0}) elif mc.get_name() == "a_method": obj = self.agent.get_object(mc.get_object_id(), mc.get_schema_id()) if obj is None: error_info = QmfData.create( { "code": -3, "description": "Unknown object id." }, _object_id="_error") self.agent.method_response(wi.get_handle(), _error=error_info) elif obj.get_object_id() != "01545": error_info = QmfData.create( { "code": -4, "description": "Unexpected id." }, _object_id="_error") self.agent.method_response(wi.get_handle(), _error=error_info) else: args = mc.get_args() if ("arg1" in args and args["arg1"] == 1 and "arg2" in args and args["arg2"] == "Now set!" and "arg3" in args and args["arg3"] == 1966): self.agent.method_response( wi.get_handle(), {"code": 0}) else: error_info = QmfData.create( { "code": -5, "description": "Bad Args." }, _object_id="_error") self.agent.method_response(wi.get_handle(), _error=error_info) else: error_info = QmfData.create( { "code": -1, "description": "Unknown method call." }, _object_id="_error") self.agent.method_response(wi.get_handle(), _error=error_info) self.agent.release_workitem(wi) wi = self.agent.get_next_workitem(timeout=0) if self.conn: self.agent.remove_connection(10) self.agent.destroy(10)
while not _done: # try: _notifier.waitForWork() _wi = _agent.get_next_workitem(timeout=0) while _wi: if _wi.get_type() == WorkItem.METHOD_CALL: mc = _wi.get_params() if mc.get_name() == "set_meth": print("!!! Calling 'set_meth' on Object_id = %s" % mc.get_object_id()) print("!!! args='%s'" % str(mc.get_args())) print("!!! userid=%s" % str(mc.get_user_id())) print("!!! handle=%s" % _wi.get_handle()) _agent.method_response(_wi.get_handle(), {"rc1": 100, "rc2": "Success"}) else: print("!!! Unknown Method name = %s" % mc.get_name()) _agent.method_response(_wi.get_handle(), _error=_error_data) else: print("TBD: work item %d:%s" % (_wi.get_type(), str(_wi.get_params()))) _agent.release_workitem(_wi) _wi = _agent.get_next_workitem(timeout=0) # except: # print( "shutting down...") # _done = True print( "Removing connection... TBD!!!" ) #_myConsole.remove_connection( _c, 10 )
class _agentApp(Thread): def __init__(self, name, broker_url, heartbeat): Thread.__init__(self) self.notifier = _testNotifier() self.broker_url = broker_url self.agent = Agent(name, _notifier=self.notifier, heartbeat_interval=heartbeat) # Dynamically construct a management database _schema = SchemaObjectClass( _classId=SchemaClassId("MyPackage", "MyClass"), _desc="A test data schema", _object_id_names=["index1", "index2"] ) # add properties _schema.add_property( "index1", SchemaProperty(qmfTypes.TYPE_UINT8)) _schema.add_property( "index2", SchemaProperty(qmfTypes.TYPE_LSTR)) # these two properties are statistics _schema.add_property( "query_count", SchemaProperty(qmfTypes.TYPE_UINT32)) _schema.add_property( "method_call_count", SchemaProperty(qmfTypes.TYPE_UINT32)) # These two properties can be set via the method call _schema.add_property( "set_string", SchemaProperty(qmfTypes.TYPE_LSTR)) _schema.add_property( "set_int", SchemaProperty(qmfTypes.TYPE_UINT32)) # add method _meth = SchemaMethod( _desc="Method to set string and int in object." ) _meth.add_argument( "arg_int", SchemaProperty(qmfTypes.TYPE_UINT32) ) _meth.add_argument( "arg_str", SchemaProperty(qmfTypes.TYPE_LSTR) ) # the input value of cookie is returned in the response _meth.add_argument( "cookie", SchemaProperty(qmfTypes.TYPE_LSTR, kwargs={"dir":"IO"})) _schema.add_method( "set_meth", _meth ) # Add schema to Agent self.agent.register_object_class(_schema) # instantiate managed data objects matching the schema _obj1 = QmfAgentData( self.agent, _schema=_schema, _values={"index1":100, "index2":"a name"}) _obj1.set_value("set_string", "UNSET") _obj1.set_value("set_int", 0) _obj1.set_value("query_count", 0) _obj1.set_value("method_call_count", 0) self.agent.add_object( _obj1 ) self.agent.add_object( QmfAgentData( self.agent, _schema=_schema, _values={"index1":99, "index2": "another name", "set_string": "UNSET", "set_int": 0, "query_count": 0, "method_call_count": 0} )) # add an "unstructured" object to the Agent _obj2 = QmfAgentData(self.agent, _object_id="01545") _obj2.set_value("field1", "a value") _obj2.set_value("field2", 2) _obj2.set_value("field3", {"a":1, "map":2, "value":3}) _obj2.set_value("field4", ["a", "list", "value"]) self.agent.add_object(_obj2) self.running = False self.ready = Event() def start_app(self): self.running = True self.start() self.ready.wait(10) if not self.ready.is_set(): raise Exception("Agent failed to connect to broker.") def stop_app(self): self.running = False # wake main thread self.notifier.indication() # hmmm... collide with daemon??? self.join(10) if self.isAlive(): raise Exception("AGENT DID NOT TERMINATE AS EXPECTED!!!") def run(self): # broker_url = "user/passwd@hostname:port" self.conn = qpid.messaging.Connection(self.broker_url) self.conn.open() self.agent.set_connection(self.conn) self.ready.set() # Agent application main processing loop while self.running: self.notifier.wait_for_work(None) wi = self.agent.get_next_workitem(timeout=0) while wi is not None: if wi.get_type() == WorkItem.METHOD_CALL: mc = wi.get_params() if not isinstance(mc, MethodCallParams): raise Exception("Unexpected method call parameters") if mc.get_name() == "set_meth": obj = self.agent.get_object(mc.get_object_id(), mc.get_schema_id()) if obj is None: error_info = QmfData.create({"code": -2, "description": "Bad Object Id."}, _object_id="_error") self.agent.method_response(wi.get_handle(), _error=error_info) else: obj.inc_value("method_call_count") out_args = {"code" : 0} if "cookie" in mc.get_args(): out_args["cookie"] = mc.get_args()["cookie"] if "arg_int" in mc.get_args(): obj.set_value("set_int", mc.get_args()["arg_int"]) if "arg_str" in mc.get_args(): obj.set_value("set_string", mc.get_args()["arg_str"]) self.agent.method_response(wi.get_handle(), out_args) elif mc.get_name() == "a_method": obj = self.agent.get_object(mc.get_object_id(), mc.get_schema_id()) if obj is None: error_info = QmfData.create({"code": -3, "description": "Unknown object id."}, _object_id="_error") self.agent.method_response(wi.get_handle(), _error=error_info) elif obj.get_object_id() != "01545": error_info = QmfData.create( {"code": -4, "description": "Unexpected id."}, _object_id="_error") self.agent.method_response(wi.get_handle(), _error=error_info) else: args = mc.get_args() if ("arg1" in args and args["arg1"] == 1 and "arg2" in args and args["arg2"] == "Now set!" and "arg3" in args and args["arg3"] == 1966): out_args = {"code" : 0} if "cookie" in mc.get_args(): out_args["cookie"] = mc.get_args()["cookie"] self.agent.method_response(wi.get_handle(), out_args) else: error_info = QmfData.create( {"code": -5, "description": "Bad Args."}, _object_id="_error") self.agent.method_response(wi.get_handle(), _error=error_info) else: error_info = QmfData.create( {"code": -1, "description": "Unknown method call."}, _object_id="_error") self.agent.method_response(wi.get_handle(), _error=error_info) self.agent.release_workitem(wi) wi = self.agent.get_next_workitem(timeout=0) if self.conn: self.agent.remove_connection(10) self.agent.destroy(10)
_notifier.waitForWork() _wi = _agent.get_next_workitem(timeout=0) while _wi: if _wi.get_type() == WorkItem.METHOD_CALL: mc = _wi.get_params() if mc.get_name() == "set_meth": print("!!! Calling 'set_meth' on Object_id = %s" % mc.get_object_id()) print("!!! args='%s'" % str(mc.get_args())) print("!!! userid=%s" % str(mc.get_user_id())) print("!!! handle=%s" % _wi.get_handle()) _agent.method_response(_wi.get_handle(), { "rc1": 100, "rc2": "Success" }) else: print("!!! Unknown Method name = %s" % mc.get_name()) _agent.method_response(_wi.get_handle(), _error=_error_data) else: print("TBD: work item %d:%s" % (_wi.get_type(), str(_wi.get_params()))) _agent.release_workitem(_wi) _wi = _agent.get_next_workitem(timeout=0) # except: # print( "shutting down...") # _done = True