SamWaf/service/waf_service/waf_privateinfo_service.go

package waf_service

import (
	"SamWaf/common/uuid"
	"SamWaf/customtype"
	"SamWaf/global"
	"SamWaf/model"
	"SamWaf/model/baseorm"
	"SamWaf/model/request"
	"errors"
	"time"
)

type WafPrivateInfoService struct{}

var WafPrivateInfoServiceApp = new(WafPrivateInfoService)

func (receiver *WafPrivateInfoService) AddApi(req request.WafPrivateInfoAddReq) error {
	var bean = &model.PrivateInfo{
		BaseOrm: baseorm.BaseOrm{
			Id:          uuid.GenUUID(),
			USER_CODE:   global.GWAF_USER_CODE,
			Tenant_ID:   global.GWAF_TENANT_ID,
			CREATE_TIME: customtype.JsonTime(time.Now()),
			UPDATE_TIME: customtype.JsonTime(time.Now()),
		},

		PrivateKey:   req.PrivateKey,
		PrivateValue: req.PrivateValue,
		Remarks:      req.Remarks,
	}
	global.GWAF_LOCAL_DB.Create(bean)
	return nil
}

func (receiver *WafPrivateInfoService) CheckIsExistApi(req request.WafPrivateInfoAddReq) int {
	var total int64 = 0
	/*where条件*/
	var whereField = ""
	var whereValues []interface{}
	//where字段
	whereField = ""

	if len(req.PrivateKey) > 0 {
		if len(whereField) > 0 {
			whereField = whereField + " and "
		}
		whereField = whereField + " private_key=? "
	}

	//where字段赋值

	if len(req.PrivateKey) > 0 {
		if len(whereField) > 0 {
			whereValues = append(whereValues, req.PrivateKey)
		}
	}

	global.GWAF_LOCAL_DB.Model(&model.PrivateInfo{}).Where(whereField, whereValues...).Count(&total)
	return int(total)
}

func (receiver *WafPrivateInfoService) ModifyWithOutValueApi(req request.WafPrivateInfoEditReq) error {
	// 根据唯一字段生成查询条件（只有在UniFields不为空时才进行存在性检查）
	var total int64 = 0
	/*where条件*/
	var whereField = ""
	var whereValues []interface{}
	//where字段
	whereField = ""

	if len(req.PrivateKey) > 0 {
		if len(whereField) > 0 {
			whereField = whereField + " and "
		}
		whereField = whereField + " private_key=? "
	}

	//where字段赋值

	if len(req.PrivateKey) > 0 {
		whereValues = append(whereValues, req.PrivateKey)
	}

	global.GWAF_LOCAL_DB.Model(&model.PrivateInfo{}).Where(whereField, whereValues...).Count(&total)
	// 查询是否已存在记录
	var bean model.PrivateInfo
	global.GWAF_LOCAL_DB.Model(&model.PrivateInfo{}).Where(whereField, whereValues...).Limit(1).Find(&bean)

	if int(total) > 0 && bean.PrivateKey != "" && bean.PrivateKey != req.PrivateKey {
		return errors.New("当前记录已经存在")
	}

	beanMap := map[string]interface{}{
		"PrivateKey": req.PrivateKey,
		"Remarks":    req.Remarks,

		"UPDATE_TIME": customtype.JsonTime(time.Now()),
	}
	err := global.GWAF_LOCAL_DB.Model(model.PrivateInfo{}).Where("private_key = ?", req.PrivateKey).Updates(beanMap).Error

	return err
}

func (receiver *WafPrivateInfoService) ModifyApi(req request.WafPrivateInfoEditReq) error {
	// 根据唯一字段生成查询条件（只有在UniFields不为空时才进行存在性检查）

	var total int64 = 0
	/*where条件*/
	var whereField = ""
	var whereValues []interface{}
	//where字段
	whereField = ""

	if len(req.PrivateKey) > 0 {
		if len(whereField) > 0 {
			whereField = whereField + " and "
		}
		whereField = whereField + " private_key=? "
	}

	//where字段赋值

	if len(req.PrivateKey) > 0 {
		whereValues = append(whereValues, req.PrivateKey)
	}

	global.GWAF_LOCAL_DB.Model(&model.PrivateInfo{}).Where(whereField, whereValues...).Count(&total)
	// 查询是否已存在记录
	var bean model.PrivateInfo
	global.GWAF_LOCAL_DB.Model(&model.PrivateInfo{}).Where(whereField, whereValues...).Limit(1).Find(&bean)

	if int(total) > 0 && bean.PrivateKey != "" && bean.PrivateKey != req.PrivateKey {
		return errors.New("当前记录已经存在")
	}

	beanMap := map[string]interface{}{

		"PrivateKey":   req.PrivateKey,
		"PrivateValue": req.PrivateValue,
		"Remarks":      req.Remarks,

		"UPDATE_TIME": customtype.JsonTime(time.Now()),
	}
	err := global.GWAF_LOCAL_DB.Model(model.PrivateInfo{}).Where("private_key = ?", req.PrivateKey).Updates(beanMap).Error

	return err
}
func (receiver *WafPrivateInfoService) GetDetailApi(req request.WafPrivateInfoDetailReq) model.PrivateInfo {
	var bean model.PrivateInfo
	global.GWAF_LOCAL_DB.Where("id=?", req.Id).Find(&bean)
	return bean
}
func (receiver *WafPrivateInfoService) GetDetailByIdApi(id string) model.PrivateInfo {
	var bean model.PrivateInfo
	global.GWAF_LOCAL_DB.Where("id=?", id).Find(&bean)
	return bean
}
func (receiver *WafPrivateInfoService) GetListApi(req request.WafPrivateInfoSearchReq) ([]model.PrivateInfo, int64, error) {
	var list []model.PrivateInfo
	var total int64 = 0
	global.GWAF_LOCAL_DB.Model(&model.PrivateInfo{}).Limit(req.PageSize).Offset(req.PageSize * (req.PageIndex - 1)).Find(&list)
	global.GWAF_LOCAL_DB.Model(&model.PrivateInfo{}).Count(&total)

	return list, total, nil
}
func (receiver *WafPrivateInfoService) DelApi(req request.WafPrivateInfoDelReq) error {
	var bean model.PrivateInfo
	err := global.GWAF_LOCAL_DB.Where("id = ?", req.Id).First(&bean).Error
	if err != nil {
		return err
	}
	err = global.GWAF_LOCAL_DB.Where("id = ?", req.Id).Delete(model.PrivateInfo{}).Error
	return err
}

func (receiver *WafPrivateInfoService) GetListPureApi() ([]model.PrivateInfo, int64, error) {
	var list []model.PrivateInfo
	var total int64 = 0
	global.GWAF_LOCAL_DB.Model(&model.PrivateInfo{}).Find(&list)
	global.GWAF_LOCAL_DB.Model(&model.PrivateInfo{}).Count(&total)

	return list, total, nil
}