| #!/usr/bin/env python |
| import zmq |
| import TZmqServer |
| import storage.ttypes |
| import storage.Storage |
| |
| |
| class StorageHandler(storage.Storage.Iface): |
| def __init__(self): |
| self.value = 0 |
| |
| def incr(self, amount): |
| self.value += amount |
| |
| def get(self): |
| return self.value |
| |
| |
| def main(): |
| handler = StorageHandler() |
| processor = storage.Storage.Processor(handler) |
| |
| ctx = zmq.Context() |
| reqrep_server = TZmqServer.TZmqServer(processor, ctx, "tcp://0.0.0.0:9090", zmq.REP) |
| oneway_server = TZmqServer.TZmqServer(processor, ctx, "tcp://0.0.0.0:9091", zmq.UPSTREAM) |
| multiserver = TZmqServer.TZmqMultiServer() |
| multiserver.servers.append(reqrep_server) |
| multiserver.servers.append(oneway_server) |
| multiserver.serveForever() |
| |
| |
| if __name__ == "__main__": |
| main() |