class Recv(MessagingHandler):
def __init__(self, url, count):
super(Recv, self).__init__(auto_accept=False)
self.url = url
self.delay = 0
self.last_id = None
self.expected = count
self.received = 0
self.accepted = 0
self.db = Db("dst_db", EventInjector())
def on_start(self, event):
event.container.selectable(self.db.injector)
e = ApplicationEvent("id_loaded")
e.container = event.container
self.db.get_id(e)
def on_id_loaded(self, event):
self.last_id = event.id
event.container.create_receiver(self.url)
def on_record_inserted(self, event):
self.accept(event.delivery)
self.accepted += 1
if self.accepted == self.expected:
event.connection.close()
self.db.close()
def on_message(self, event):
id = int(event.message.id)
if (not self.last_id) or id > self.last_id:
if self.expected == 0 or self.received < self.expected:
self.received += 1
self.last_id = id
self.db.insert(
id, event.message.body,
ApplicationEvent("record_inserted",
delivery=event.delivery))
print("inserted message %s" % id)
else:
self.release(event.delivery)
else:
self.accept(event.delivery)
class Recv(MessagingHandler):
def __init__(self, url, count):
super(Recv, self).__init__(auto_accept=False)
self.url = url
self.delay = 0
self.last_id = None
self.expected = count
self.received = 0
self.accepted = 0
self.db = Db("dst_db", EventInjector())
def on_start(self, event):
event.container.selectable(self.db.injector)
e = ApplicationEvent("id_loaded")
e.container = event.container
self.db.get_id(e)
def on_id_loaded(self, event):
self.last_id = event.id
event.container.create_receiver(self.url)
def on_record_inserted(self, event):
self.accept(event.delivery)
self.accepted += 1
if self.accepted == self.expected:
event.connection.close()
self.db.close()
def on_message(self, event):
id = int(event.message.id)
if (not self.last_id) or id > self.last_id:
if self.received < self.expected:
self.received += 1
self.last_id = id
self.db.insert(id, event.message.body, ApplicationEvent("record_inserted", delivery=event.delivery))
print("inserted message %s" % id)
else:
self.release(event.delivery)
else:
self.accept(event.delivery)
