aws-xray 0.13.0 → 0.14.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 2671bed4cd8bbe16d65b0b3419d55769afb82bd2
-  data.tar.gz: a22eb453d5126e550f79f8d5d514aebc75b2134d
+  metadata.gz: e367a0bf45a2a7055628dc8e2f41626f164ece6a
+  data.tar.gz: 0da9a4d1438e8ffe38f369c341af17f44342b0b3
 SHA512:
-  metadata.gz: 0407a0498d6d6f3ffc04f5548a230336ac9d80d0b5574ae5ee33f95bc98a692032f6212ece489baad95871b513bd64024ff20719a43a8314b7b1884fa8762fb9
-  data.tar.gz: e09183316b940b6ba40af92d0e95b34b1e984a3b81882807eac6c016237512c17afe8a22bea5c3c4f7dcb1ee26b442f928e01f3397c13ec8903364501a9dea4d
+  metadata.gz: c287819c6e3ea6b726675c420aec2f1a2ca94f315c85a4feba9e019e26f68a9ab59adfc8e02b7f6a3d42d40a449168031695e3f9f05bb6b501f79c25350a2860
+  data.tar.gz: 6ec72a4fa10a76c7ed6ff305a93d7a821fa2a002dcf04e27431673ac7c273afee571b90538e64e3ce3f160ad842f77bd9f3521185f8cfd3cdd5f302a2215575f

data/lib/aws/xray/context.rb

@@ -69,6 +69,7 @@ module Aws
         @trace = trace
         @base_segment_id = base_segment_id
         @disabled_ids = []
+        @sub_segment_name = nil
       end
 
       # Curretly context object is thread safe, so copying is not necessary,
@@ -111,7 +112,7 @@ module Aws
       # @return [Object] A value which given block returns.
       def child_trace(remote:, name:)
         raise SegmentDidNotStartError unless @base_segment_id
-        sub = SubSegment.build(@trace, @base_segment_id, remote: remote, name: name)
+        sub = SubSegment.build(@trace, @base_segment_id, remote: remote, name: overwrite_name(name))
 
         begin
           yield sub
@@ -141,6 +142,31 @@ module Aws
       def disabled?(id)
         @disabled_ids.include?(id)
       end
+
+      # Temporary overwrite sub segment with given data in given block.
+      # The overwriting will be occured only one time.
+      # CAUTION: the injection will NOT be propagated between threads!!
+      #
+      # @param [String] name
+      def overwrite_sub_segment(name:)
+        @sub_segment_name = name.to_s
+
+        begin
+          yield
+        ensure
+          @sub_segment_name = nil
+        end
+      end
+
+      private
+
+      def overwrite_name(name)
+        return name unless @sub_segment_name
+
+        name = @sub_segment_name
+        @sub_segment_name = nil
+        name
+      end
     end
   end
 end

data/lib/aws/xray/version.rb

@@ -1,5 +1,5 @@
 module Aws
   module Xray
-    VERSION = '0.13.0'
+    VERSION = '0.14.0'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aws-xray
 version: !ruby/object:Gem::Version
-  version: 0.13.0
+  version: 0.14.0
 platform: ruby
 authors:
 - Taiki Ono