Release 0.1.3

Author: 闫明

beans/ab_config.go

@@ -39,8 +39,22 @@ type ABTestConfig struct {
 	*/
 	APIUrl string
 
+	/*
+		HTTP连接参数
+	*/
+	HTTPTransportParam HTTPTransportParam
+
 	/**
 	用于 SDK 埋点 SensorsAnalytics
 	*/
 	SensorsAnalytics sensorsanalytics.SensorsAnalytics
 }
+
+type HTTPTransportParam struct {
+	MaxIdleConnsPerHost         int
+	MaxIdleConns                int
+	MaxConnsPerHost             int
+	IdleConnTimeoutMilliSeconds int
+	DialTimeoutMilliSeconds     int
+	DialKeepAliveMilliSeconds   int
+}

beans/request_params.go

@@ -1,7 +1,5 @@
 package beans
 
-import "time"
-
 type RequestParam struct {
 	// 试验变量名
 	ParamName string
@@ -16,7 +14,7 @@ type RequestParam struct {
 	CustomIDs map[string]interface{}
 
 	// 网络请求超时时间，单位 ms，默认 3s
-	TimeoutMilliseconds time.Duration
+	TimeoutMilliseconds int
 
 	// 是否自动采集 A/B Testing 埋点事件
 	EnableAutoTrackABEvent bool

sensors_abtesting.go

@@ -8,7 +8,7 @@ import (
 )
 
 const (
-	SDK_VERSION = "0.1.2"
+	SDK_VERSION = "0.1.3"
 	LIB_NAME    = "Golang"
 )
 
@@ -158,5 +158,45 @@ func initConfig(abConfig beans.ABTestConfig) (error, beans.ABTestConfig) {
 	config.EnableRecordRequestCostTime = abConfig.EnableRecordRequestCostTime
 	config.APIUrl = abConfig.APIUrl
 	initCache(config)
+	utils.InitTransport(getHTTPTransPortParam(abConfig))
 	return nil, config
 }
+
+func getHTTPTransPortParam(abConfig beans.ABTestConfig) beans.HTTPTransportParam {
+	param := beans.HTTPTransportParam{}
+	if abConfig.HTTPTransportParam.MaxIdleConnsPerHost <= 0 {
+		param.MaxIdleConnsPerHost = 5
+	} else {
+		param.MaxIdleConnsPerHost = abConfig.HTTPTransportParam.MaxIdleConnsPerHost
+	}
+	if abConfig.HTTPTransportParam.MaxIdleConns <= 0 {
+		param.MaxIdleConns = 20
+	} else {
+		param.MaxIdleConns = abConfig.HTTPTransportParam.MaxIdleConns
+	}
+
+	if abConfig.HTTPTransportParam.MaxConnsPerHost <= 0 {
+		param.MaxConnsPerHost = 200
+	} else {
+		param.MaxConnsPerHost = abConfig.HTTPTransportParam.MaxConnsPerHost
+	}
+
+	if abConfig.HTTPTransportParam.IdleConnTimeoutMilliSeconds <= 0 {
+		param.IdleConnTimeoutMilliSeconds = 30 * 1000
+	} else {
+		param.IdleConnTimeoutMilliSeconds = abConfig.HTTPTransportParam.IdleConnTimeoutMilliSeconds
+	}
+
+	if abConfig.HTTPTransportParam.DialTimeoutMilliSeconds <= 0 {
+		param.DialTimeoutMilliSeconds = 30 * 1000
+	} else {
+		param.DialTimeoutMilliSeconds = abConfig.HTTPTransportParam.DialTimeoutMilliSeconds
+	}
+
+	if abConfig.HTTPTransportParam.DialKeepAliveMilliSeconds <= 0 {
+		param.DialKeepAliveMilliSeconds = 30 * 1000
+	} else {
+		param.DialKeepAliveMilliSeconds = abConfig.HTTPTransportParam.DialKeepAliveMilliSeconds
+	}
+	return param
+}

utils/network.go

@@ -7,12 +7,28 @@ import (
 	"fmt"
 	"github.com/sensorsdata/abtesting-sdk-go/beans"
 	"io/ioutil"
+	"net"
 	"net/http"
 	"strconv"
 	"strings"
 	"time"
 )
 
+var httpTransport = &http.Transport{}
+
+func InitTransport(httpTrans beans.HTTPTransportParam) {
+	httpTransport = &http.Transport{
+		DialContext: (&net.Dialer{
+			Timeout:   time.Duration(httpTrans.DialTimeoutMilliSeconds) * time.Millisecond,
+			KeepAlive: time.Duration(httpTrans.DialKeepAliveMilliSeconds) * time.Millisecond,
+		}).DialContext,
+		MaxIdleConns:        httpTrans.MaxIdleConns,
+		MaxIdleConnsPerHost: httpTrans.MaxIdleConnsPerHost,
+		MaxConnsPerHost:     httpTrans.MaxConnsPerHost,
+		IdleConnTimeout:     time.Duration(httpTrans.IdleConnTimeoutMilliSeconds) * time.Millisecond,
+	}
+}
+
 func RequestExperiment(url string, requestPrams map[string]interface{}, to time.Duration, enableRecordRequestCostTime bool) (Response, error) {
 	var resp *http.Response
 
@@ -24,7 +40,7 @@ func RequestExperiment(url string, requestPrams map[string]interface{}, to time.
 	req.Header.Add("X-AB-Request-Start-Time", fmt.Sprintf("%v", abRequestStartTime))
 	req.Header.Add("Content-Type", "application/json")
 
-	client := &http.Client{Timeout: to}
+	client := &http.Client{Timeout: to, Transport: httpTransport}
 	resp, err := client.Do(req)
 
 	if enableRecordRequestCostTime {