diff --git a/backend/internal/cert/deploy/btwaf.go b/backend/internal/cert/deploy/btwaf.go index 01594bf..6fc56c2 100644 --- a/backend/internal/cert/deploy/btwaf.go +++ b/backend/internal/cert/deploy/btwaf.go @@ -87,13 +87,13 @@ func RequestBtWaf(data *map[string]any, method, providerID, requestUrl string) ( return res, nil } -func GetBTWafSiteList(page int, pageSize int, siteName string) ([]any, error) { +func GetBTWafSiteList(page int, pageSize int, siteName string, providerId string) ([]any, error) { data := map[string]any{ "p": page, "p_size": pageSize, "site_name": siteName, } - response, err := RequestBtWaf(&data, "POST", "1", "api/wafmastersite/get_site_list") + response, err := RequestBtWaf(&data, "POST", providerId, "api/wafmastersite/get_site_list") res := response["res"].(map[string]any) if err != nil { return nil, err @@ -102,7 +102,7 @@ func GetBTWafSiteList(page int, pageSize int, siteName string) ([]any, error) { return res["list"].([]any), nil } -//btwaf不支持通过API设置SSL +// btwaf不支持通过API设置SSL func DeployBtWaf(cfg map[string]any) error { cert, ok := cfg["certificate"].(map[string]any) if !ok { @@ -166,7 +166,7 @@ func DeployBtWafSite(cfg map[string]any) error { return fmt.Errorf("参数错误:siteName") } - sitelist, err := GetBTWafSiteList(1, 10, siteName) + sitelist, err := GetBTWafSiteList(1, 10, siteName, providerID) if len(sitelist) == 0 || err != nil { return fmt.Errorf("找不到网站:%s", siteName) } diff --git a/backend/internal/cert/deploy/btwaf_test.go b/backend/internal/cert/deploy/btwaf_test.go index 7ac1b08..463f304 100644 --- a/backend/internal/cert/deploy/btwaf_test.go +++ b/backend/internal/cert/deploy/btwaf_test.go @@ -33,7 +33,7 @@ func TestBTWAFSite(t *testing.T) { //} func TestGetBTWAFSiteList(t *testing.T) { - res, err := GetBTWafSiteList(1, 10, "xxx.cn") + res, err := GetBTWafSiteList(1, 10, "xxx.cn", "1") for _, site := range res { sites := site.(map[string]any) fmt.Printf("网站名: %s ID:%s \n", sites["site_name"], sites["site_id"])