048fb36ac3
Storage Management Agent is a gRPC server that provides an abstraction layer above the SPDK RPC interface. The interface aims to provide a set of methods for managing various protocols (e.g. NVMe, virtio-blk) while hiding the details of a particular transport. The external API is defined by `lib/python/spdk/sma/proto/sma.proto` protobuf file. It defines the generic gRPC service methods and their requests/responses. Device-specific messages are defined in their own files. This patch also defines messages for creating NVMe and NVMe/TCP devices. This patch implements a gRPC service that delegates the work to a specific device type. A DeviceManager is a class that implements some of the methods defined by the service for a given type of devices (e.g. NVMe, virtio-blk, NVMe/TCP, etc.). For now, only the RPC for creating a device is implemented, others are added in subsequent patches. The series implements the generic calls as well as their NVMe/TCP implementation. Support for other devce types could be easily added by creating a new device manager and defining its protobuf parameter definition. Signed-off-by: Konrad Sztyber <konrad.sztyber@intel.com> Change-Id: I17cde3b31d3514878f1027cfcd112b48848f6123 Reviewed-on: https://review.spdk.io/gerrit/c/spdk/spdk/+/10273 Community-CI: Broadcom CI <spdk-ci.pdl@broadcom.com> Community-CI: Mellanox Build Bot Tested-by: SPDK CI Jenkins <sys_sgci@intel.com> Reviewed-by: Ben Walker <benjamin.walker@intel.com> Reviewed-by: Jim Harris <james.r.harris@intel.com>
50 lines
1.7 KiB
Python
Executable File
50 lines
1.7 KiB
Python
Executable File
#!/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:])
|