Update RPC with simpler methods for connecting
This commit is contained in:
@@ -32,8 +32,8 @@ at the given address and does some RPCs')
|
||||
return parser.parse_args()
|
||||
|
||||
|
||||
def main(sock):
|
||||
client = capnp.TwoPartyClient(sock)
|
||||
def main(host):
|
||||
client = capnp.TwoPartyClient(host)
|
||||
|
||||
# Pass "calculator" to ez_restore (there's also a `restore` function that
|
||||
# takes a struct or AnyPointer as an argument), and then cast the returned
|
||||
@@ -282,11 +282,4 @@ def main(sock):
|
||||
print("PASS")
|
||||
|
||||
if __name__ == '__main__':
|
||||
host, port = parse_args().host.split(':')
|
||||
|
||||
sock = socket.create_connection((host, port))
|
||||
|
||||
# Set TCP_NODELAY on socket to disable Nagle's algorithm. This is not
|
||||
# neccessary, but it speeds things up.
|
||||
sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
|
||||
main(sock)
|
||||
main(parse_args().host)
|
||||
|
||||
Reference in New Issue
Block a user