SensitiveWord相关修改测试

This commit is contained in:
平凡梦
2020-07-10 09:06:24 +08:00
parent c4b62a05dc
commit 946810c177
3 changed files with 23 additions and 9 deletions

View File

@@ -727,7 +727,7 @@ func CheckHasSensitiveWord(word string) (bool, error) {
}
func IsSensitiveWordInList(str string) (bool, string) {
wordList, err := dao.GetSensitiveWordList()
wordList, err := dao.GetSensitiveWordList(2)
if err == nil {
for _, value := range wordList {
keyWord := value.Word

View File

@@ -7,22 +7,32 @@ import (
"time"
)
func GetSensitiveWordList() (wordList []*model.SensitiveWord, err error) {
sql := `SELECT * FROM sensitive_word WHERE deleted_at = ?`
err = GetRows(nil, &wordList, sql, utils.DefaultTimeValue)
/*VendorID = 2 的时候是通用保留之前的1看后面会不会有扩展*/
func GetSensitiveWordList(vendorID int) (wordList []*model.SensitiveWord, err error) {
if vendorID == 0 {
vendorID = 2
}
sql := `SELECT * FROM sensitive_word WHERE deleted_at = ? AND vendor_id = ?`
err = GetRows(nil, &wordList, sql, utils.DefaultTimeValue, vendorID)
return wordList, err
}
func InsertSensitiveWord(word string, vendorID int, userName string) error {
if vendorID == 0 {
vendorID = 2
}
sensitiveWord := &model.SensitiveWord{Word: word, VendorID: vendorID}
WrapAddIDCULDEntity(sensitiveWord, userName)
return CreateEntity(nil, sensitiveWord)
}
func DeleteSensitiveWord(wordList *model.SensitiveWord, id int, userName string) (word *model.SensitiveWord, err error) {
func DeleteSensitiveWord(wordList *model.SensitiveWord, id int, userName string, vendorID int) (word *model.SensitiveWord, err error) {
if id == 0 {
return nil, errors.New("入参错误找不到相应ID")
}
if vendorID == 0 {
vendorID = 2
}
wordList.ID = id
wordList.DeletedAt = time.Now()
wordList.LastOperator = userName
@@ -32,10 +42,13 @@ func DeleteSensitiveWord(wordList *model.SensitiveWord, id int, userName string)
return wordList, nil
}
func UpdateSensitiveWord(wordList *model.SensitiveWord, id int, userName string, params ...string) (word *model.SensitiveWord, err error) {
func UpdateSensitiveWord(wordList *model.SensitiveWord, vendorID int, id int, userName string, params ...string) (word *model.SensitiveWord, err error) {
if id == 0 {
return nil, errors.New("入参错误找不到相应ID")
}
if vendorID == 0 {
vendorID = 2
}
wordList.ID = id
wordList.UpdatedAt = time.Now()
wordList.LastOperator = userName