Add client request tracing.

Author: ioquatix

lib/async/http/protocol/http1/client.rb

@@ -5,6 +5,8 @@
 
 require_relative "connection"
 
+require "traces/provider"
+
 module Async
 	module HTTP
 		module Protocol
@@ -79,6 +81,20 @@ def call(request, task: Task.current)
 						self.close(error)
 						raise
 					end
+					
+					Traces::Provider(self) do
+						def write_request(...)
+							Traces.trace("async.http.protocol.http1.client.write_request") do
+								super
+							end
+						end
+						
+						def read_response(...)
+							Traces.trace("async.http.protocol.http1.client.read_response") do
+								super
+							end
+						end
+					end
 				end
 			end
 		end

lib/async/http/protocol/http2/client.rb

@@ -6,6 +6,7 @@
 require_relative "connection"
 require_relative "response"
 
+require "traces/provider"
 require "protocol/http2/client"
 
 module Async
@@ -34,10 +35,32 @@ def call(request)
 						@count += 1
 
 						response = create_response
+						write_request(response, request)
+						read_response(response)
+						
+						return response
+					end
+					
+					def write_request(response, request)
 						response.send_request(request)
+					end
+					
+					def read_response(response)
 						response.wait
+					end
+					
+					Traces::Provider(self) do
+						def write_request(...)
+							Traces.trace("async.http.protocol.http2.client.write_request") do
+								super
+							end
+						end
 
-						return response
+						def read_response(...)
+							Traces.trace("async.http.protocol.http2.client.read_response") do
+								super
+							end
+						end
 					end
 				end
 			end