TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2026-01-10 01:45:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加为WAF分组添加规则集的API

Browse Source
This commit is contained in:
GoEdgeLab
2021-10-25 12:01:16 +08:00
parent a241186633
commit b742f4d93c
4 changed files with 99 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/setup/sql_upgrade.go
View File

@@ -56,6 +56,9 @@ var upgradeFuncs = []*upgradeVersion{
{
"0.3.2", upgradeV0_3_2,
},
{
"0.3.3", upgradeV0_3_3,
},
}
// UpgradeSQLData 升级SQL数据
@@ -511,3 +514,14 @@ func upgradeV0_3_2(db *dbs.DB) error {
return nil
}
// v0.3.3
func upgradeV0_3_3(db *dbs.DB) error {
// 升级CC请求数Code
_, err := db.Exec("UPDATE edgeHTTPFirewallRuleSets SET code='8002' WHERE name='CC请求数' AND code='8001'")
if err != nil {
return err
}
return nil
}

20
internal/setup/sql_upgrade_test.go
View File

@@ -22,7 +22,7 @@ func TestUpgradeSQLData(t *testing.T) {
}
func TestUpgradeSQLData_v1_3_1(t *testing.T) {
func TestUpgradeSQLData_v0_3_1(t *testing.T) {
db, err := dbs.NewInstanceFromConfig(&dbs.DBConfig{
Driver: "mysql",
Dsn: "root:123456@tcp(127.0.0.1:3306)/db_edge_new?charset=utf8mb4&timeout=30s",
@@ -38,7 +38,7 @@ func TestUpgradeSQLData_v1_3_1(t *testing.T) {
t.Log("ok")
}
func TestUpgradeSQLData_v1_3_2(t *testing.T) {
func TestUpgradeSQLData_v0_3_2(t *testing.T) {
db, err := dbs.NewInstanceFromConfig(&dbs.DBConfig{
Driver: "mysql",
Dsn: "root:123456@tcp(127.0.0.1:3306)/db_edge?charset=utf8mb4&timeout=30s",
@@ -52,4 +52,20 @@ func TestUpgradeSQLData_v1_3_2(t *testing.T) {
t.Fatal(err)
}
t.Log("ok")
}
func TestUpgradeSQLData_v0_3_3(t *testing.T) {
db, err := dbs.NewInstanceFromConfig(&dbs.DBConfig{
Driver: "mysql",
Dsn: "root:123456@tcp(127.0.0.1:3306)/db_edge?charset=utf8mb4&timeout=30s",
Prefix: "edge",
})
if err != nil {
t.Fatal(err)
}
err = upgradeV0_3_3(db)
if err != nil {
t.Fatal(err)
}
t.Log("ok")
}