Using jaeger and adding timeout to request

topfreegames · Jun 14, 2018 · dc086e7 · dc086e7
1 parent 41f83fc
commit dc086e7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/Gopkg.lock b/Gopkg.lock
diff --git a/lib/lib.go b/lib/lib.go
@@ -9,8 +9,10 @@ import (
 	"net/http"
 	"strings"
 	"sync"
+	"time"
 
 	"github.com/spf13/viper"
+	ehttp "github.com/topfreegames/extensions/http"
 )
 
 // Khan is a struct that represents a khan API application
@@ -28,17 +30,21 @@ var (
 	once   sync.Once
 )
 
-func getHTTPClient() *http.Client {
+func getHTTPClient(timeoutMs int) *http.Client {
 	once.Do(func() {
-		client = &http.Client{}
+		client = &http.Client{
+			Timeout: time.Duration(timeoutMs) * time.Millisecond,
+		}
+		ehttp.Instrument(client)
 	})
 	return client
 }
 
 // NewKhan returns a new khan API application
 func NewKhan(config *viper.Viper) KhanInterface {
+	config.SetDefault("khan.timeout", 1000)
 	k := &Khan{
-		httpClient: getHTTPClient(),
+		httpClient: getHTTPClient(config.GetInt("khan.timeout")),
 		Config:     config,
 		url:        config.GetString("khan.url"),
 		user:       config.GetString("khan.user"),