50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import grpc
|
||
|
|
import google.protobuf.json_format as json_format
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
sys.path.append(os.path.dirname(__file__) + '/../python')
|
||
|
|
|
||
|
|
import spdk.sma.proto.sma_pb2 as sma_pb2 # noqa
|
||
|
|
import spdk.sma.proto.sma_pb2_grpc as sma_pb2_grpc # noqa
|
||
|
|
import spdk.sma.proto.nvmf_tcp_pb2 as nvmf_tcp_pb2 # noqa
|
||
|
|
import spdk.sma.proto.nvmf_tcp_pb2_grpc as nvmf_tcp_pb2_grpc # noqa
|
||
|
|
|
||
|
|
|
||
|
|
class Client:
|
||
|
|
def __init__(self, addr, port):
|
||
|
|
self._service = sma_pb2.DESCRIPTOR.services_by_name['StorageManagementAgent']
|
||
|
|
self.addr = addr
|
||
|
|
self.port = port
|
||
|
|
|
||
|
|
def _get_message_type(self, descriptor):
|
||
|
|
return getattr(sma_pb2, descriptor.name)
|
||
|
|
|
||
|
|
def _get_method_types(self, method_name):
|
||
|
|
method = self._service.methods_by_name.get(method_name)
|
||
|
|
return (self._get_message_type(method.input_type),
|
||
|
|
self._get_message_type(method.output_type))
|
||
|
|
|
||
|
|
def call(self, method, params):
|
||
|
|
with grpc.insecure_channel(f'{self.addr}:{self.port}') as channel:
|
||
|
|
stub = sma_pb2_grpc.StorageManagementAgentStub(channel)
|
||
|
|
func = getattr(stub, method)
|
||
|
|
input, output = self._get_method_types(method)
|
||
|
|
response = func(request=json_format.ParseDict(params, input()))
|
||
|
|
return json_format.MessageToDict(response,
|
||
|
|
preserving_proto_field_name=True)
|
||
|
|
|
||
|
|
|
||
|
|
def main(args):
|
||
|
|
client = Client('localhost', 8080)
|
||
|
|
request = json.loads(sys.stdin.read())
|
||
|
|
result = client.call(request['method'], request.get('params', {}))
|
||
|
|
print(json.dumps(result, indent=2))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main(sys.argv[1:])
|