rails_twirp 0.5.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8df8457b09993d7d5c986130e724948df049c201b6797f33a7b66458bde98918
-  data.tar.gz: 0a2100cbefd37754f9dbd6cf666def1c29d0936322f72837d7e60653b5510b79
+  metadata.gz: 34fc8b4e84ba8c456ac2fa01c3c35dd0fd11224360d6e21b832b5d17ed1f57a5
+  data.tar.gz: 9626b771c35bfdd034fa1dccad6ef5ed921aad0abdf3086bfe27a6b3b26f50b9
 SHA512:
-  metadata.gz: 1b50c23cd6fa8852686ca96066c0a65a532f06ec0225b5ed04ac860218760d187e868d0b3d499855527ed23d355dacf8bf33eb1aaf2fd1b590b29aa12c8e836a
-  data.tar.gz: 461fe47bbafdd0bcf304f48b4938f307fd380d7bbce7049a703c60f7bc781b0f58b1cf06b7410c4a168e39441b1a9bd49b5091a0dbfc3472870c002bef6f51e1
+  metadata.gz: e4621381feff50a031faf89079aa49054d246bd52f0e9d0fe4fe588f3014c2dd9a779f249aa16ac15340d9e259b61e273b393dd6a69ecaa15b5bf76c29578434
+  data.tar.gz: e91d714d2ef9c5e6a45e34911a29de4c1927d2e8442c5bf3538c5c43f137c10010b0282d9a5f2d4c5901d9c1733191cd0e7ff4e2764d8ae546233798ef81d64b

data/lib/rails_twirp.rb

@@ -4,6 +4,7 @@ require "rails_twirp/application"
 require "rails_twirp/base"
 require "rails_twirp/route_set"
 require "rails_twirp/testing/integration_test"
+require "rails_twirp/log_subscriber"
 
 module RailsTwirp
   mattr_accessor :test_app

data/lib/rails_twirp/base.rb

@@ -10,6 +10,8 @@ require "abstract_controller/callbacks"
 require "action_controller/metal/helpers"
 require "rails_twirp/rescue"
 require "rails_twirp/url_for"
+require "rails_twirp/implicit_render"
+require "rails_twirp/instrumentation"
 
 module RailsTwirp
   class Base < AbstractController::Base
@@ -24,15 +26,16 @@ module RailsTwirp
     include UrlFor
     include AbstractController::AssetPaths
     include AbstractController::Caching
-    include AbstractController::Logger
 
     include ActionView::Rendering
     include RenderPb
     include Errors
+    include ImplicitRender
 
     # These need to be last so errors can be handled as early as possible.
     include AbstractController::Callbacks
     include Rescue
+    include Instrumentation
 
     attr_internal :request, :env, :response_class
     def initialize
@@ -55,8 +58,6 @@ module RailsTwirp
 
       process(action)
 
-      # Implicit render
-      self.response_body = render unless response_body
       response_body
     end
 

data/lib/rails_twirp/implicit_render.rb

@@ -0,0 +1,11 @@
+require "action_controller/metal/basic_implicit_render"
+
+module RailsTwirp
+  module ImplicitRender
+    include ActionController::BasicImplicitRender
+
+    def default_render
+      render
+    end
+  end
+end

data/lib/rails_twirp/instrumentation.rb

@@ -0,0 +1,32 @@
+module RailsTwirp
+  module Instrumentation
+    extend ActiveSupport::Concern
+
+    include AbstractController::Logger
+
+    def process_action(*)
+      raw_payload = {
+        controller: self.class.name,
+        action: action_name,
+        request: request,
+        http_request: http_request,
+        headers: http_request.headers,
+        path: http_request.fullpath
+      }
+
+      ActiveSupport::Notifications.instrument("start_processing.rails_twirp", raw_payload)
+
+      ActiveSupport::Notifications.instrument("process_action.rails_twirp", raw_payload) do |payload|
+        result = super
+        if response_body.is_a?(Twirp::Error)
+          payload[:code] = response_body.code
+          payload[:msg] = response_body.msg
+        else
+          payload[:code] = :success
+        end
+        payload[:response] = response_body
+        result
+      end
+    end
+  end
+end

data/lib/rails_twirp/log_subscriber.rb

