minato-stackdriver 0.1.0.pre.3 → 0.1.0.pre.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 35e5b9fc732eef1bea6278908dc6a579aff7c970da01a13ec1b69759db729e7f
-  data.tar.gz: 675285811e330721194acb1057c72caa4fe3df5a370272d092f80a23cc59b4a0
+  metadata.gz: cd1d1350fad7bba843363307fc6ccbe853c646a4a78390fd0c9057a8b726d426
+  data.tar.gz: 80b2445e66eab767ea7856ffa4e224ef304cd8f379a476920990aea69eb249e7
 SHA512:
-  metadata.gz: 51cbf2d0a086d0fc0723fdbdd5f7e96a7bc4de7a6ade987312ee2130b62e1b994a5af38be9483ebd18050fb91c0c14cc9f839b09024fa7dbb23a1acf9c5b13c4
-  data.tar.gz: 3012d935e6a0b75c6f886dbc989c4aad12151bb3bd72d0b2231618a306a408e6ac0a074b28e7646cf54262ad67b3e4ed970a70248270b9caf190b5d4694b26b6
+  metadata.gz: 6ad75ec9e669e9488cf0592a273d26b118874fe4b1d3a271a132fe5188cb1246fae1e8ddfe26f3acd989a0c0aa97debe4462eb627aa4aef18ba547d10ace970d
+  data.tar.gz: 5db6c1139eeca524038c72d1f304444ddd823ce6312f5ff91a616608ce4f1a35859d3f75ebfb04c526176dd0ba52c7eb91554fe544cf9bb20837c124612d018e

data/lib/minato/stackdriver/middleware/distributed_trace_context.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module Minato
+  module Stackdriver
+    module Middleware
+      class DistributedTraceContext
+        TRACE_HEADER_KEY = "X-Cloud-Trace-Context"
+
+        def initialize(app)
+          @app = app
+        end
+
+        def call(env)
+          add_trace_context_header(env)
+
+          @app.call(env)
+        end
+
+        private
+
+        def add_trace_context_header(env)
+          return unless Rails.env.production?
+
+          env.request_headers.merge!({ "#{TRACE_HEADER_KEY}": "#{current_cloud_trace.trace.trace_id};o=1" })
+        end
+
+        def current_cloud_trace
+          @current_cloud_trace ||= current_trace_from_google_cloud
+        end
+
+        def current_trace_from_google_cloud
+          _trace_client = Google::Cloud::Trace.new
+          Google::Cloud::Trace.get
+        end
+      end
+    end
+  end
+end

data/lib/minato/stackdriver/middleware.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require "minato/stackdriver/middleware/distributed_trace_context"
+
+module Minato
+  module Stackdriver
+    module Middleware
+    end
+  end
+end

data/lib/minato/stackdriver/railtie.rb

@@ -3,8 +3,9 @@
 module Minato
   module Stackdriver
     class Railtie < ::Rails::Railtie
-      config.before_configuration do |app|
-        Minato::Stackdriver.init(app)
+      initializer "minato_stackdriver_setup",
+                  before: "load_environment_config" do
+        Minato::Stackdriver.init
       end
     end
   end

data/lib/minato/stackdriver/version.rb

@@ -2,6 +2,6 @@
 
 module Minato
   module Stackdriver
-    VERSION = "0.1.0-3"
+    VERSION = "0.1.0-5"
   end
 end

data/lib/minato/stackdriver.rb

@@ -2,16 +2,19 @@
 
 require "minato/stackdriver/version"
 require "minato/stackdriver/railtie"
+require "minato/stackdriver/middleware"
 require "stackdriver"
 
 module Minato
   module Stackdriver
     BLACKLIST_PATHS = ["/health/alive", "/health/ready"].freeze
 
-    def self.init(app)
-      app.config.google_cloud.use_logging = false
-      app.config.google_cloud.use_trace = true
-      app.config.google_cloud.trace_sampler = Google::Cloud::Trace::TimeSampler.new(path_blacklist: BLACKLIST_PATHS.dup)
+    def self.init
+      Google::Cloud.configure do |config|
+        config.use_logging = false
+        config.use_trace = true
+        config.trace.sampler = Google::Cloud::Trace::TimeSampler.new(path_blacklist: BLACKLIST_PATHS.dup)
+      end
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: minato-stackdriver
 version: !ruby/object:Gem::Version
-  version: 0.1.0.pre.3
+  version: 0.1.0.pre.5
 platform: ruby
 authors:
 - Ferreri
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-01-25 00:00:00.000000000 Z
+date: 2024-02-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -49,6 +49,8 @@ files:
 - README.md
 - Rakefile
 - lib/minato/stackdriver.rb
+- lib/minato/stackdriver/middleware.rb
+- lib/minato/stackdriver/middleware/distributed_trace_context.rb
 - lib/minato/stackdriver/railtie.rb
 - lib/minato/stackdriver/version.rb
 - lib/tasks/minato/stackdriver_tasks.rake