aws-xray 0.15.1 → 0.16.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ab1ededbeaec72532b3fd8640450ed0c72ef255f
-  data.tar.gz: 984d6f352917b1c3b279a2d859977e332aeac7fb
+  metadata.gz: cb65c48a877e87915776fa765ab77e7274287d1a
+  data.tar.gz: 5522ae2e4792cae8c313021d1705237d3734ad23
 SHA512:
-  metadata.gz: f5d81c1b147dbffa156ec45eb8b459c54a4c3a849e1a6bdbc45295b5436674bac29782d81884b0b562fd497f1363f3bdae464a026b5272da38668f60c0471c73
-  data.tar.gz: a27c730779d6fcda961858a7b97f95a9bea3380f93fe0aa29e74bb071aae1e6d5a0b7755e89297c5ba300de7255a66a04a6c35c4b230ee0877532fd33a9f4d6c
+  metadata.gz: a76f973fa970b3b29dd4d4755ac103fefe15905d235ee5cd231682d4bbc3d8dee3fef70e3a619a896daf45561f870f63805e1a50ca194c434e21415c74640f61
+  data.tar.gz: 8cd98e3d5d76249fd18aec23a6f883046cfd130813c68c35c09f2d3cc4b87b49285e5641746b04bd618c2aa20a49785cc5d027c154d21c40bbd9dd6b56dca681

data/lib/aws/xray/context.rb

@@ -27,8 +27,11 @@ module Aws
         end
 
         # @param [Aws::Xray::Context] context
-        def set_current(context)
+        def with_given_context(context)
           Thread.current.thread_variable_set(VAR_NAME, context)
+          yield
+        ensure
+          remove_current
         end
 
         # @return [Boolean]

data/lib/aws/xray/hooks/net_http.rb

@@ -6,7 +6,7 @@ module Aws
       module NetHttp
         NAME_HEADER = 'X-Aws-Xray-Name'.freeze
 
-        def request_with_aws_xray(req, *args)
+        def request_with_aws_xray(req, *args, &block)
           return request_without_aws_xray(req, *args) unless Context.started?
           return request_without_aws_xray(req, *args) if Context.current.disabled?(:net_http)
 
@@ -26,7 +26,7 @@ module Aws
             req[TRACE_HEADER] = propagate_trace.to_header_value
             sub.set_http_request(request_record)
 
-            res = request_without_aws_xray(req, *args)
+            res = request_without_aws_xray(req, *args, &block)
 
             sub.set_http_response(res.code.to_i, res['Content-Length'])
             case res.code.to_i

data/lib/aws/xray/version.rb

@@ -1,5 +1,5 @@
 module Aws
   module Xray
-    VERSION = '0.15.1'
+    VERSION = '0.16.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-xray
 version: !ruby/object:Gem::Version
-  version: 0.15.1
+  version: 0.16.0
 platform: ruby
 authors:
 - Taiki Ono
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-06-09 00:00:00.000000000 Z
+date: 2017-06-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday