Release 0.1.2

Author: qinwenshi

.gitignore

@@ -4,4 +4,6 @@
 # --------
 # IntelliJ
 *.iml
-.idea/*
+.idea/*
+go.mod
+go.sum

beans/ab_config.go

@@ -29,6 +29,11 @@ type ABTestConfig struct {
 	*/
 	EnableEventCache bool
 
+	/**
+	开启请求耗时记录
+	*/
+	EnableRecordRequestCostTime bool
+
 	/*
 		API 地址
 	*/

experiment_helper.go

@@ -22,7 +22,7 @@ func loadExperimentFromNetwork(sensors *SensorsABTest, distinctId string, isLogi
 	if requestParam.TimeoutMilliseconds <= 0 {
 		requestParam.TimeoutMilliseconds = 3 * 1000
 	}
-	response, err := requestExperimentOnNetwork(sensors.config.APIUrl, distinctId, isLoginId, requestParam)
+	response, err := requestExperimentOnNetwork(sensors.config.APIUrl, distinctId, isLoginId, requestParam, sensors.config.EnableRecordRequestCostTime)
 	if err != nil {
 		return err, beans.Experiment{
 			Result: requestParam.DefaultValue,
@@ -88,7 +88,7 @@ func loadExperimentFromCache(sensors *SensorsABTest, distinctId string, isLoginI
 	var outExperiments []beans.InnerExperiment
 	if isRequestNetwork {
 		// 从网络请求试验
-		response, err := requestExperimentOnNetwork(sensors.config.APIUrl, distinctId, isLoginId, requestParam)
+		response, err := requestExperimentOnNetwork(sensors.config.APIUrl, distinctId, isLoginId, requestParam, sensors.config.EnableRecordRequestCostTime)
 		if err != nil {
 			return err, beans.Experiment{
 				Result: requestParam.DefaultValue,
@@ -134,11 +134,11 @@ func loadExperimentFromCache(sensors *SensorsABTest, distinctId string, isLoginI
 }
 
 // 从网络加载试验
-func requestExperimentOnNetwork(apiUrl string, distinctId string, isLoginId bool, requestParam beans.RequestParam) (utils.Response, error) {
+func requestExperimentOnNetwork(apiUrl string, distinctId string, isLoginId bool, requestParam beans.RequestParam, enableRecordRequestCostTime bool) (utils.Response, error) {
 	if requestParam.TimeoutMilliseconds <= 0 {
 		requestParam.TimeoutMilliseconds = 3 * 1000
 	}
-	return utils.RequestExperiment(apiUrl, buildRequestParam(distinctId, isLoginId, requestParam), time.Duration(requestParam.TimeoutMilliseconds)*time.Millisecond)
+	return utils.RequestExperiment(apiUrl, buildRequestParam(distinctId, isLoginId, requestParam), time.Duration(requestParam.TimeoutMilliseconds)*time.Millisecond, enableRecordRequestCostTime)
 }
 
 func trackABTestEventOuter(distinctId string, isLoginId bool, experiment beans.Experiment, sensors *SensorsABTest, properties map[string]interface{}, customIDs map[string]interface{}) {

sensors_abtesting.go

@@ -8,7 +8,7 @@ import (
 )
 
 const (
-	SDK_VERSION = "0.1.1"
+	SDK_VERSION = "0.1.2"
 	LIB_NAME    = "Golang"
 )
 
@@ -155,6 +155,7 @@ func initConfig(abConfig beans.ABTestConfig) (error, beans.ABTestConfig) {
 
 	config.SensorsAnalytics = abConfig.SensorsAnalytics
 	config.EnableEventCache = abConfig.EnableEventCache
+	config.EnableRecordRequestCostTime = abConfig.EnableRecordRequestCostTime
 	config.APIUrl = abConfig.APIUrl
 	initCache(config)
 	return nil, config

utils/network.go

@@ -4,25 +4,34 @@ import (
 	"bytes"
 	"encoding/json"
 	"errors"
+	"fmt"
 	"github.com/sensorsdata/abtesting-sdk-go/beans"
 	"io/ioutil"
 	"net/http"
+	"strconv"
 	"strings"
 	"time"
 )
 
-func RequestExperiment(url string, requestPrams map[string]interface{}, to time.Duration) (Response, error) {
+func RequestExperiment(url string, requestPrams map[string]interface{}, to time.Duration, enableRecordRequestCostTime bool) (Response, error) {
 	var resp *http.Response
 
 	data, _ := json.Marshal(requestPrams)
 
 	req, _ := http.NewRequest("POST", url, bytes.NewReader(data))
 
+	abRequestStartTime := time.Now().UnixNano() / int64(time.Millisecond)
+	req.Header.Add("X-AB-Request-Start-Time", fmt.Sprintf("%v", abRequestStartTime))
 	req.Header.Add("Content-Type", "application/json")
 
 	client := &http.Client{Timeout: to}
 	resp, err := client.Do(req)
 
+	if enableRecordRequestCostTime {
+		abRequestEndTime := time.Now().UnixNano() / int64(time.Millisecond)
+		recordAbRequestCostTime(resp, abRequestStartTime, abRequestEndTime)
+	}
+
 	if err != nil {
 		return Response{}, err
 	}
@@ -49,6 +58,36 @@ func RequestExperiment(url string, requestPrams map[string]interface{}, to time.
 	}
 }
 
+func recordAbRequestCostTime(response *http.Response, abRequestStartTime int64, abRequestEndTime int64) {
+	abRequestId := getAbRequestIdFromResponse(response)
+	abRequestProcessTime := getAbRequestProcessTimeFromResponse(response)
+	abRequestTotalTime := strconv.FormatInt(abRequestEndTime-abRequestStartTime, 10)
+	fmt.Println("record ab request time consumption. requestId: ", abRequestId, ", requestTotalTime:", abRequestTotalTime, "ms, abRequestProcessTime:", abRequestProcessTime, "ms")
+}
+
+func getAbRequestIdFromResponse(response *http.Response) (abRequestId string) {
+	if response != nil && response.Header != nil {
+		abRequestId = response.Header.Get("X-AB-Request-Id")
+		if abRequestId == "" {
+			abRequestId = response.Header.Get("X-Request-Id")
+		}
+		if abRequestId != "" {
+			return abRequestId
+		}
+	}
+	return "unknown (not found)"
+}
+
+func getAbRequestProcessTimeFromResponse(response *http.Response) (abRequestProcessTime string) {
+	if response != nil && response.Header != nil {
+		abRequestProcessTime = response.Header.Get("X-AB-Request-Process-Time")
+		if abRequestProcessTime != "" {
+			return abRequestProcessTime
+		}
+	}
+	return "unknown (not found)"
+}
+
 func defaultTrackConfig(response *Response, resMaps map[string]interface{}) {
 	if !response.TrackConfig.TriggerSwitch {
 		return