#!/usr/bin/env python | |
import sys | |
import time | |
import zmq | |
import TZmqClient | |
import thrift.protocol.TBinaryProtocol | |
import storage.ttypes | |
import storage.Storage | |
def main(args): | |
endpoint = "tcp://127.0.0.1:9090" | |
socktype = zmq.REQ | |
incr = 0 | |
if len(args) > 1: | |
incr = int(args[1]) | |
if incr: | |
socktype = zmq.DOWNSTREAM | |
endpoint = "tcp://127.0.0.1:9091" | |
ctx = zmq.Context() | |
transport = TZmqClient.TZmqClient(ctx, endpoint, socktype) | |
protocol = thrift.protocol.TBinaryProtocol.TBinaryProtocolAccelerated(transport) | |
client = storage.Storage.Client(protocol) | |
transport.open() | |
if incr: | |
client.incr(incr) | |
time.sleep(0.05) | |
else: | |
value = client.get() | |
print value | |
if __name__ == "__main__": | |
main(sys.argv) |