Fix custom https client

AlexxIT · Nov 5, 2023 · bcbebd5 · bcbebd5
1 parent 50e2a62
commit bcbebd5
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pkg/tcp/request.go b/pkg/tcp/request.go
@@ -27,6 +27,11 @@ func Do(req *http.Request) (*http.Response, error) {
 		}
 	}
 
+	if secure != nil {
+		ctx := context.WithValue(req.Context(), secureKey, secure)
+		req = req.WithContext(ctx)
+	}
+
 	if client == nil {
 		transport := http.DefaultTransport.(*http.Transport).Clone()
 
@@ -43,6 +48,7 @@ func Do(req *http.Request) (*http.Response, error) {
 			if err != nil {
 				return nil, err
 			}
+			secure := ctx.Value(connKey).(*tls.Config)
 			tlsConn := tls.Client(conn, secure)
 			if err = tlsConn.Handshake(); err != nil {
 				return nil, err
@@ -122,7 +128,7 @@ func Do(req *http.Request) (*http.Response, error) {
 }
 
 var client *http.Client
-var connKey struct{}
+var connKey, secureKey struct{}
 
 func WithConn() (context.Context, *net.Conn) {
 	pconn := new(net.Conn)