8d445c1c77
* Moving from govendor to dep. * Making the pull request template more friendly. * Fixing akward space in PR template. * goimports run on whole project using ` goimports -w $(find . -type f -name '*.go' -not -path "./vendor/*" -not -path "./gen-go/*")` source of command: https://gist.github.com/bgentry/fd1ffef7dbde01857f66
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)
|