feat: support compute throttling time left

Author: yndu13

service/service.go

@@ -633,3 +633,40 @@ func shaHmac1(source, secret string) []byte {
 	hmac.Write([]byte(source))
 	return hmac.Sum(nil)
 }
+
+func getTimeLeft(rateLimit *string) (_result *int64) {
+	if rateLimit != nil {
+		pairs := strings.Split(tea.StringValue(rateLimit), ",")
+		for _, pair := range pairs {
+			kv := strings.Split(pair, ":")
+			if len(kv) == 2 {
+				key, value := kv[0], kv[1]
+				if key == "TimeLeft" {
+					timeLeftValue, err := strconv.ParseInt(value, 10, 64)
+					if err != nil {
+						return nil
+					}
+					return tea.Int64(timeLeftValue)
+				}
+			}
+		}
+	}
+	return nil
+}
+
+/**
+ * Get throttling param
+ * @param the response headers
+ * @return time left
+ */
+func GetThrottlingTimeLeft(headers map[string]*string) (_result *int64) {
+	rateLimitForUserApi := headers["X-RateLimit-User-API"]
+	rateLimitForUser := headers["X-RateLimit-User"]
+	timeLeftForUserApi := getTimeLeft(rateLimitForUserApi)
+	timeLeftForUser := getTimeLeft(rateLimitForUser)
+	if tea.Int64Value(timeLeftForUserApi) > tea.Int64Value(timeLeftForUser) {
+		return timeLeftForUserApi
+	} else {
+		return timeLeftForUser
+	}
+}

service/service_test.go

@@ -255,3 +255,40 @@ func Test_SignatureMethod(t *testing.T) {
 	res = SignatureMethod(priKey, "source", "ACS3-RSA-SHA256")
 	utils.AssertEqual(t, "a00b88ae04f651a8ab645e724949ff435bbb2cf9a37aa54323024477f8031f4e13dc948484c5c5a81ba53a55eb0571dffccc1e953c93269d6da23ed319e0f1ef699bcc9823a646574628ae1b70ed569b5a07d139dda28996b5b9231f5ba96141f0893deec2fbf54a0fa2c203b8ae74dd26f457ac29c873745a5b88273d2b3d12", tea.StringValue(HexEncode(res)))
 }
+
+func Test_GetThrottlingTimeLeft(t *testing.T) {
+	headers := map[string]*string{
+		"X-RateLimit-User-API": nil,
+		"X-RateLimit-User":     nil,
+	}
+	timeLeft := GetThrottlingTimeLeft(headers)
+	utils.AssertNil(t, timeLeft)
+
+	headers = map[string]*string{
+		"X-RateLimit-User-API": nil,
+		"X-RateLimit-User":     tea.String("Limit:1,Remain:0,TimeLeft:2000,Reset:1234"),
+	}
+	timeLeft = GetThrottlingTimeLeft(headers)
+	utils.AssertEqual(t, int64(2000), tea.Int64Value(timeLeft))
+
+	headers = map[string]*string{
+		"X-RateLimit-User-API": tea.String("Limit:1,Remain:0,TimeLeft:2000,Reset:1234"),
+		"X-RateLimit-User":     nil,
+	}
+	timeLeft = GetThrottlingTimeLeft(headers)
+	utils.AssertEqual(t, int64(2000), tea.Int64Value(timeLeft))
+
+	headers = map[string]*string{
+		"X-RateLimit-User-API": tea.String("Limit:1,Remain:0,TimeLeft:2000,Reset:1234"),
+		"X-RateLimit-User":     tea.String("Limit:1,Remain:0,TimeLeft:0,Reset:1234"),
+	}
+	timeLeft = GetThrottlingTimeLeft(headers)
+	utils.AssertEqual(t, int64(2000), tea.Int64Value(timeLeft))
+
+	headers = map[string]*string{
+		"X-RateLimit-User-API": tea.String("Limit:1,Remain:0,TimeLeft:0,Reset:1234"),
+		"X-RateLimit-User":     tea.String("Limit:1,Remain:0,TimeLeft:0,Reset:1234"),
+	}
+	timeLeft = GetThrottlingTimeLeft(headers)
+	utils.AssertEqual(t, int64(0), tea.Int64Value(timeLeft))
+}