- basic mtps api added.

platform/common/common.go

@@ -0,0 +1,83 @@
+package common
+
+import (
+	"errors"
+	"net"
+	"net/http"
+	"time"
+
+	"go.uber.org/zap"
+)
+
+type AccessPlatformAPIWithRetryParams struct {
+	MaxExceedLimitRetryCount   int
+	MaxRecoverableRetryCount   int
+	SleepSecondWhenExceedLimit time.Duration
+	Client                     *http.Client
+	Request                    *http.Request
+	SugarLogger                *zap.SugaredLogger
+}
+
+const (
+	PAErrorLevelSuccess     = 0
+	PAErrorLevelExceedLimit = 1
+	PAErrorLevelRecoverable = 2
+	PAErrorLevelFailed      = 3
+)
+
+var (
+	ErrRecoverableErrMaxRetry = errors.New("recoverable error reach max retry count!")
+	ErrLimitReachMaxRetry     = errors.New("Reach max retry count!")
+	ErrHttpCode               = errors.New("HTTP Code is not 200")
+	ErrBusinessCode           = errors.New("Business code is not ok")
+)
+
+func AccessPlatformAPIWithRetry(params *AccessPlatformAPIWithRetryParams, handleResponse func(response *http.Response) (int, error)) error {
+	exceedLimitRetryCount := 0
+	recoverableErrorRetryCount := 0
+	for {
+		response, err := params.Client.Do(params.Request)
+		if err != nil {
+			params.SugarLogger.Debugf("client.Get return err:%v", err)
+			err, ok := err.(net.Error)
+			recoverableErrorRetryCount++
+			if ok && err.Timeout() && recoverableErrorRetryCount <= params.MaxRecoverableRetryCount {
+				continue
+			} else {
+				return err
+			}
+		}
+		defer response.Body.Close()
+		if response.StatusCode != 200 {
+			params.SugarLogger.Debugf("http code is:%d", response.StatusCode)
+			recoverableErrorRetryCount++
+			if recoverableErrorRetryCount <= params.MaxRecoverableRetryCount {
+				continue
+			}
+
+			return ErrHttpCode
+		}
+
+		errLevel, err := handleResponse(response)
+		if err != nil {
+			return err
+		}
+		if errLevel == PAErrorLevelSuccess {
+			return nil
+		} else if errLevel == PAErrorLevelExceedLimit {
+			exceedLimitRetryCount++
+			if exceedLimitRetryCount <= params.MaxExceedLimitRetryCount {
+				time.Sleep(params.SleepSecondWhenExceedLimit)
+			} else {
+				return ErrLimitReachMaxRetry
+			}
+		} else if errLevel == PAErrorLevelRecoverable {
+			if recoverableErrorRetryCount <= params.MaxRecoverableRetryCount {
+				continue
+			}
+			return ErrRecoverableErrMaxRetry
+		} else {
+			return ErrBusinessCode
+		}
+	}
+}