grpc_interceptors 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e5788b629b92902268c009af0d015ce195ea82ef6841f2adc75be44a80841acf
-  data.tar.gz: 175875a825b39b6bbf21256c0a97310ffa83feff87a9c9e9ff6ad4a6ebbc83f5
+  metadata.gz: 2d7842d003c6ac5d3ed32a99b351e1c4637b2a9f2cc492ae4e53d7035c36e3e5
+  data.tar.gz: 67486ab61fb8f18703281e86233ec99fd501567aeb0887475e5cf5e07573e49a
 SHA512:
-  metadata.gz: b1b68eeb8f1d1bd06e99153c160caea6353043fc14e2c3c61268246f7fe32bd87755ff0d509d4748c85128f04349e1ca61550fcef3f640b645672afe81c7bd84
-  data.tar.gz: ca9653d07430adaecbe7ca6ed07e629703275435a0d0f067240ff93a13c3b5a61e80b704305f60365db70b3977f5d7b8f37bafd2529f8942d43c9899ae3e9a65
+  metadata.gz: dc309aaa264a867dc0e20bc6f657de562fd15e75f9ca1da2a9f05f955b64a62e0432b695b57c21584094ce8ad8ee9b2fd63a69d3327674f64670b811553706d3
+  data.tar.gz: b797809c323717c89c4ef73f61151748e0b2d0a26697684328bae47d9d263a9c5ea9f9d98b9aa9a32cdf381d7bed4af55f8e2960730b7cddfe1f3b4fd503d8b4

data/README.md

@@ -29,7 +29,11 @@ _WIP_
 
 ### Logging
 
