gRPC integration

docs/ext/grpc/grpc.client_interceptor.rst

@@ -0,0 +1,7 @@
+grpc.client\_interceptor module
+===============================
+
+.. automodule:: opentelemetry.ext.grpc.client_interceptor
+   :members:
+   :undoc-members:
+   :show-inheritance:

docs/ext/grpc/grpc.rst

@@ -0,0 +1,17 @@
+.. include:: ../../../ext/opentelemetry-ext-grpc/README.rst
+
+Submodules
+----------
+
+.. toctree::
+
+   grpc.client_interceptor
+   grpc.server_interceptor
+
+Module contents
+---------------
+
+.. automodule:: opentelemetry.ext.grpc
+   :members:
+   :undoc-members:
+   :show-inheritance:

docs/ext/grpc/grpc.server_interceptor.rst

@@ -0,0 +1,7 @@
+grpc.server\_interceptor module
+===============================
+
+.. automodule:: opentelemetry.ext.grpc.server_interceptor
+   :members:
+   :undoc-members:
+   :show-inheritance:

ext/opentelemetry-ext-grpc/README.rst

@@ -0,0 +1,16 @@
+OpenTelemetry gRPC Integration
+==============================
+
+|pypi|
+
+.. |pypi| image:: https://badge.fury.io/py/opentelemetry-ext-grpc.svg
+   :target: https://pypi.org/project/opentelemetry-ext-grpc/
+
+gRPC support!
+
+Installation
+------------
+
+::
+
+     pip install opentelemetry-ext-grpc

ext/opentelemetry-ext-grpc/examples/codegen.py

@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+# Copyright 2020, OpenTelemetry 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.
+
+from grpc_tools import protoc
+
+
+def main():
+    return protoc.main(
+        ["-I.", "--python_out=.", "--grpc_python_out=.", "helloworld.proto"]
+    )
+
+
+if __name__ == "__main__":
+    main()

ext/opentelemetry-ext-grpc/examples/hello_world_client.py

@@ -0,0 +1,46 @@
+#!/usr/bin/env python
+# Copyright 2020, OpenTelemetry 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.
+
+# https://github.com/grpc/grpc/blob/master/examples/python/helloworld/greeter_client.py
+# https://github.com/grpc/grpc/blob/v1.16.x/examples/python/interceptors/default_value/greeter_client.py
+"""The Python implementation of the GRPC helloworld.Greeter client."""
+
+import logging
+
+import grpc
+
+import helloworld_pb2
+import helloworld_pb2_grpc
+from opentelemetry.ext.grpc import client_interceptor
+
+interceptor = client_interceptor.OpenTelemetryClientInterceptor()
+
+
+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:
+
+        intercepted_channel = grpc.intercept_channel(channel, interceptor)
+        stub = helloworld_pb2_grpc.GreeterStub(intercepted_channel)
+
+        response = stub.SayHello(helloworld_pb2.HelloRequest(name="YOU"))
+    print("Greeter client received: " + response.message)
+
+
+if __name__ == "__main__":
+    logging.basicConfig()
+    run()

ext/opentelemetry-ext-grpc/examples/hello_world_server.py

@@ -0,0 +1,61 @@
+#!/usr/bin/env python
+# Copyright 2020, OpenTelemetry 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.
+
+# https://github.com/grpc/grpc/blob/master/examples/python/helloworld/greeter_server.py
+"""The Python implementation of the GRPC helloworld.Greeter server."""
+
+import logging
+from concurrent import futures
+
+import grpc
+
+import helloworld_pb2
+import helloworld_pb2_grpc
+from opentelemetry import trace
+from opentelemetry.ext.grpc import server_interceptor
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import (
+    ConsoleSpanExporter,
+    SimpleExportSpanProcessor,
+)
+
+trace.set_preferred_tracer_provider_implementation(lambda T: TracerProvider())
+trace.tracer_provider().add_span_processor(
+    SimpleExportSpanProcessor(ConsoleSpanExporter())
+)
+tracer = trace.get_tracer(__name__)
+
+
+class Greeter(helloworld_pb2_grpc.GreeterServicer):
+    def SayHello(self, request, context):
+        return helloworld_pb2.HelloReply(message="Hello, %s!" % request.name)
+
+
+def serve():
+
+    interceptor = server_interceptor.OpenTelemetryServerInterceptor(tracer)
+    server = grpc.server(
+        futures.ThreadPoolExecutor(max_workers=10), interceptors=(interceptor,)
+    )
+
+    helloworld_pb2_grpc.add_GreeterServicer_to_server(Greeter(), server)
+    server.add_insecure_port("[::]:50051")
+    server.start()
+    server.wait_for_termination()
+
+
+if __name__ == "__main__":
+    logging.basicConfig()
+    serve()

ext/opentelemetry-ext-grpc/examples/helloworld.proto

@@ -0,0 +1,35 @@
+// Copyright 2020, OpenTelemetry 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.
+
+// https://github.com/grpc/grpc/blob/master/examples/protos/helloworld.proto
+
+syntax = "proto3";
+
+package helloworld;
+
+// The greeting service definition.
+service Greeter {
+  // Sends a greeting
+  rpc SayHello (HelloRequest) returns (HelloReply) {}
+}
+
+// The request message containing the user's name.
+message HelloRequest {
+  string name = 1;
+}
+
+// The response message containing the greetings
+message HelloReply {
+  string message = 1;
+}