Compare commits

...

1 Commits

Author SHA1 Message Date
yndu13 8ac445bc69 [v2.x]Rebuild retry policy 2024-03-19 21:22:17 +08:00
16 changed files with 1639 additions and 628 deletions
+1 -1
View File
@@ -1,4 +1,4 @@
module github.com/alibabacloud-go/tea
module github.com/alibabacloud-go/tea/v2
go 1.14
+117
View File
@@ -0,0 +1,117 @@
package tea
import (
"math"
"math/rand"
)
type BackoffPolicy interface {
GetDelayTimeMillis(ctx *RetryPolicyContext) *int64
}
func NewBackoffPolicy(options map[string]interface{}) (backoffPolicy BackoffPolicy) {
policy := StringValue(TransInterfaceToString(options["policy"]))
switch policy {
case "Fixed":
backoffPolicy = &FixedBackoffPolicy{
Period: TransInterfaceToInt(options["period"]),
}
return
case "Random":
backoffPolicy = &RandomBackoffPolicy{
Period: TransInterfaceToInt(options["period"]),
Cap: TransInterfaceToInt64(options["cap"]),
}
return
case "Exponential":
backoffPolicy = &ExponentialBackoffPolicy{
Period: TransInterfaceToInt(options["period"]),
Cap: TransInterfaceToInt64(options["cap"]),
}
return
case "EqualJitter":
backoffPolicy = &EqualJitterBackoffPolicy{
Period: TransInterfaceToInt(options["period"]),
Cap: TransInterfaceToInt64(options["cap"]),
}
return
case "ExponentialWithEqualJitter":
backoffPolicy = &EqualJitterBackoffPolicy{
Period: TransInterfaceToInt(options["period"]),
Cap: TransInterfaceToInt64(options["cap"]),
}
return
case "FullJitter":
backoffPolicy = &FullJitterBackoffPolicy{
Period: TransInterfaceToInt(options["period"]),
Cap: TransInterfaceToInt64(options["cap"]),
}
return
case "ExponentialWithFullJitter":
backoffPolicy = &FullJitterBackoffPolicy{
Period: TransInterfaceToInt(options["period"]),
Cap: TransInterfaceToInt64(options["cap"]),
}
return
}
return nil
}
type FixedBackoffPolicy struct {
Period *int
}
func (fixedBackoff *FixedBackoffPolicy) GetDelayTimeMillis(ctx *RetryPolicyContext) *int64 {
return Int64(int64(IntValue(fixedBackoff.Period)))
}
type RandomBackoffPolicy struct {
Period *int
Cap *int64
}
func (randomBackoff *RandomBackoffPolicy) GetDelayTimeMillis(ctx *RetryPolicyContext) *int64 {
if randomBackoff.Cap == nil {
randomBackoff.Cap = Int64(20 * 1000)
}
ceil := math.Min(float64(*randomBackoff.Cap), float64(IntValue(randomBackoff.Period))*float64(IntValue(ctx.RetriesAttempted)))
return Int64(int64(rand.Float64() * ceil))
}
type ExponentialBackoffPolicy struct {
Period *int
Cap *int64
}
func (exponentialBackoff *ExponentialBackoffPolicy) GetDelayTimeMillis(ctx *RetryPolicyContext) *int64 {
if exponentialBackoff.Cap == nil {
exponentialBackoff.Cap = Int64(3 * 24 * 60 * 60 * 1000)
}
return Int64(int64(math.Min(float64(*exponentialBackoff.Cap), float64(IntValue(exponentialBackoff.Period))*math.Pow(2.0, float64(IntValue(ctx.RetriesAttempted))))))
}
type EqualJitterBackoffPolicy struct {
Period *int
Cap *int64
}
func (equalJitterBackoff *EqualJitterBackoffPolicy) GetDelayTimeMillis(ctx *RetryPolicyContext) *int64 {
if equalJitterBackoff.Cap == nil {
equalJitterBackoff.Cap = Int64(3 * 24 * 60 * 60 * 1000)
}
ceil := math.Min(float64(*equalJitterBackoff.Cap), float64(IntValue(equalJitterBackoff.Period))*math.Pow(2.0, float64(IntValue(ctx.RetriesAttempted))))
return Int64(int64(ceil/2 + rand.Float64()*(ceil/2+1)))
}
type FullJitterBackoffPolicy struct {
Period *int
Cap *int64
}
func (fullJitterBackof *FullJitterBackoffPolicy) GetDelayTimeMillis(ctx *RetryPolicyContext) *int64 {
if fullJitterBackof.Cap == nil {
fullJitterBackof.Cap = Int64(3 * 24 * 60 * 60 * 1000)
}
ceil := math.Min(float64(*fullJitterBackof.Cap), float64(IntValue(fullJitterBackof.Period))*math.Pow(2.0, float64(IntValue(ctx.RetriesAttempted))))
return Int64(int64(rand.Float64() * ceil))
}
+135
View File
@@ -0,0 +1,135 @@
package tea
import (
"reflect"
"testing"
"github.com/alibabacloud-go/tea/v2/utils"
)
func TestBackoffPolicy(t *testing.T) {
var backoffPolicy BackoffPolicy
backoffPolicy = NewBackoffPolicy(map[string]interface{}{
"policy": "Any",
})
utils.AssertEqual(t, nil, backoffPolicy)
backoffPolicy = NewBackoffPolicy(map[string]interface{}{
"policy": "Fixed",
"period": 1000,
})
typeOfPolicy := reflect.TypeOf(backoffPolicy)
utils.AssertEqual(t, "FixedBackoffPolicy", typeOfPolicy.Elem().Name())
backoffPolicy = NewBackoffPolicy(map[string]interface{}{
"policy": "Random",
"period": 2,
"cap": int64(60 * 1000),
})
typeOfPolicy = reflect.TypeOf(backoffPolicy)
utils.AssertEqual(t, "RandomBackoffPolicy", typeOfPolicy.Elem().Name())
backoffPolicy = NewBackoffPolicy(map[string]interface{}{
"policy": "Exponential",
"period": 2,
"cap": int64(60 * 1000),
})
typeOfPolicy = reflect.TypeOf(backoffPolicy)
utils.AssertEqual(t, "ExponentialBackoffPolicy", typeOfPolicy.Elem().Name())
backoffPolicy = NewBackoffPolicy(map[string]interface{}{
"policy": "EqualJitter",
"period": 2,
"cap": int64(60 * 1000),
})
typeOfPolicy = reflect.TypeOf(backoffPolicy)
utils.AssertEqual(t, "EqualJitterBackoffPolicy", typeOfPolicy.Elem().Name())
backoffPolicy = NewBackoffPolicy(map[string]interface{}{
"policy": "ExponentialWithEqualJitter",
"period": 2,
"cap": int64(60 * 1000),
})
typeOfPolicy = reflect.TypeOf(backoffPolicy)
utils.AssertEqual(t, "EqualJitterBackoffPolicy", typeOfPolicy.Elem().Name())
backoffPolicy = NewBackoffPolicy(map[string]interface{}{
"policy": "FullJitter",
"period": 2,
"cap": int64(60 * 1000),
})
typeOfPolicy = reflect.TypeOf(backoffPolicy)
utils.AssertEqual(t, "FullJitterBackoffPolicy", typeOfPolicy.Elem().Name())
backoffPolicy = NewBackoffPolicy(map[string]interface{}{
"policy": "ExponentialWithFullJitter",
"period": 2,
"cap": int64(60 * 1000),
})
typeOfPolicy = reflect.TypeOf(backoffPolicy)
utils.AssertEqual(t, "FullJitterBackoffPolicy", typeOfPolicy.Elem().Name())
}
func TestFixedBackoffPolicy(t *testing.T) {
backoffPolicy := FixedBackoffPolicy{
Period: Int(1000),
}
utils.AssertEqual(t, int64(1000), Int64Value(backoffPolicy.GetDelayTimeMillis(nil)))
retryPolicyContext := RetryPolicyContext{
RetriesAttempted: Int(1),
}
utils.AssertEqual(t, int64(1000), Int64Value(backoffPolicy.GetDelayTimeMillis(&retryPolicyContext)))
retryPolicyContext = RetryPolicyContext{
RetriesAttempted: Int(2),
}
utils.AssertEqual(t, int64(1000), Int64Value(backoffPolicy.GetDelayTimeMillis(&retryPolicyContext)))
}
func TestRandomBackoffPolicy(t *testing.T) {
backoffPolicy := RandomBackoffPolicy{
Period: Int(2),
}
retryPolicyContext := RetryPolicyContext{
RetriesAttempted: Int(1),
}
utils.AssertEqual(t, true, Int64Value(backoffPolicy.GetDelayTimeMillis(&retryPolicyContext)) < 2)
retryPolicyContext = RetryPolicyContext{
RetriesAttempted: Int(2),
}
utils.AssertEqual(t, true, Int64Value(backoffPolicy.GetDelayTimeMillis(&retryPolicyContext)) < 4)
}
func TestExponentialBackoffPolicy(t *testing.T) {
backoffPolicy := ExponentialBackoffPolicy{
Period: Int(2),
}
retryPolicyContext := RetryPolicyContext{
RetriesAttempted: Int(1),
}
utils.AssertEqual(t, int64(4), Int64Value(backoffPolicy.GetDelayTimeMillis(&retryPolicyContext)))
retryPolicyContext = RetryPolicyContext{
RetriesAttempted: Int(2),
}
utils.AssertEqual(t, int64(8), Int64Value(backoffPolicy.GetDelayTimeMillis(&retryPolicyContext)))
}
func TestEqualJitterBackoffPolicy(t *testing.T) {
backoffPolicy := EqualJitterBackoffPolicy{
Period: Int(2),
}
retryPolicyContext := RetryPolicyContext{
RetriesAttempted: Int(1),
}
utils.AssertEqual(t, true, Int64Value(backoffPolicy.GetDelayTimeMillis(&retryPolicyContext)) < 5)
retryPolicyContext = RetryPolicyContext{
RetriesAttempted: Int(2),
}
utils.AssertEqual(t, true, Int64Value(backoffPolicy.GetDelayTimeMillis(&retryPolicyContext)) < 9)
}
func TestFullJitterBackoffPolicy(t *testing.T) {
backoffPolicy := FullJitterBackoffPolicy{
Period: Int(2),
}
retryPolicyContext := RetryPolicyContext{
RetriesAttempted: Int(1),
}
utils.AssertEqual(t, true, Int64Value(backoffPolicy.GetDelayTimeMillis(&retryPolicyContext)) < 4)
retryPolicyContext = RetryPolicyContext{
RetriesAttempted: Int(2),
}
utils.AssertEqual(t, true, Int64Value(backoffPolicy.GetDelayTimeMillis(&retryPolicyContext)) < 8)
}
File diff suppressed because it is too large Load Diff
File diff suppressed because it is too large Load Diff
+152
View File
@@ -0,0 +1,152 @@
package tea
import (
"bytes"
"encoding/json"
"fmt"
"reflect"
"strconv"
)
// BaseError is an interface for getting actual error
type BaseError interface {
error
ErrorName() *string
ErrorCode() *string
RetryAfterTimeMillis() *int64
}
// CastError is used for cast type fails
type CastError struct {
Message *string
Code *string
}
// NewCastError is used for cast type fails
func NewCastError(message *string) *CastError {
return &CastError{
Message: message,
Code: nil,
}
}
// Return message of CastError
func (err *CastError) Error() string {
return StringValue(err.Message)
}
func (err *CastError) ErrorName() *string {
return String("CastError")
}
func (err *CastError) ErrorCode() *string {
return err.Code
}
func (err *CastError) RetryAfterTimeMillis() *int64 {
return nil
}
// SDKError struct is used save error code and message
type SDKError struct {
Code *string
StatusCode *int
Message *string
Data *string
Stack *string
errMsg *string
Description *string
AccessDeniedDetail map[string]interface{}
}
// NewSDKError is used for shortly create SDKError object
func NewSDKError(obj map[string]interface{}) *SDKError {
err := &SDKError{}
if val, ok := obj["code"].(int); ok {
err.Code = String(strconv.Itoa(val))
} else if val, ok := obj["code"].(string); ok {
err.Code = String(val)
}
if obj["message"] != nil {
err.Message = String(obj["message"].(string))
}
if obj["description"] != nil {
err.Description = String(obj["description"].(string))
}
if detail := obj["accessDeniedDetail"]; detail != nil {
r := reflect.ValueOf(detail)
if r.Kind().String() == "map" {
res := make(map[string]interface{})
tmp := r.MapKeys()
for _, key := range tmp {
res[key.String()] = r.MapIndex(key).Interface()
}
err.AccessDeniedDetail = res
}
}
if data := obj["data"]; data != nil {
r := reflect.ValueOf(data)
if r.Kind().String() == "map" {
res := make(map[string]interface{})
tmp := r.MapKeys()
for _, key := range tmp {
res[key.String()] = r.MapIndex(key).Interface()
}
if statusCode := res["statusCode"]; statusCode != nil {
if code, ok := statusCode.(int); ok {
err.StatusCode = Int(code)
} else if tmp, ok := statusCode.(string); ok {
code, err_ := strconv.Atoi(tmp)
if err_ == nil {
err.StatusCode = Int(code)
}
} else if code, ok := statusCode.(*int); ok {
err.StatusCode = code
}
}
}
byt := bytes.NewBuffer([]byte{})
jsonEncoder := json.NewEncoder(byt)
jsonEncoder.SetEscapeHTML(false)
jsonEncoder.Encode(data)
err.Data = String(string(bytes.TrimSpace(byt.Bytes())))
}
if statusCode, ok := obj["statusCode"].(int); ok {
err.StatusCode = Int(statusCode)
} else if status, ok := obj["statusCode"].(string); ok {
statusCode, err_ := strconv.Atoi(status)
if err_ == nil {
err.StatusCode = Int(statusCode)
}
}
return err
}
// Set ErrMsg by msg
func (err *SDKError) SetErrMsg(msg string) {
err.errMsg = String(msg)
}
func (err *SDKError) Error() string {
if err.errMsg == nil {
str := fmt.Sprintf("SDKError:\n StatusCode: %d\n Code: %s\n Message: %s\n Data: %s\n",
IntValue(err.StatusCode), StringValue(err.Code), StringValue(err.Message), StringValue(err.Data))
err.SetErrMsg(str)
}
return StringValue(err.errMsg)
}
func (err *SDKError) ErrorName() *string {
return String("SDKError")
}
func (err *SDKError) ErrorCode() *string {
return err.Code
}
func (err *SDKError) RetryAfterTimeMillis() *int64 {
return nil
}
+115
View File
@@ -0,0 +1,115 @@
package tea
import (
"testing"
"github.com/alibabacloud-go/tea/v2/utils"
)
func TestCastError(t *testing.T) {
var err BaseError
err = NewCastError(String("cast error"))
utils.AssertEqual(t, "cast error", err.Error())
utils.AssertEqual(t, "", StringValue(err.ErrorCode()))
utils.AssertEqual(t, "CastError", StringValue(err.ErrorName()))
utils.AssertNil(t, err.RetryAfterTimeMillis())
}
func TestSDKError(t *testing.T) {
var err0 BaseError
err0 = NewSDKError(map[string]interface{}{
"code": "code",
"statusCode": 404,
"message": "message",
"data": map[string]interface{}{
"httpCode": "404",
"requestId": "dfadfa32cgfdcasd4313",
"hostId": "github.com/alibabacloud/tea",
"recommend": "https://中文?q=a.b&product=c&requestId=123",
},
"description": "description",
"accessDeniedDetail": map[string]interface{}{
"AuthAction": "ram:ListUsers",
"AuthPrincipalType": "SubUser",
"PolicyType": "ResourceGroupLevelIdentityBassdPolicy",
"NoPermissionType": "ImplicitDeny",
"UserId": 123,
},
})
utils.AssertNotNil(t, err0)
utils.AssertEqual(t, "SDKError:\n StatusCode: 404\n Code: code\n Message: message\n Data: {\"hostId\":\"github.com/alibabacloud/tea\",\"httpCode\":\"404\",\"recommend\":\"https://中文?q=a.b&product=c&requestId=123\",\"requestId\":\"dfadfa32cgfdcasd4313\"}\n", err0.Error())
var err *SDKError
err = err0.(*SDKError)
err.SetErrMsg("test")
utils.AssertEqual(t, "test", err.Error())
utils.AssertEqual(t, 404, *err.StatusCode)
utils.AssertEqual(t, "description", *err.Description)
utils.AssertEqual(t, "ImplicitDeny", err.AccessDeniedDetail["NoPermissionType"])
utils.AssertEqual(t, 123, err.AccessDeniedDetail["UserId"])
utils.AssertNil(t, err.RetryAfterTimeMillis())
err = NewSDKError(map[string]interface{}{
"statusCode": "404",
"data": map[string]interface{}{
"statusCode": 500,
},
})
utils.AssertNotNil(t, err)
utils.AssertEqual(t, 404, *err.StatusCode)
err = NewSDKError(map[string]interface{}{
"data": map[string]interface{}{
"statusCode": 500,
},
})
utils.AssertNotNil(t, err)
utils.AssertEqual(t, 500, *err.StatusCode)
err = NewSDKError(map[string]interface{}{
"data": map[string]interface{}{
"statusCode": Int(500),
},
})
utils.AssertNotNil(t, err)
utils.AssertEqual(t, 500, *err.StatusCode)
err = NewSDKError(map[string]interface{}{
"data": map[string]interface{}{
"statusCode": "500",
},
})
utils.AssertNotNil(t, err)
utils.AssertEqual(t, 500, *err.StatusCode)
err = NewSDKError(map[string]interface{}{
"code": "code",
"message": "message",
"data": map[string]interface{}{
"requestId": "dfadfa32cgfdcasd4313",
},
})
utils.AssertNotNil(t, err)
utils.AssertNil(t, err.StatusCode)
err = NewSDKError(map[string]interface{}{
"code": 400,
"message": "message",
"data": "string data",
})
utils.AssertNotNil(t, err)
utils.AssertNotNil(t, err.Data)
utils.AssertEqual(t, "400", StringValue(err.Code))
utils.AssertNil(t, err.StatusCode)
}
func TestSDKErrorCode404(t *testing.T) {
err := NewSDKError(map[string]interface{}{
"statusCode": 404,
"code": "NOTFOUND",
"message": "message",
})
utils.AssertNotNil(t, err)
utils.AssertEqual(t, "SDKError:\n StatusCode: 404\n Code: NOTFOUND\n Message: message\n Data: \n", err.Error())
utils.AssertEqual(t, "NOTFOUND", StringValue(err.ErrorCode()))
utils.AssertEqual(t, "SDKError", StringValue(err.ErrorName()))
}
+81
View File
@@ -0,0 +1,81 @@
package tea
import (
"bytes"
"encoding/json"
"errors"
"fmt"
"net/http"
"time"
)
func Prettify(i interface{}) string {
resp, _ := json.MarshalIndent(i, "", " ")
return string(resp)
}
func Sleep(backoffTime *int64) {
sleeptime := time.Duration(Int64Value(backoffTime)) * time.Millisecond
time.Sleep(sleeptime)
}
func Merge(args ...interface{}) map[string]*string {
finalArg := make(map[string]*string)
for _, obj := range args {
switch obj.(type) {
case map[string]*string:
arg := obj.(map[string]*string)
for key, value := range arg {
if value != nil {
finalArg[key] = value
}
}
default:
byt, _ := json.Marshal(obj)
arg := make(map[string]string)
err := json.Unmarshal(byt, &arg)
if err != nil {
return finalArg
}
for key, value := range arg {
if value != "" {
finalArg[key] = String(value)
}
}
}
}
return finalArg
}
// Convert is use convert map[string]interface object to struct
func Convert(in interface{}, out interface{}) error {
byt, _ := json.Marshal(in)
decoder := jsonParser.NewDecoder(bytes.NewReader(byt))
decoder.UseNumber()
err := decoder.Decode(&out)
return err
}
// Recover is used to format error
func Recover(in interface{}) error {
if in == nil {
return nil
}
return errors.New(fmt.Sprint(in))
}
// Deprecated
func Retryable(err error) *bool {
if err == nil {
return Bool(false)
}
if realErr, ok := err.(*SDKError); ok {
if realErr.StatusCode == nil {
return Bool(false)
}
code := IntValue(realErr.StatusCode)
return Bool(code >= http.StatusInternalServerError)
}
return Bool(true)
}
+132
View File
@@ -0,0 +1,132 @@
package tea
import (
"errors"
"testing"
"time"
"github.com/alibabacloud-go/tea/v2/utils"
)
type PrettifyTest struct {
name string
Strs []string
Nums8 []int8
Unum8 []uint8
Value string
Mapvalue map[string]string
}
func Test_Prettify(t *testing.T) {
prettifyTest := &PrettifyTest{
name: "prettify",
Nums8: []int8{0, 1, 2, 4},
Unum8: []uint8{0},
Value: "ok",
Mapvalue: map[string]string{"key": "ccp", "value": "ok"},
}
str := Prettify(prettifyTest)
utils.AssertContains(t, str, "Nums8")
str = Prettify(nil)
utils.AssertEqual(t, str, "null")
}
func Test_Sleep(t *testing.T) {
start := time.Now()
Sleep(Int64(1000))
Sleep(Int64(0))
Sleep(nil)
cost := time.Since(start)
utils.AssertEqual(t, cost.Seconds() >= 1, true)
}
func Test_Merge(t *testing.T) {
in := map[string]*string{
"tea": String("test"),
}
valid := map[string]interface{}{
"valid": "test",
}
invalidStr := "sdfdg"
result := Merge(in, valid, invalidStr)
utils.AssertEqual(t, "test", StringValue(result["tea"]))
utils.AssertEqual(t, "test", StringValue(result["valid"]))
result = Merge(nil)
utils.AssertEqual(t, map[string]*string{}, result)
}
func Test_Convert(t *testing.T) {
in := map[string]interface{}{
"key": "value",
"body": []byte("test"),
}
out := new(test)
err := Convert(in, &out)
utils.AssertNil(t, err)
utils.AssertEqual(t, "value", out.Key)
utils.AssertEqual(t, "test", string(out.Body))
in = map[string]interface{}{
"key": 123,
"body": []byte("test"),
}
out = new(test)
err = Convert(in, &out)
utils.AssertNil(t, err)
utils.AssertEqual(t, "123", out.Key)
utils.AssertEqual(t, "test", string(out.Body))
}
func Test_Recover(t *testing.T) {
err := Recover(nil)
utils.AssertNil(t, err)
defer func() {
if r := Recover(recover()); r != nil {
utils.AssertEqual(t, "test", r.Error())
}
}()
panic("test")
}
func Test_Retryable(t *testing.T) {
ifRetry := Retryable(nil)
utils.AssertEqual(t, false, BoolValue(ifRetry))
err := errors.New("tea")
ifRetry = Retryable(err)
utils.AssertEqual(t, true, BoolValue(ifRetry))
errmsg := map[string]interface{}{
"code": "err",
}
err = NewSDKError(errmsg)
ifRetry = Retryable(err)
utils.AssertEqual(t, false, BoolValue(ifRetry))
errmsg["statusCode"] = 400
err = NewSDKError(errmsg)
ifRetry = Retryable(err)
utils.AssertEqual(t, false, BoolValue(ifRetry))
errmsg["statusCode"] = "400"
err = NewSDKError(errmsg)
ifRetry = Retryable(err)
utils.AssertEqual(t, false, BoolValue(ifRetry))
errmsg["statusCode"] = 500
err = NewSDKError(errmsg)
ifRetry = Retryable(err)
utils.AssertEqual(t, true, BoolValue(ifRetry))
errmsg["statusCode"] = "500"
err = NewSDKError(errmsg)
ifRetry = Retryable(err)
utils.AssertEqual(t, true, BoolValue(ifRetry))
errmsg["statusCode"] = "test"
err = NewSDKError(errmsg)
ifRetry = Retryable(err)
utils.AssertEqual(t, false, BoolValue(ifRetry))
}
+1 -1
View File
@@ -4,7 +4,7 @@ import (
"reflect"
"testing"
"github.com/alibabacloud-go/tea/utils"
"github.com/alibabacloud-go/tea/v2/utils"
jsoniter "github.com/json-iterator/go"
"github.com/modern-go/reflect2"
)
+107
View File
@@ -0,0 +1,107 @@
package tea
import (
"math"
"time"
"github.com/alibabacloud-go/tea/v2/utils"
)
const (
// DefaultMaxAttempts sets maximum number of retries
DefaultMaxAttempts = 3
// DefaultMinDelay sets minimum retry delay
DefaultMinDelay = 100 * time.Millisecond
// DefaultMaxDelayTimeMillis sets maximum retry delay
DefaultMaxDelay = 120 * time.Second
)
type RetryCondition struct {
MaxAttempts *int
MaxDelayTimeMillis *int64
Backoff *BackoffPolicy
Exception []*string
ErrorCode []*string
}
type RetryOptions struct {
Retryable *bool
RetryCondition []*RetryCondition
NoRetryCondition []*RetryCondition
}
type RetryPolicyContext struct {
RetriesAttempted *int
Request *Request
Response *Response
Error error
}
func ShouldRetry(options *RetryOptions, ctx *RetryPolicyContext) *bool {
if IntValue(ctx.RetriesAttempted) == 0 {
return Bool(true)
}
if options == nil || !BoolValue(options.Retryable) {
return Bool(false)
}
if err, ok := ctx.Error.(BaseError); ok {
noRetryConditions := options.NoRetryCondition
retryConditions := options.RetryCondition
if noRetryConditions != nil {
for _, noRetryCondition := range noRetryConditions {
if utils.Contains(noRetryCondition.Exception, err.ErrorName()) || utils.Contains(noRetryCondition.ErrorCode, err.ErrorCode()) {
return Bool(false)
}
}
}
if retryConditions != nil {
for _, retryCondition := range retryConditions {
if !utils.Contains(retryCondition.Exception, err.ErrorName()) && !utils.Contains(retryCondition.ErrorCode, err.ErrorCode()) {
continue
}
if IntValue(ctx.RetriesAttempted) > IntValue(retryCondition.MaxAttempts) {
return Bool(false)
}
return Bool(true)
}
}
}
return Bool(false)
}
func GetBackoffDelay(options *RetryOptions, ctx *RetryPolicyContext) *int64 {
if IntValue(ctx.RetriesAttempted) == 0 {
return Int64(0)
}
if err, ok := ctx.Error.(BaseError); ok {
if options != nil {
retryConditions := options.RetryCondition
if retryConditions != nil {
for _, retryCondition := range retryConditions {
if !utils.Contains(retryCondition.Exception, err.ErrorName()) && !utils.Contains(retryCondition.ErrorCode, err.ErrorCode()) {
continue
}
var maxDelay int64
if retryCondition.MaxDelayTimeMillis != nil {
maxDelay = Int64Value(retryCondition.MaxDelayTimeMillis)
} else {
maxDelay = DefaultMaxDelay.Milliseconds()
}
if err.RetryAfterTimeMillis() != nil {
return Int64(int64(math.Min(float64(Int64Value(err.RetryAfterTimeMillis())), float64(maxDelay))))
}
if retryCondition.Backoff == nil {
return Int64(DefaultMinDelay.Milliseconds())
}
delayTimeMillis := (*retryCondition.Backoff).GetDelayTimeMillis(ctx)
return Int64(int64(math.Min(float64(Int64Value(delayTimeMillis)), float64(maxDelay))))
}
}
}
}
return Int64(DefaultMinDelay.Milliseconds())
}
+536
View File
File diff suppressed because it is too large Load Diff
+75
View File
@@ -1,5 +1,12 @@
package tea
import (
"encoding/json"
"fmt"
"io"
"strings"
)
func String(a string) *string {
return &a
}
@@ -489,3 +496,71 @@ func BoolSliceValue(a []*bool) []bool {
}
return res
}
func TransInterfaceToBool(val interface{}) *bool {
if val == nil {
return nil
}
return Bool(val.(bool))
}
func TransInterfaceToInt(val interface{}) *int {
if val == nil {
return nil
}
return Int(val.(int))
}
func TransInterfaceToInt64(val interface{}) *int64 {
if val == nil {
return nil
}
return Int64(val.(int64))
}
func TransInterfaceToString(val interface{}) *string {
if val == nil {
return nil
}
return String(val.(string))
}
func ToInt(a *int32) *int {
return Int(int(Int32Value(a)))
}
func ToInt32(a *int) *int32 {
return Int32(int32(IntValue(a)))
}
func ToString(val interface{}) string {
return fmt.Sprintf("%v", val)
}
func ToObject(obj interface{}) map[string]interface{} {
result := make(map[string]interface{})
byt, _ := json.Marshal(obj)
err := json.Unmarshal(byt, &result)
if err != nil {
return nil
}
return result
}
func ToReader(obj interface{}) io.Reader {
switch obj.(type) {
case *string:
tmp := obj.(*string)
return strings.NewReader(StringValue(tmp))
case []byte:
return strings.NewReader(string(obj.([]byte)))
case io.Reader:
return obj.(io.Reader)
default:
panic("Invalid Body. Please set a valid Body.")
}
}
+85 -1
View File
@@ -1,9 +1,11 @@
package tea
import (
"io/ioutil"
"strings"
"testing"
"github.com/alibabacloud-go/tea/utils"
"github.com/alibabacloud-go/tea/v2/utils"
)
func Test_Trans(t *testing.T) {
@@ -161,3 +163,85 @@ func Test_Trans(t *testing.T) {
utils.AssertNil(t, Uint64Slice(nil))
utils.AssertNil(t, Uint64ValueSlice(nil))
}
func Test_TransInterfaceToInt(t *testing.T) {
a := TransInterfaceToInt(nil)
utils.AssertNil(t, a)
a = TransInterfaceToInt(10)
utils.AssertEqual(t, IntValue(a), 10)
}
func Test_TransInterfaceToInt64(t *testing.T) {
a := TransInterfaceToInt64(nil)
utils.AssertNil(t, a)
a = TransInterfaceToInt64(int64(10))
utils.AssertEqual(t, Int64Value(a), int64(10))
}
func Test_TransInterfaceToString(t *testing.T) {
a := TransInterfaceToString(nil)
utils.AssertNil(t, a)
a = TransInterfaceToString("10")
utils.AssertEqual(t, StringValue(a), "10")
}
func Test_TransInt32AndInt(t *testing.T) {
a := ToInt(Int32(10))
utils.AssertEqual(t, IntValue(a), 10)
b := ToInt32(a)
utils.AssertEqual(t, Int32Value(b), int32(10))
}
func Test_ToString(t *testing.T) {
str := ToString(10)
utils.AssertEqual(t, "10", str)
str = ToString("10")
utils.AssertEqual(t, "10", str)
}
func Test_ToObject(t *testing.T) {
str := "{sdsfdsd:"
result := ToObject(str)
utils.AssertNil(t, result)
input := map[string]string{
"name": "test",
}
result = ToObject(input)
utils.AssertEqual(t, "test", result["name"].(string))
}
func Test_ToReader(t *testing.T) {
str := "abc"
reader := ToReader(String(str))
byt, err := ioutil.ReadAll(reader)
utils.AssertNil(t, err)
utils.AssertEqual(t, "abc", string(byt))
read := strings.NewReader("bcd")
reader = ToReader(read)
byt, err = ioutil.ReadAll(reader)
utils.AssertNil(t, err)
utils.AssertEqual(t, "bcd", string(byt))
byts := []byte("cdf")
reader = ToReader(byts)
byt, err = ioutil.ReadAll(reader)
utils.AssertNil(t, err)
utils.AssertEqual(t, "cdf", string(byt))
num := 10
defer func() {
err := recover()
utils.AssertEqual(t, "Invalid Body. Please set a valid Body.", err.(string))
}()
reader = ToReader(num)
byt, err = ioutil.ReadAll(reader)
utils.AssertNil(t, err)
utils.AssertEqual(t, "", string(byt))
}
+13
View File
@@ -0,0 +1,13 @@
package utils
func Contains(s []*string, str *string) bool {
if s == nil {
return false
}
for _, v := range s {
if str != nil && v != nil && *v == *str {
return true
}
}
return false
}
+21
View File
@@ -0,0 +1,21 @@
package utils
import (
"testing"
)
func Test_Contains(t *testing.T) {
apple := "apple"
banana := "banana"
cherry := "cherry"
slice := []*string{&apple, &banana, &cherry, nil}
AssertEqual(t, true, Contains(slice, &banana))
toFind := "banana"
AssertEqual(t, true, Contains(slice, &toFind))
notFind := "notFind"
AssertEqual(t, false, Contains(slice, &notFind))
notFind = ""
AssertEqual(t, false, Contains(slice, &notFind))
AssertEqual(t, false, Contains(slice, nil))
AssertEqual(t, false, Contains(nil, nil))
}