chore: Add basic samples

Author: yevgenypats

Makefile

@@ -5,4 +5,5 @@ clone-proto:
 gen:
 	cd plugin-pb && git pull && cd ..
 	cp ./plugin-pb/plugin/v3/*.proto ./protos/cloudquery/plugin_v3/.
-	python -m grpc_tools.protoc -I./protos --python_out=. --pyi_out=. --grpc_python_out=. ./protos/cloudquery/plugin_v3/*.proto
+	python -m grpc_tools.protoc -I./protos --python_out=. --pyi_out=. --grpc_python_out=. ./protos/cloudquery/plugin_v3/*.proto
+

README.md

@@ -11,7 +11,7 @@ This is a low-level auto-generate gRPC client and server for CloudQuery plugin f
 we recommend using virtualenv to manage your python environment.
 
 ```bash
-virtualenv -p python3.7 venv
+virtualenv -p python3.7 venv # or any python >= 3.7
 source venv/bin/activate
 ```
 

samples/plugin_client.py

@@ -0,0 +1,16 @@
+import logging
+import grpc
+from cloudquery.plugin_v3 import plugin_pb2, plugin_pb2_grpc
+
+def run():
+    # NOTE(gRPC Python Team): .close() is possible on a channel and should be
+    # used in circumstances in which the with statement does not fit the needs
+    # of the code.
+    with grpc.insecure_channel('localhost:50051') as channel:
+        stub = plugin_pb2_grpc.PluginStub(channel)
+        response = stub.GetName(plugin_pb2.GetName.Request())
+        print(response.name)
+
+if __name__ == '__main__':
+    logging.basicConfig()
+    run()

samples/plugin_server.py

@@ -0,0 +1,48 @@
+import grpc
+from concurrent import futures
+import logging
+from cloudquery.plugin_v3 import plugin_pb2, plugin_pb2_grpc
+
+class PluginServicer(plugin_pb2_grpc.PluginServicer):
+    def __init__(self):
+        pass
+
+    def GetName(self, request, context):
+        return plugin_pb2.GetName.Response(name="plugin test")
+
+    def GetVersion(self, request, context):
+        return plugin_pb2.GetVersion.Response(version="0.0.1")
+
+    def Init(self, request, context):
+        return plugin_pb2.Init.Response()
+
+    def GetTables(self, request, context):
+        return plugin_pb2.GetTables.Response(tables=[])
+
+    def Sync(self, request, context):
+        return plugin_pb2.Sync.Response()
+
+    def Read(self, request, context):
+        return plugin_pb2.Read.Response()
+
+    def Write(self, request_iterator, context):
+        for request in request_iterator:
+            print(request)
+        return plugin_pb2.Write.Response()
+
+    def Close(self, request, context):
+        return plugin_pb2.Close.Response()
+
+def serve():
+    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
+    plugin_pb2_grpc.add_PluginServicer_to_server(
+        PluginServicer(), server)
+    server.add_insecure_port('[::]:50051')
+    print("Starting server. Listening on port 50051")
+    server.start()
+    server.wait_for_termination()
+
+
+if __name__ == '__main__':
+    logging.basicConfig()
+    serve()