@@ -0,0 +1,64 @@
+require "active_support/log_subscriber"
+
+module RailsTwirp
+  class LogSubscriber < ActiveSupport::LogSubscriber
+    def start_processing(event)
+      return unless logger.info?
+
+      payload = event.payload
+
+      info "Processing by #{payload[:controller]}##{payload[:action]}"
+    end
+
+    def process_action(event)
+      payload = event.payload
+      exception_raised(payload[:http_request], payload[:exception_object]) if payload[:exception_object]
+
+      info do
+        code = payload.fetch(:code, :internal)
+
+        message = +"Completed #{code} in #{event.duration.round}ms (Allocations: #{event.allocations})"
+        message << "\n\n" if defined?(Rails.env) && Rails.env.development?
+
+        message
+      end
+    end
+
+    private
+
+    def exception_raised(request, exception)
+      backtrace_cleaner = request.get_header("action_dispatch.backtrace_cleaner")
+      wrapper = ActionDispatch::ExceptionWrapper.new(backtrace_cleaner, exception)
+
+      log_error(wrapper)
+    end
+
+    def log_error(wrapper)
+      exception = wrapper.exception
+      trace = wrapper.exception_trace
+
+      message = []
+      message << "  "
+      message << "#{exception.class} (#{exception.message}):"
+      message.concat(exception.annotated_source_code) if exception.respond_to?(:annotated_source_code)
+      message << "  "
+      message.concat(trace)
+
+      log_array(message)
+    end
+
+    def log_array(array)
+      lines = Array(array)
+
+      return if lines.empty?
+
+      if logger.formatter&.respond_to?(:tags_text)
+        fatal { lines.join("\n#{logger.formatter.tags_text}") }
+      else
+        fatal { lines.join("\n") }
+      end
+    end
+  end
+end
+
+RailsTwirp::LogSubscriber.attach_to :rails_twirp

data/lib/rails_twirp/testing/integration_test.rb

@@ -1,11 +1,17 @@
+require "twirp/encoding"
+
 module RailsTwirp
   class IntegrationTest < ActiveSupport::TestCase
+    Response = Struct.new(:status, :body, :headers)
+
     attr_reader :response, :request, :controller
+    attr_writer :mount_path
 
     def initialize(name)
       super
       reset!
       @before_rpc = []
+      @mount_path = "/twirp"
     end
 
     def reset!
@@ -19,32 +25,59 @@ module RailsTwirp
 
     def rpc(service, rpc, request, headers: nil)
       @request = request
-      service = app.twirp.routes.services[service].to_service
-
-      rack_env = {
-        "HTTP_HOST" => "localhost"
-      }
-      http_request = ActionDispatch::Request.new(rack_env)
-      http_request.headers.merge! headers if headers.present?
-      env = {rack_env: rack_env}
 
+      env = build_rack_env(service, rpc, request, headers)
       @before_rpc.each do |hook|
         hook.call(env)
       end
 
-      response = begin
-        service.call_rpc rpc, request, env
-      rescue => e
-        Twirp::Error.internal_with(e)
-      end
+      status, headers, body = app.call(env)
+      @response = decode_rack_response(service, rpc, status, headers, body)
+      set_controller_from_rack_env(env)
 
-      @response = response
-      @controller = http_request.controller_instance
-      response
+      @response
     end
 
     def app
       RailsTwirp.test_app
     end
+
+    private
+
+    def build_rack_env(service, rpc, request, headers)
+      env = {
+        "CONTENT_TYPE" => request_content_type,
+        "HTTP_HOST" => "localhost",
+        "PATH_INFO" => "#{@mount_path}/#{service.service_full_name}/#{rpc}",
+        "REQUEST_METHOD" => "POST"
+      }
+      if headers.present?
+        http_request = ActionDispatch::Request.new(env)
+        http_request.headers.merge! headers
+      end
+
+      input_class = service.rpcs[rpc][:input_class]
+      env["rack.input"] = StringIO.new(Twirp::Encoding.encode(request, input_class, request_content_type))
+      env
+    end
+
+    def request_content_type
+      Twirp::Encoding::PROTO
+    end
+
+    def decode_rack_response(service, rpc, status, headers, body)
+      body = body.join # body is an Enumerable
+
+      if status === 200
+        output_class = service.rpcs[rpc][:output_class]
+        Twirp::Encoding.decode(body, output_class, headers["Content-Type"])
+      else
+        Twirp::Client.error_from_response(Response.new(status, body, headers))
+      end
+    end
+
+    def set_controller_from_rack_env(env)
+      @controller = ActionDispatch::Request.new(env).controller_class
+    end
   end
 end

data/lib/rails_twirp/version.rb

@@ -1,3 +1,3 @@
 module RailsTwirp
-  VERSION = "0.5.0"
+  VERSION = "0.6.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rails_twirp
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.6.0
 platform: ruby
 authors:
 - Bouke van der Bijl
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-04-20 00:00:00.000000000 Z
+date: 2021-04-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -59,6 +59,9 @@ files:
 - lib/rails_twirp/base.rb
 - lib/rails_twirp/engine.rb
 - lib/rails_twirp/errors.rb
+- lib/rails_twirp/implicit_render.rb
+- lib/rails_twirp/instrumentation.rb
+- lib/rails_twirp/log_subscriber.rb
 - lib/rails_twirp/mapper.rb
 - lib/rails_twirp/render_pb.rb
 - lib/rails_twirp/rescue.rb