add missing proto stubs; refactor

Author: parthea

.github/.OwlBot.yaml

@@ -21,12 +21,18 @@ deep-remove-regex:
 deep-copy-regex:
   - source: /google/spanner/(v.*)/.*-py/(.*)
     dest: /owl-bot-staging/spanner/$1/$2
+  - source: /google/spanner/(v.*)/.*-pb2-py/(commit_response.*|keys.*|mutation.*|query_plan.*|result_set.*|spanner.*|transaction.*|type.*)
+    dest: /owl-bot-staging/spanner_pb2/$1/$2
   - source: /google/spanner/admin/instance/(v.*)/.*-py/(.*)
     dest: /owl-bot-staging/spanner_admin_instance/$1/$2
+  - source: /google/spanner/admin/instance/(v.*)/.*-pb2-py/(common.*|spanner_instance_admin.*)
+    dest: /owl-bot-staging/spanner_admin_instance_pb2/$1/$2
   - source: /google/spanner/admin/database/(v.*)/.*-py/(.*)
     dest: /owl-bot-staging/spanner_admin_database/$1/$2
-  - source: /google/spanner/executor/(v.*)/.*-py/(cloud_executor.*)
-    dest: /owl-bot-staging/spanner_executor/$1/$2
+  - source: /google/spanner/admin/database/(v.*)/.*-pb2-py/(common.*|backup.*|spanner_database_admin.*)
+    dest: /owl-bot-staging/spanner_admin_database_pb2/$1/$2
+  - source: /google/spanner/executor/(v.*)/.*-pb2-py/(cloud_executor.*)
+    dest: /owl-bot-staging/spanner_executor_pb2/$1/$2
 
 begin-after-commit-hash: b154da710c5c9eedee127c07f74b6158c9c22382
 

owlbot.py

@@ -119,9 +119,24 @@ def get_staging_dirs(
     )
 
 for library in get_staging_dirs(default_version="None",
-    sub_directory="spanner_executor"
+    sub_directory="spanner_executor_pb2"
 ):
-    s.move(library, "test_proxy/protos")
+    s.move(library, "test_proxy/protos/google/spanner/executor/v1")
+
+for library in get_staging_dirs(default_version="None",
+    sub_directory="spanner_pb2"
+):
+    s.move(library, "test_proxy/protos/google/spanner/v1")
+
+for library in get_staging_dirs(default_version="None",
+    sub_directory="spanner_admin_database_pb2"
+):
+    s.move(library, "test_proxy/protos/google/spanner/admin/database/v1")
+
+for library in get_staging_dirs(default_version="None",
+    sub_directory="spanner_admin_instance_pb2"
+):
+    s.move(library, "test_proxy/protos/google/spanner/admin/instance/v1")
 
 s.remove_staging_dirs()
 

test_proxy/protos/cloud_executor_server.py

@@ -0,0 +1,43 @@
+# Copyright 2015 gRPC authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""The Python implementation of the GRPC helloworld.Greeter server."""
+
+from concurrent import futures
+import logging
+
+import grpc
+from google.spanner.executor.v1 import cloud_executor_pb2
+from google.spanner.executor.v1 import cloud_executor_pb2_grpc
+
+
+class TestProxyServer(cloud_executor_pb2_grpc.SpannerExecutorProxyServicer):
+    def ExecuteActionAsync(self, request, context):
+        print("Execute action async")
+        print(f"{request=}")
+        # TODO: do something with the request
+        yield cloud_executor_pb2.SpannerAsyncActionResponse()
+
+def serve():
+    port = '50055'
+    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
+    cloud_executor_pb2_grpc.add_SpannerExecutorProxyServicer_to_server(TestProxyServer(), server)
+    server.add_insecure_port('[::]:' + port)
+    server.start()
+    print("Server started, listening on " + port)
+    server.wait_for_termination()
+
+
+if __name__ == '__main__':
+    logging.basicConfig()
+    serve()

test_proxy/protos/google/spanner/admin/database/v1/backup_pb2.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

test_proxy/protos/google/spanner/admin/database/v1/backup_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+