bitfab 0.58.3 → 0.58.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 959d7652dd7c6e3fdfea18c79e759300b2ebf826401f0e0601ea3642eae84034
-  data.tar.gz: bc7820f643314689a9440247edf0e81d4c234208ce1c903db7a9cb904da685fe
+  metadata.gz: 13b552211978874936f8b732245e948396758a5adf6d0485859cf98748bf4933
+  data.tar.gz: 612bbee201e37c8c425c5aa42bbcc6f713ca848e5cf43ad7eae5797beeefd660
 SHA512:
-  metadata.gz: a92544b809ea07e087e55b031d82130baf820f636cf41b4d90f4d1a0d1d3513024d733c6cdbe57e74dd7832552c0029816424f25c0fe65e1e585168b8908221e
-  data.tar.gz: 691217d535f95027773b65454c40470901a393aef97906283fe9f33ea38a05bd21b70105dbc8d0344c9fffaa1fb6839ecc99fc5b7480c879761189ca343e0b0d
+  metadata.gz: 7150bd10acde90cfe101845b6b6da248ae4cfb57a3742d8a33ef8d045e93d0ebfa6eaf6672376304f957206801c4f0b6ae60707d881d7621dce2d61b887405d4
+  data.tar.gz: beb42456465d624c22fc12d2294011be076748011bfac9dfad6a6397083bcebf6902a2fd7f7fadbdfa2c8e88c2df6b862c3a1f1bff0afb3f947e45256c68cc28

data/lib/bitfab/client.rb

@@ -425,6 +425,13 @@ module Bitfab
         span_id = SecureRandom.uuid
         parent_span_id = parent&.dig(:span_id)
         is_root_span = parent_span_id.nil?
+        ancestor_span_id = nearest_captured_ancestor_span_id
+        open_instrumentation = (surface == :trace) ? "trace" : "span"
+        sends_on_completion = lambda do
+          is_root_span || declared_node || !content_off_by_simulation_plan?(
+            trace_id:, trace_function_key:, span_name:, instrumentation: open_instrumentation
+          )
+        end
         started_at = Bitfab.now_iso_timestamp
         resolved_test_run_id = replay_ctx ? replay_ctx[:test_run_id] : test_run_id
         resolved_input_source_span_id = replay_ctx&.dig(:input_source_span_id)
@@ -550,6 +557,7 @@ module Bitfab
             span_data: content_off ? span_data.merge("content_off_by_simulation_plan" => true) : span_data,
             instrumentation: recorded_instrumentation,
             declared_node:,
+            nearest_captured_ancestor_span_id: ancestor_span_id,
             trace_function_key:,
             trace_id:,
             span_id:,
@@ -618,7 +626,8 @@ module Bitfab
             span_id:,
             explicit_span_receiver:,
             explicit_span_method_name: explicit_span_method_name&.to_sym,
-            surface:, recording_override:, content_withheld:
+            surface:, recording_override:, content_withheld:,
+            sends_on_completion: sends_on_completion
           ) do
             invoke = -> { around_body ? around_body.call { yield } : yield }
             result = if explicit_span_receiver
@@ -683,6 +692,14 @@ module Bitfab
       raise ArgumentError, "id must be a valid Bitfab span ID" unless id.is_a?(String) && UUID_PATTERN.match?(id)
     end
 
+    def nearest_captured_ancestor_span_id
+      SpanContext.stack.reverse_each do |entry|
+        sends = entry[:sends_on_completion]
+        return entry[:span_id] if sends.nil? || sends.call
+      end
+      nil
+    end
+
     def content_off_by_simulation_plan?(trace_id:, trace_function_key:, span_name:, instrumentation: "span")
       return false if SimulationPlan::FRAMEWORKS.include?(instrumentation)
 
@@ -927,7 +944,8 @@ module Bitfab
     def send_span(trace_function_key:, trace_id:, span_id:, parent_span_id:,
       span_name:, span_type:, function_name:, contexts:, prompt:, args:, kwargs:, result:, error:,
       started_at:, ended_at:, test_run_id: nil, input_source_span_id: nil, mocked: false,
-      mock_target: nil, mock_source: nil, span_data: {}, instrumentation: "span", declared_node: false)
+      mock_target: nil, mock_source: nil, span_data: {}, instrumentation: "span", declared_node: false,
+      nearest_captured_ancestor_span_id: nil)
       extra_span_data = span_data
       # If drop() was called on this trace, suppress the span PAYLOAD upload for
       # every span that completes after the flag was set. The trace completion
@@ -983,6 +1001,9 @@ module Bitfab
         "span_origin" => Bitfab.make_span_origin(instrumentation)
       }
       raw_span["parent_id"] = parent_span_id if parent_span_id
+      if nearest_captured_ancestor_span_id && nearest_captured_ancestor_span_id != parent_span_id
+        raw_span["nearest_captured_ancestor_id"] = nearest_captured_ancestor_span_id
+      end
       raw_span["input_source_span_id"] = input_source_span_id if input_source_span_id
 
       payload = {

data/lib/bitfab/span_context.rb

@@ -150,8 +150,9 @@ module Bitfab
 
     # Execute a block with a new span pushed onto the stack.
     # The span is automatically popped when the block completes.
-    def with_span(trace_id:, span_id:, explicit_span_receiver: nil, explicit_span_method_name: nil, surface: nil, recording_override: nil, content_withheld: nil)
+    def with_span(trace_id:, span_id:, explicit_span_receiver: nil, explicit_span_method_name: nil, surface: nil, recording_override: nil, content_withheld: nil, sends_on_completion: nil)
       entry = {trace_id:, span_id:}
+      entry[:sends_on_completion] = sends_on_completion if sends_on_completion
       entry[:surface] = surface if surface
       entry[:recording_override] = recording_override if recording_override
       entry[:content_withheld] = content_withheld if content_withheld

data/lib/bitfab/version.rb

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

metadata

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