feat: not check sensitive if content empty

OpenTreeHole · Apr 2, 2024 · edd8dae · edd8dae
1 parent af931a4
commit edd8dae
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/models/hole.go b/models/hole.go
@@ -2,6 +2,7 @@ package models
 
 import (
 	"fmt"
+	"golang.org/x/exp/maps"
 	"time"
 
 	"github.com/gofiber/fiber/v2"
@@ -118,7 +119,7 @@ func loadTags(holes Holes) (err error) {
 	}
 
 	var tags Tags
-	err = DB.Where("id in ?", utils.Keys(tagIDs)).Find(&tags).Error
+	err = DB.Where("id in ?", maps.Keys(tagIDs)).Find(&tags).Error
 	if err != nil {
 		return err
 	}

diff --git a/utils/sensitive/api.go b/utils/sensitive/api.go
@@ -6,6 +6,7 @@ import (
 	"github.com/yidun/yidun-golang-sdk/yidun/service/antispam/image/v5"
 	"github.com/yidun/yidun-golang-sdk/yidun/service/antispam/image/v5/check"
 	"strconv"
+	"strings"
 	"time"
 	"treehole_next/config"
 	"treehole_next/utils"
@@ -66,7 +67,13 @@ func CheckSensitive(params ParamsForCheck) (resp *ResponseForCheck, err error) {
 			Labels: nil,
 		}, nil
 	}
-	params.Content = clearContent
+	params.Content = strings.TrimSpace(clearContent)
+	if params.Content == "" {
+		return &ResponseForCheck{
+			Pass:   true,
+			Labels: nil,
+		}, nil
+	}
 
 	return CheckSensitiveText(params)
 }