feat: add untraced ctx method

open-telemetry · May 8, 2024 · 44cd1a4 · 44cd1a4
1 parent a1f1629
commit 44cd1a4
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/common/lib/opentelemetry/common/utilities.rb b/common/lib/opentelemetry/common/utilities.rb
@@ -93,6 +93,10 @@ def untraced
         end
       end
 
+      def untraced_context(parent_context: Context.current)
+        parent_context.set_value(UNTRACED_KEY, true)
+      end
+
       # Detects whether the current context has been set to disable tracing.
       def untraced?(context = nil)
         context ||= Context.current

diff --git a/common/test/opentelemetry/common/utilities_test.rb b/common/test/opentelemetry/common/utilities_test.rb
@@ -26,6 +26,13 @@ def shutdown(timeout: nil); end
       common_utils.untraced {}
       assert_equal(false, common_utils.untraced?)
     end
+
+    it 'supports non block format' do
+      token = OpenTelemetry::Context.attach(common_utils.untraced_context)
+      assert_equal(true, common_utils.untraced?)
+      OpenTelemetry::Context.detach(token)
+      assert_equal(false, common_utils.untraced?)
+    end
   end
 
   describe '#utf8_encode' do