36 lines
813 B
Python
Executable file
36 lines
813 B
Python
Executable file
#!/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)
|