bitfab 0.10.5 → 0.10.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b789df310c91a7e9130f3bdb6a35e6237a8ddadede4df49a5f7dc4afc0525e91
-  data.tar.gz: 0cd79b9c9bc8bf15ef926625c0aa1aba8f3cf1798062c0d077e371d98b9f6aba
+  metadata.gz: f20353e1bb03affda1786ed5f8566ab1b3a401f0f9635023bae2fd350bcc7de6
+  data.tar.gz: 8e10f199e9d81cfdbfa2bc9172327131301dc3959307340bbe6c55e1ef2bc235
 SHA512:
-  metadata.gz: fbf62eeabc5741c36ef1b867730f548039f20331d6591ccae7a66cc48882cd4638877edc1b80d57c3a326827edfbedff1d3ec226ed3b4e3358f9041e3405ea5f
-  data.tar.gz: 579cb384585a42160d5f1c9941cbf0762f53d35ffda7f109cb40c4c52a291a02d446e113ca664a2c84a08916481fe86912eb386a58956fb5651ff1a6dc108771
+  metadata.gz: ade88c2b4d5878a3f6e0cee6f70c4a2d63b1a2f0359f425e9da6ab787806c8ef9396ca0e0a71f5ec7c75647e658419f0e9eb099fbbb4ce22a56c38adf26632e8
+  data.tar.gz: 9aa7b82e532a84893533b97d67d413cd4012729624d6bde8433bfdecf73c1d929269e6a1d779e50ed7d13e6dcc84a050f22fdf41282f2de4469bd0c5a6a35b30

data/lib/bitfab/client.rb

@@ -65,10 +65,15 @@ module Bitfab
       replay_ctx = ReplayContext.current
       resolved_test_run_id = replay_ctx&.dig(:test_run_id)
       resolved_input_source_span_id = replay_ctx&.dig(:input_source_span_id)
+      resolved_input_source_trace_id = replay_ctx&.dig(:input_source_trace_id)
 
       # Register trace state for root spans
       if is_root_span && !TraceState.get(trace_id)
-        TraceState.create(trace_id, test_run_id: resolved_test_run_id)
+        TraceState.create(
+          trace_id,
+          test_run_id: resolved_test_run_id,
+          input_source_trace_id: resolved_input_source_trace_id
+        )
       end
 
       if is_root_span
@@ -228,6 +233,9 @@ module Bitfab
       if trace_state&.dig(:contexts)
         raw_trace["contexts"] = trace_state[:contexts]
       end
+      if trace_state&.dig(:input_source_trace_id)
+        raw_trace["input_source_trace_id"] = trace_state[:input_source_trace_id]
+      end
 
       payload = {
         "type" => "sdk-function",

data/lib/bitfab/replay.rb

@@ -14,11 +14,12 @@ module Bitfab
 
     # Execute a block with replay context set on the current thread.
     # The context is automatically cleared when the block completes.
-    def with_context(test_run_id:, input_source_span_id: nil)
+    def with_context(test_run_id:, input_source_span_id: nil, input_source_trace_id: nil)
       previous = Thread.current[REPLAY_CONTEXT_KEY]
       Thread.current[REPLAY_CONTEXT_KEY] = {
         test_run_id:,
-        input_source_span_id:
+        input_source_span_id:,
+        input_source_trace_id:
       }
       yield
     ensure
@@ -117,7 +118,15 @@ module Bitfab
       span = http_client.get_external_span(server_item["externalSpanId"])
       item_data = extract_span_data(span)
       metrics = extract_server_item_metrics(server_item)
-      execute_item(item_data, receiver, method_name, test_run_id, span["id"], metrics)
+      execute_item(
+        item_data,
+        receiver,
+        method_name,
+        test_run_id,
+        span["id"],
+        metrics,
+        input_source_trace_id: span["externalTraceId"]
+      )
     end
 
     # Extract input/output data from an external span's rawData.
@@ -155,13 +164,14 @@ module Bitfab
     end
 
     # Execute a single replay item: deserialize inputs, call method with replay context.
-    def execute_item(item, receiver, method_name, test_run_id, input_source_span_id = nil, metrics = {})
+    def execute_item(item, receiver, method_name, test_run_id, input_source_span_id = nil, metrics = {},
+      input_source_trace_id: nil)
       args, kwargs = Serialize.deserialize_inputs(item)
 
       fn_result = nil
       fn_error = nil
 
-      ReplayContext.with_context(test_run_id:, input_source_span_id:) do
+      ReplayContext.with_context(test_run_id:, input_source_span_id:, input_source_trace_id:) do
         fn_result = if kwargs.empty?
           receiver.send(method_name, *args)
         else

data/lib/bitfab/span_context.rb

@@ -131,12 +131,13 @@ module Bitfab
       @states_mutex.synchronize { @states[trace_id] }
     end
 
-    def create(trace_id, test_run_id: nil)
+    def create(trace_id, test_run_id: nil, input_source_trace_id: nil)
       @states_mutex.synchronize do
         @states[trace_id] ||= {
           trace_id:,
           started_at: Time.now.utc.strftime("%Y-%m-%dT%H:%M:%S.%3NZ"),
-          test_run_id:
+          test_run_id:,
+          input_source_trace_id:
         }.compact
       end
     end

data/lib/bitfab/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Bitfab
-  VERSION = "0.10.5"
+  VERSION = "0.10.6"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bitfab
 version: !ruby/object:Gem::Version
-  version: 0.10.5
+  version: 0.10.6
 platform: ruby
 authors:
 - Harvest Team