memoize inheritable tags in TestSession

DataDog · Nov 22, 2023 · 56d0c20 · 56d0c20
1 parent 550d0d3
commit 56d0c20
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/lib/datadog/ci/test_session.rb b/lib/datadog/ci/test_session.rb
@@ -19,13 +19,15 @@ def finish
       end
 
       def inheritable_tags
+        return @inheritable_tags if defined?(@inheritable_tags)
+
         # this method is not synchronized because it does not iterate over the tags, but rather
         # uses synchronized method to get each tag value
         res = {}
         Ext::Test::INHERITABLE_TAGS.each do |tag|
           res[tag] = get_tag(tag)
         end
-        res
+        @inheritable_tags = res
       end
     end
   end

diff --git a/sig/datadog/ci/test_session.rbs b/sig/datadog/ci/test_session.rbs
@@ -1,6 +1,8 @@
 module Datadog
   module CI
     class TestSession < ConcurrentSpan
+      @inheritable_tags: Hash[untyped, untyped]
+
       def inheritable_tags: () -> Hash[untyped, untyped]
     end
   end