-When the `LOG_LEVEL` env variable is set to `INFO` then the server logs out
+When the `LOG_LEVEL` env variable is set to `INFO` then the server logs out.
+
+When the `LOG_LEVEL` env variable is set to `DEBUG` then the server additionally adds the request to the log message. (Note, adding the response is currently blocked by [this gRPC issue](https://github.com/grpc/grpc/pull/26547).)
+
+
 
 ### StatsD metrics
 

data/lib/grpc_interceptors/client/logging.rb

@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+require_relative '../common/grpc_helper'
+require_relative '../common/log_payload'
+
+module GrpcInterceptors
+  module Client
+    class Logging < ::GRPC::ServerInterceptor
+      def initialize(logger)
+        @logger = logger
+
+        super()
+      end
+
+      def request_response(
+        request: nil, call: nil, method: nil, metadata: nil, &block
+      )
+        log(request, method, 'unary', &block)
+      end
+
+      # def client_streamer(_requests: nil, call: nil, method: nil, metadata: nil)
+      #  yield
+      # end
+
+      # def server_streamer(_request: nil, call: nil, method: nil, metadata: nil)
+      #  yield
+      # end
+
+      # def bidi_streamer(_requests: nil, call: nil, method: nil, metadata: nil)
+      #  yield
+      # end
+
+      private
+
+      # if the server responds with error, then the error is attached to the log
+      # if the current level is INFO, then it logs out basic facts
+      # if the current level is DEBUG, then it additionally includes the request
+      def log(request, method, method_type)
+        grpc_code = ::GRPC::Core::StatusCodes::OK
+
+        response = yield
+      rescue StandardError => e
+        grpc_code = e.is_a?(::GRPC::BadStatus) ? e.code : ::GRPC::Core::StatusCodes::UNKNOWN
+
+        raise
+      ensure
+        payload = Common::LogPayload.build(
+          method, method_type, grpc_code, 'client'
+        )
+
+        if e
+          payload['error'] = e.class.to_s
+          payload['error_message'] = e.message
+          payload['backtrace'] = e.backtrace
+        end
+
+        if @logger.level == Logger::Severity::INFO
+          @logger.info(payload)
+        elsif @logger.level == Logger::Severity::DEBUG
+          payload['request'] = Common::GrpcHelper.proto_to_json(request)
+          payload['response'] = Common::GrpcHelper.proto_to_json(response)
+          @logger.debug(payload)
+        end
+      end
+    end
+  end
+end

data/lib/grpc_interceptors/client/{open_telemetry_tracing_instrument.rb → opentelemetry_tracing_instrument.rb}

@@ -1,15 +1,18 @@
 # frozen_string_literal: true
 
 require_relative '../common/grpc_helper'
+require_relative '../common/opentelemetry_helper'
 
 module GrpcInterceptors
   module Client
     class OpenTelemetryTracingInstrument < ::GRPC::ClientInterceptor
       def request_response(request: nil, call: nil, method: nil, metadata: nil)
         kind = OpenTelemetry::Trace::SpanKind::CLIENT
-        attributes = tracing_attributes(method)
+        attributes = Common::OpenTelemetryHelper.tracing_attributes(method)
 
-        GrpcHelper.tracer.in_span(method, kind: kind, attributes: attributes) do
+        Common::OpenTelemetryHelper.tracer.in_span(
+          method, kind: kind, attributes: attributes
+        ) do
           OpenTelemetry.propagation.inject(metadata)
           yield
         end
@@ -26,20 +29,6 @@ module GrpcInterceptors
       # def bidi_streamer(_requests: nil, call: nil, method: nil, metadata: nil)
       # yield
       # end
-
-      private
-
-      def tracing_attributes(method)
-        method_parts = method.to_s.sub(%r{^/}, '').split('/')
-        service_name = method_parts.shift
-        method_name = method_parts.join('/')
-
-        {
-          OpenTelemetry::SemanticConventions::Trace::RPC_SYSTEM => 'grpc',
-          OpenTelemetry::SemanticConventions::Trace::RPC_SERVICE => service_name,
-          OpenTelemetry::SemanticConventions::Trace::RPC_METHOD => method_name,
-        }
-      end
     end
   end
 end

data/lib/grpc_interceptors/common/grpc_helper.rb

@@ -1,13 +1,54 @@
 # frozen_string_literal: true
 
 module GrpcInterceptors
-  module GrpcHelper
-    def self.route_name(method)
-      "/#{method.owner.service_name}/#{method.original_name}"
-    end
+  module Common
+    module GrpcHelper
+      def self.route_name(method)
+        return method unless method.is_a?(Method) # client case
+
+        # server case
+        service_name = service_name_from_server(method)
+        method_name = method_name_from_server(method)
+
+        "/#{service_name}/#{method_name}"
+      end
+
+      def self.service_name(method)
+        return service_name_from_server(method) if method.is_a?(Method) # server case
+
+        service_name_from_client(method) # client case
+      end
+
+      def self.method_name(method)
+        return method_name_from_server(method) if method.is_a?(Method) # server case
+
+        method_name_from_client(method) # client case
+      end
+
+      def self.service_name_from_server(method)
+        method.owner.service_name.to_s
+      end
+
+      def self.method_name_from_server(method)
+        method.original_name.to_s
+      end
+
+      def self.service_name_from_client(method)
+        method_parts = method.to_s.sub(%r{^/}, '').split('/')
+        method_parts.first
+      end
+
+      def self.method_name_from_client(method)
+        method_parts = method.to_s.sub(%r{^/}, '').split('/')
+        method_parts[1..].join('/')
+      end
 
-    def self.tracer
-      OpenTelemetry.tracer_provider.tracer('grpc')
+      def self.proto_to_json(proto)
+        proto.to_json(
+          emit_defaults: true,
+          preserve_proto_fieldnames: true
+        )
+      end
     end
   end
 end

data/lib/grpc_interceptors/common/log_payload.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module GrpcInterceptors
+  module Common
+    module LogPayload
+      def self.build(method, method_type, grpc_code, kind)
+        service = GrpcHelper.service_name(method)
+        method = GrpcHelper.method_name(method)
+
+        payload = {
+          'pid' => Process.pid,
+          'grpc.component' => kind, # the caller, server or client
+          'grpc.service' => service,
+          'grpc.method' => method,
+          'grpc.method_type' => method_type,
+          'grpc.code' => grpc_code
+        }
+
+        if defined?(OpenTelemetry) && OpenTelemetry::Trace.current_span.recording?
+          tracing_context = OpenTelemetry::Trace.current_span.context
+          payload['span_id'] = tracing_context.hex_span_id
+          payload['trace_id'] = tracing_context.hex_trace_id
+        end
+
+        payload
+      end
+    end
+  end
+end

data/lib/grpc_interceptors/common/opentelemetry_helper.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module GrpcInterceptors
+  module Common
+    module OpenTelemetryHelper
+      def self.tracer
+        OpenTelemetry.tracer_provider.tracer('grpc')
+      end
+
+      def self.tracing_attributes(method)
+        service_name = Common::GrpcHelper.service_name(method)
+        method_name = Common::GrpcHelper.method_name(method)
+
+        {
+          OpenTelemetry::SemanticConventions::Trace::RPC_SYSTEM => 'grpc',
+          OpenTelemetry::SemanticConventions::Trace::RPC_SERVICE => service_name,
+          OpenTelemetry::SemanticConventions::Trace::RPC_METHOD => method_name
+        }
+      end
+    end
+  end
+end

data/lib/grpc_interceptors/server/logging.rb

@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+require_relative '../common/grpc_helper'
+require_relative '../common/log_payload'
+
+module GrpcInterceptors
+  module Server
+    class Logging < ::GRPC::ServerInterceptor
+      def initialize(logger)
+        @logger = logger
+
+        super()
+      end
+
+      def request_response(request: nil, call: nil, method: nil, &block)
+        log(request, method, 'unary', &block)
+      end
+
+      # def client_streamer(call: nil, method: nil)
+      #  yield
+      # end
+
+      # def server_streamer(_request: nil, call: nil, method: nil)
+      #  yield
+      # end
+
+      # def bidi_streamer(_requests: nil, call: nil, method: nil)
+      #  yield
+      # end
+
+      private
+
+      # in case of an exception, it logs out on the ERROR level including error details
+      # if the current level is INFO, then it logs out basic facts
+      # if the current level is DEBUG, then it additionally includes the request
+      def log(request, method, method_type)
+        grpc_code = ::GRPC::Core::StatusCodes::OK
+
+        yield
+      rescue StandardError => e
+        grpc_code = e.is_a?(::GRPC::BadStatus) ? e.code : ::GRPC::Core::StatusCodes::UNKNOWN
+
+        raise
+      ensure
+        payload = Common::LogPayload.build(
+          method, method_type, grpc_code, 'server'
+        )
+
+        if e
+          payload['error'] = e.class.to_s
+          payload['error_message'] = e.message
+          payload['backtrace'] = e.backtrace
+          @logger.error(payload)
+        elsif @logger.level == Logger::Severity::INFO
+          @logger.info(payload)
+        elsif @logger.level == Logger::Severity::DEBUG
+          payload['request'] = Common::GrpcHelper.proto_to_json(request)
+          @logger.debug(payload)
+        end
+      end
+    end
+  end
+end

data/lib/grpc_interceptors/server/opentelemetry_tracing_instrument.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require_relative '../common/grpc_helper'
+require_relative '../common/opentelemetry_helper'
+
+module GrpcInterceptors
+  module Server
+    # https://github.com/grpc/grpc/blob/master/src/ruby/lib/grpc/generic/interceptors.rb
+    class OpenTelemetryTracingInstrument < ::GRPC::ServerInterceptor
+      def request_response(request: nil, call: nil, method: nil, &block)
+        context = OpenTelemetry.propagation.extract(call.metadata)
+        route_name = Common::GrpcHelper.route_name(method)
+        attributes = Common::OpenTelemetryHelper.tracing_attributes(method)
+        kind = OpenTelemetry::Trace::SpanKind::SERVER
+
+        OpenTelemetry::Context.with_current(context) do
+          Common::OpenTelemetryHelper.tracer.in_span(
+            route_name,
+            attributes: attributes,
+            kind: kind,
+            &block
+          )
+        end
+      end
+
+      # def client_streamer(call: nil, method: nil)
+      #  yield
+      # end
+
+      # def server_streamer(_request: nil, call: nil, method: nil)
+      #  yield
+      # end
+
+      # def bidi_streamer(_requests: nil, call: nil, method: nil)
+      #  yield
+      # end
+    end
+  end
+end

data/lib/grpc_interceptors/server/statsd_metrics.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative '../common/grpc_helper'
+
 module GrpcInterceptors
   module Server
     class StatsDMetrics < ::GRPC::ServerInterceptor
@@ -32,8 +34,8 @@ module GrpcInterceptors
 
       def common_labels(method)
         {
-          grpc_method: method.original_name.to_s,
-          grpc_service: method.owner.service_name
+          grpc_method: Common::GrpcHelper.method_name(method),
+          grpc_service: Common::GrpcHelper.service_name(method)
         }
       end
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: grpc_interceptors
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - michal-kazmierczak
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-03-01 00:00:00.000000000 Z
+date: 2024-03-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: opentelemetry-api
@@ -272,10 +272,13 @@ files:
 - LICENSE
 - README.md
 - lib/grpc_interceptors.rb
-- lib/grpc_interceptors/client/open_telemetry_tracing_instrument.rb
+- lib/grpc_interceptors/client/logging.rb
+- lib/grpc_interceptors/client/opentelemetry_tracing_instrument.rb
 - lib/grpc_interceptors/common/grpc_helper.rb
-- lib/grpc_interceptors/common/logger.rb
-- lib/grpc_interceptors/server/open_telemetry_tracing_instrument.rb
+- lib/grpc_interceptors/common/log_payload.rb
+- lib/grpc_interceptors/common/opentelemetry_helper.rb
+- lib/grpc_interceptors/server/logging.rb
+- lib/grpc_interceptors/server/opentelemetry_tracing_instrument.rb
 - lib/grpc_interceptors/server/statsd_metrics.rb
 homepage: https://github.com/michal-kazmierczak/ruby-grpc-interceptors
 licenses:

data/lib/grpc_interceptors/common/logger.rb

@@ -1,11 +0,0 @@
-# frozen_string_literal: true
-
-module GrpcInterceptors
-  module Common
-    class Logger
-      # extend self
-
-      def log; end
-    end
-  end
-end

data/lib/grpc_interceptors/server/open_telemetry_tracing_instrument.rb

@@ -1,55 +0,0 @@
-# frozen_string_literal: true
-
-require_relative '../common/grpc_helper'
-
-module GrpcInterceptors
-  module Server
-    # https://github.com/grpc/grpc/blob/master/src/ruby/lib/grpc/generic/interceptors.rb
-    class OpenTelemetryTracingInstrument < ::GRPC::ServerInterceptor
-      def request_response(request: nil, call: nil, method: nil)
-        parent_context = OpenTelemetry.propagation.extract(call.metadata)
-        route_name = GrpcHelper.route_name(method)
-        attributes = tracing_attributes(method)
-        kind = OpenTelemetry::Trace::SpanKind::SERVER
-        span = GrpcHelper.tracer.start_span(
-          route_name,
-          with_parent: parent_context,
-          attributes: attributes,
-          kind: kind
-        )
-
-        yield
-
-        span.finish
-      rescue StandardError => e
-        OpenTelemetry.handle_error(exception: e)
-
-        raise e
-      ensure
-        span.finish if span.recording?
-      end
-
-      # def client_streamer(call: nil, method: nil)
-      #  yield
-      # end
-
-      # def server_streamer(_request: nil, call: nil, method: nil)
-      #  yield
-      # end
-
-      # def bidi_streamer(_requests: nil, call: nil, method: nil)
-      #  yield
-      # end
-
-      private
-
-      def tracing_attributes(method)
-        {
-          OpenTelemetry::SemanticConventions::Trace::RPC_SYSTEM => 'grpc',
-          OpenTelemetry::SemanticConventions::Trace::RPC_SERVICE => method.owner.service_name,
-          OpenTelemetry::SemanticConventions::Trace::RPC_METHOD => method.original_name.to_s,
-        }
-      end
-    end
-  end
-end