timeouts for http;

x-yield · Apr 9, 2020 · 7a54e2d · 7a54e2d
1 parent f441621
commit 7a54e2d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/tankapi/session.go b/tankapi/session.go
@@ -7,17 +7,28 @@ import (
 	"fmt"
 	"io/ioutil"
 	"log"
+	"net"
 	"net/http"
 	"strings"
+	"time"
 )
 
 const (
 	createBreakpoint  = "init"
 	prepareBreakpoint = "start"
 )
 
+func dialTimeout(network, addr string) (net.Conn, error) {
+	return net.DialTimeout(network, addr, time.Second*5)
+}
+
+var transport = http.Transport{
+	Dial: dialTimeout,
+}
+
 var netClient = &http.Client{
-	Timeout: time.Second * 10,
+	Transport: &transport,
+	Timeout:   time.Second * 10,
 }
 
 type Session struct {

diff --git a/tankapi/worker.go b/tankapi/worker.go
@@ -4,7 +4,6 @@ import (
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
-	"net/http"
 )
 
 // Tank - represents tank api server; might have several sessions simultaneously