mirror of
https://gitee.com/samwaf/SamWaf.git
synced 2025-12-06 06:58:54 +08:00
84 lines
2.6 KiB
Go
84 lines
2.6 KiB
Go
package waf_service
|
|
|
|
import (
|
|
"SamWaf/customtype"
|
|
"SamWaf/global"
|
|
"SamWaf/model"
|
|
"SamWaf/model/baseorm"
|
|
"SamWaf/model/request"
|
|
"errors"
|
|
uuid "github.com/satori/go.uuid"
|
|
"time"
|
|
)
|
|
|
|
type WafLdpUrlService struct{}
|
|
|
|
var WafLdpUrlServiceApp = new(WafLdpUrlService)
|
|
|
|
func (receiver *WafLdpUrlService) AddApi(req request.WafLdpUrlAddReq) error {
|
|
var bean = &model.LDPUrl{
|
|
BaseOrm: baseorm.BaseOrm{
|
|
Id: uuid.NewV4().String(),
|
|
USER_CODE: global.GWAF_USER_CODE,
|
|
Tenant_ID: global.GWAF_TENANT_ID,
|
|
CREATE_TIME: customtype.JsonTime(time.Now()),
|
|
UPDATE_TIME: customtype.JsonTime(time.Now()),
|
|
},
|
|
HostCode: req.HostCode,
|
|
CompareType: req.CompareType,
|
|
Url: req.Url,
|
|
Remarks: req.Remarks,
|
|
}
|
|
global.GWAF_LOCAL_DB.Create(bean)
|
|
return nil
|
|
}
|
|
|
|
func (receiver *WafLdpUrlService) CheckIsExistApi(req request.WafLdpUrlAddReq) error {
|
|
return global.GWAF_LOCAL_DB.First(&model.LDPUrl{}, "host_code = ? and url= ?", req.HostCode,
|
|
req.Url).Error
|
|
}
|
|
func (receiver *WafLdpUrlService) ModifyApi(req request.WafLdpUrlEditReq) error {
|
|
var ipWhite model.LDPUrl
|
|
global.GWAF_LOCAL_DB.Where("host_code = ? and url= ?", req.HostCode,
|
|
req.Url).Find(&ipWhite)
|
|
if ipWhite.Id != "" && ipWhite.Url != req.Url {
|
|
return errors.New("当前网站和url已经存在")
|
|
}
|
|
ipWhiteMap := map[string]interface{}{
|
|
"Host_Code": req.HostCode,
|
|
"Url": req.Url,
|
|
"Remarks": req.Remarks,
|
|
"Compare_Type": req.CompareType,
|
|
"UPDATE_TIME": customtype.JsonTime(time.Now()),
|
|
}
|
|
err := global.GWAF_LOCAL_DB.Model(model.LDPUrl{}).Where("id = ?", req.Id).Updates(ipWhiteMap).Error
|
|
|
|
return err
|
|
}
|
|
func (receiver *WafLdpUrlService) GetDetailApi(req request.WafLdpUrlDetailReq) model.LDPUrl {
|
|
var bean model.LDPUrl
|
|
global.GWAF_LOCAL_DB.Where("id=?", req.Id).Find(&bean)
|
|
return bean
|
|
}
|
|
func (receiver *WafLdpUrlService) GetDetailByIdApi(id string) model.LDPUrl {
|
|
var bean model.LDPUrl
|
|
global.GWAF_LOCAL_DB.Where("id=?", id).Find(&bean)
|
|
return bean
|
|
}
|
|
func (receiver *WafLdpUrlService) GetListApi(req request.WafLdpUrlSearchReq) ([]model.LDPUrl, int64, error) {
|
|
var ipWhites []model.LDPUrl
|
|
var total int64 = 0
|
|
global.GWAF_LOCAL_DB.Limit(req.PageSize).Offset(req.PageSize * (req.PageIndex - 1)).Find(&ipWhites)
|
|
global.GWAF_LOCAL_DB.Model(&model.LDPUrl{}).Count(&total)
|
|
return ipWhites, total, nil
|
|
}
|
|
func (receiver *WafLdpUrlService) DelApi(req request.WafLdpUrlDelReq) error {
|
|
var ipWhite model.LDPUrl
|
|
err := global.GWAF_LOCAL_DB.Where("id = ?", req.Id).First(&ipWhite).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = global.GWAF_LOCAL_DB.Where("id = ?", req.Id).Delete(model.LDPUrl{}).Error
|
|
return err
|
|
}
|