| 25 | |
| 26 | |
| 27 | class Send(MessagingHandler): |
| 28 | def __init__(self, url, messages): |
| 29 | super(Send, self).__init__() |
| 30 | self.url = url |
| 31 | self.sent = 0 |
| 32 | self.confirmed = 0 |
| 33 | self.total = messages |
| 34 | |
| 35 | def on_start(self, event): |
| 36 | event.container.create_sender(self.url) |
| 37 | |
| 38 | def on_sendable(self, event): |
| 39 | while event.sender.credit and self.sent < self.total: |
| 40 | msg = Message(id=(self.sent + 1), body={'sequence': (self.sent + 1)}) |
| 41 | event.sender.send(msg) |
| 42 | self.sent += 1 |
| 43 | |
| 44 | def on_accepted(self, event): |
| 45 | self.confirmed += 1 |
| 46 | if self.confirmed == self.total: |
| 47 | print("all messages confirmed") |
| 48 | event.connection.close() |
| 49 | |
| 50 | def on_disconnected(self, event): |
| 51 | self.sent = self.confirmed |
| 52 | |
| 53 | |
| 54 | parser = optparse.OptionParser(usage="usage: %prog [options]", |