Merge pull request #492 from samwafgo/feat_batch_copy_cache

feat:batch copy cache config

api/waf_host.go

@@ -61,6 +61,41 @@ func (w *WafHostAPi) AddApi(c *gin.Context) {
 		response.FailWithMessage("解析失败", c)
 	}
 }
+
+// BatchCopyConfigApi 批量复制配置API
+func (w *WafHostAPi) BatchCopyConfigApi(c *gin.Context) {
+	var req request.WafHostBatchCopyConfigReq
+	err := c.ShouldBindJSON(&req)
+	if err == nil {
+		// 验证源主机是否存在
+		sourceHost := wafHostService.GetDetailByCodeApi(req.SourceHostCode)
+		if sourceHost.Code == "" {
+			response.FailWithMessage("源主机不存在", c)
+			return
+		}
+
+		// 验证目标主机是否存在
+		targetHost := wafHostService.GetDetailByCodeApi(req.TargetHostCode)
+		if targetHost.Code == "" {
+			response.FailWithMessage("目标主机 "+req.TargetHostCode+" 不存在", c)
+			return
+		}
+
+		// 执行配置复制
+		err = wafHostService.CopyConfigApi(req)
+		if err != nil {
+			response.FailWithMessage("复制配置失败: "+err.Error(), c)
+			return
+		}
+
+		// 通知WAF引擎更新配置
+		global.GWAF_CHAN_HOST <- targetHost
+
+		response.OkWithMessage("复制配置成功", c)
+	} else {
+		response.FailWithMessage("解析失败", c)
+	}
+}
 func (w *WafHostAPi) GetDetailApi(c *gin.Context) {
 	var req request.WafHostDetailReq
 	err := c.ShouldBind(&req)

model/request/waf_host_req.go

@@ -110,3 +110,10 @@ type WafHostAllDomainsReq struct {
 type WafHostBatchGuardStatusReq struct {
 	GUARD_STATUS int `json:"guard_status"` // 防御状态：0-关闭防御，1-开启防御
 }
+
+// WafHostBatchCopyConfigReq 批量复制配置请求
+type WafHostBatchCopyConfigReq struct {
+	SourceHostCode string   `json:"source_host_code"` // 源主机代码
+	TargetHostCode string   `json:"target_host_code"` // 目标主机代码
+	Modules        []string `json:"modules"`          // 要复制的模块列表，如：["cache"]
+}

router/waf_host.go

@@ -21,5 +21,6 @@ func (receiver *HostRouter) InitHostRouter(group *gin.RouterGroup) {
 	hostRouter.GET("/samwaf/wafhost/host/allhost", hostApi.GetAllListApi)
 	hostRouter.GET("/samwaf/wafhost/host/alldomainbyhostcode", hostApi.GetDomainsByHostCodeApi)
 	hostRouter.POST("/samwaf/wafhost/host/modfiyallstatus", hostApi.ModifyAllGuardStatusApi)
+	hostRouter.POST("/samwaf/wafhost/host/batchcopyconfig", hostApi.BatchCopyConfigApi)
 
 }

service/waf_service/waf_host.go

@@ -8,7 +8,9 @@ import (
 	"SamWaf/model"
 	"SamWaf/model/baseorm"
 	"SamWaf/model/request"
+	"encoding/json"
 	"errors"
+	"gorm.io/gorm"
 	"strings"
 	"time"
 )
@@ -372,6 +374,139 @@ func (receiver *WafHostService) ModifyAllGuardStatusApi(req request.WafHostBatch
 	return err
 }
 
+// CopyConfigApi 复制配置
+func (receiver *WafHostService) CopyConfigApi(req request.WafHostBatchCopyConfigReq) error {
+	// 获取源主机信息
+	sourceHost := receiver.GetDetailByCodeApi(req.SourceHostCode)
+	if sourceHost.Code == "" {
+		return errors.New("源主机不存在")
+	}
+
+	// 获取目标主机信息
+	targetHost := receiver.GetDetailByCodeApi(req.TargetHostCode)
+	if targetHost.Code == "" {
+		return errors.New("目标主机不存在")
+	}
+
+	// 开始事务
+	tx := global.GWAF_LOCAL_DB.Begin()
+	defer func() {
+		if r := recover(); r != nil {
+			tx.Rollback()
+		}
+	}()
+
+	// 遍历要复制的模块
+	for _, module := range req.Modules {
+		switch module {
+		case "cache":
+			// 复制缓存配置
+			err := receiver.copyCacheConfig(tx, sourceHost, targetHost)
+			if err != nil {
+				tx.Rollback()
+				return errors.New("复制缓存配置失败: " + err.Error())
+			}
+		// 可以在这里添加其他模块的复制逻辑
+		// case "defense":
+		//     err := receiver.copyDefenseConfig(tx, sourceHost, targetHost)
+		// case "ssl":
+		//     err := receiver.copySSLConfig(tx, sourceHost, targetHost)
+		default:
+			tx.Rollback()
+			return errors.New("不支持的模块: " + module)
+		}
+	}
+
+	// 提交事务
+	return tx.Commit().Error
+}
+
+// copyCacheConfig 复制缓存配置
+func (receiver *WafHostService) copyCacheConfig(tx *gorm.DB, sourceHost, targetHost model.Hosts) error {
+	// 解析源主机的缓存配置
+	var sourceCacheConfig model.CacheConfig
+	if sourceHost.CacheJSON != "" {
+		err := json.Unmarshal([]byte(sourceHost.CacheJSON), &sourceCacheConfig)
+		if err != nil {
+			return errors.New("解析源主机缓存配置失败: " + err.Error())
+		}
+	}
+
+	// 将缓存配置应用到目标主机
+	cacheJSON, err := json.Marshal(sourceCacheConfig)
+	if err != nil {
+		return errors.New("序列化缓存配置失败: " + err.Error())
+	}
+
+	// 更新目标主机的缓存配置和开启状态
+	updateMap := map[string]interface{}{
+		"CacheJSON":   string(cacheJSON),
+		"UPDATE_TIME": customtype.JsonTime(time.Now()),
+	}
+
+	err = tx.Model(&model.Hosts{}).Where("CODE = ?", targetHost.Code).Updates(updateMap).Error
+	if err != nil {
+		return errors.New("更新目标主机缓存配置失败: " + err.Error())
+	}
+
+	// 复制缓存规则
+	err = receiver.copyCacheRules(tx, sourceHost.Code, targetHost.Code)
+	if err != nil {
+		return errors.New("复制缓存规则失败: " + err.Error())
+	}
+
+	return nil
+}
+
+// copyCacheRules 复制缓存规则
+func (receiver *WafHostService) copyCacheRules(tx *gorm.DB, sourceHostCode, targetHostCode string) error {
+	// 查询源主机的所有缓存规则
+	var sourceCacheRules []model.CacheRule
+	err := tx.Where("host_code = ?", sourceHostCode).Find(&sourceCacheRules).Error
+	if err != nil {
+		return errors.New("查询源主机缓存规则失败: " + err.Error())
+	}
+
+	// 如果源主机没有缓存规则，直接返回
+	if len(sourceCacheRules) == 0 {
+		return nil
+	}
+
+	// 先删除目标主机的现有缓存规则
+	err = tx.Where("host_code = ?", targetHostCode).Delete(&model.CacheRule{}).Error
+	if err != nil {
+		return errors.New("删除目标主机现有缓存规则失败: " + err.Error())
+	}
+
+	// 复制缓存规则到目标主机
+	for _, rule := range sourceCacheRules {
+		newRule := model.CacheRule{
+			BaseOrm:       rule.BaseOrm,
+			HostCode:      targetHostCode, // 更改为目标主机代码
+			RuleName:      rule.RuleName,
+			RuleType:      rule.RuleType,
+			RuleContent:   rule.RuleContent,
+			ParamType:     rule.ParamType,
+			CacheTime:     rule.CacheTime,
+			Priority:      rule.Priority,
+			RequestMethod: rule.RequestMethod,
+			Remarks:       rule.Remarks,
+		}
+		// 生成新的ID和时间戳
+		newRule.Id = uuid.GenUUID()
+		newRule.USER_CODE = global.GWAF_USER_CODE
+		newRule.CREATE_TIME = customtype.JsonTime(time.Now())
+		newRule.UPDATE_TIME = customtype.JsonTime(time.Now())
+
+		err = tx.Create(&newRule).Error
+		if err != nil {
+			return errors.New("创建缓存规则失败: " + err.Error())
+		}
+	}
+
+	return nil
+}
+
 // GetHostsByGuardStatus 获取指定防御状态的主机
 func (receiver *WafHostService) GetHostsByGuardStatus(guardStatus int) []model.Hosts {
 	var webHosts []model.Hosts