57 lines
1.2 KiB
Python
Executable File
57 lines
1.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) 2023, Meta
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
"""Thrift Hello Client Sample
|
|
|
|
Connect to a hello service and demonstrate the
|
|
ping(), echo(), and counter() Thrift RPC methods.
|
|
|
|
Usage:
|
|
./hello_client.py [ip]
|
|
"""
|
|
|
|
import argparse
|
|
import sys
|
|
sys.path.append('gen-py')
|
|
|
|
from thrift.protocol import TBinaryProtocol
|
|
from thrift.transport import TTransport
|
|
from thrift.transport import TSocket
|
|
from hello import Hello
|
|
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(allow_abbrev=False)
|
|
parser.add_argument('--ip', default='192.0.2.1',
|
|
help='IP address of hello server')
|
|
|
|
return parser.parse_args()
|
|
|
|
|
|
def main():
|
|
args = parse_args()
|
|
|
|
transport = TSocket.TSocket(args.ip, 4242)
|
|
transport = TTransport.TBufferedTransport(transport)
|
|
protocol = TBinaryProtocol.TBinaryProtocol(transport)
|
|
client = Hello.Client(protocol)
|
|
|
|
transport.open()
|
|
|
|
client.ping()
|
|
client.echo('Hello, world!')
|
|
|
|
# necessary to mitigate unused variable warning with for i in range(5)
|
|
i = 0
|
|
while i < 5:
|
|
client.counter()
|
|
i = i + 1
|
|
|
|
transport.close()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|