add

Author: SevenEarth

tencentcloud/services/waf/service_tencentcloud_waf.go

@@ -25,6 +25,7 @@ func (me *WafService) DescribeWafCustomRuleById(ctx context.Context, domain, rul
 	logId := tccommon.GetLogId(ctx)
 
 	request := waf.NewDescribeCustomRuleListRequest()
+	response := waf.NewDescribeCustomRuleListResponse()
 	request.Domain = &domain
 	request.Offset = common.Uint64Ptr(0)
 	request.Limit = common.Uint64Ptr(20)
@@ -42,16 +43,24 @@ func (me *WafService) DescribeWafCustomRuleById(ctx context.Context, domain, rul
 		}
 	}()
 
-	ratelimit.Check(request.GetAction())
+	err := resource.Retry(tccommon.ReadRetryTimeout, func() *resource.RetryError {
+		ratelimit.Check(request.GetAction())
+		result, e := me.client.UseWafV20180125Client().DescribeCustomRuleList(request)
+		if e != nil {
+			return tccommon.RetryError(e)
+		} else {
+			log.Printf("[DEBUG]%s api[%s] success, request body [%s], response body [%s]\n", logId, request.GetAction(), request.ToJsonString(), result.ToJsonString())
+		}
+
+		response = result
+		return nil
+	})
 
-	response, err := me.client.UseWafClient().DescribeCustomRuleList(request)
 	if err != nil {
 		errRet = err
 		return
 	}
 
-	log.Printf("[DEBUG]%s api[%s] success, request body [%s], response body [%s]\n", logId, request.GetAction(), request.ToJsonString(), response.ToJsonString())
-
 	if len(response.Response.RuleList) < 1 {
 		return
 	}
@@ -73,16 +82,23 @@ func (me *WafService) DeleteWafCustomRuleById(ctx context.Context, domain, ruleI
 		}
 	}()
 
-	ratelimit.Check(request.GetAction())
+	err := resource.Retry(tccommon.WriteRetryTimeout, func() *resource.RetryError {
+		ratelimit.Check(request.GetAction())
+		result, e := me.client.UseWafV20180125Client().DeleteCustomRule(request)
+		if e != nil {
+			return tccommon.RetryError(e)
+		} else {
+			log.Printf("[DEBUG]%s api[%s] success, request body [%s], response body [%s]\n", logId, request.GetAction(), request.ToJsonString(), result.ToJsonString())
+		}
+
+		return nil
+	})
 
-	response, err := me.client.UseWafClient().DeleteCustomRule(request)
 	if err != nil {
 		errRet = err
 		return
 	}
 
-	log.Printf("[DEBUG]%s api[%s] success, request body [%s], response body [%s]\n", logId, request.GetAction(), request.ToJsonString(), response.ToJsonString())
-
 	return
 }