增加批量导入违禁词
This commit is contained in:
@@ -2,6 +2,7 @@ package handler
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"backend/internal/repo"
|
||||
"github.com/gin-gonic/gin"
|
||||
@@ -51,6 +52,31 @@ func (h *BannedWordsHandler) Create(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, gin.H{"data": gin.H{"id": item.ID, "word": item.Word, "hits": item.Hits, "created_at": item.CreatedAt}})
|
||||
}
|
||||
|
||||
// Import bulk-adds words from a free-form text blob — split on newlines,
|
||||
// commas (英文/中文), 顿号 and semicolons — skipping blanks and existing entries.
|
||||
func (h *BannedWordsHandler) Import(c *gin.Context) {
|
||||
var body struct {
|
||||
Text string `json:"text"`
|
||||
}
|
||||
if err := c.ShouldBindJSON(&body); err != nil || strings.TrimSpace(body.Text) == "" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"detail": "请提供要导入的违禁词"})
|
||||
return
|
||||
}
|
||||
words := strings.FieldsFunc(body.Text, func(r rune) bool {
|
||||
switch r {
|
||||
case '\n', '\r', ',', ',', '、', ';', ';':
|
||||
return true
|
||||
}
|
||||
return false
|
||||
})
|
||||
added, skipped, err := h.words.BulkCreate(c.Request.Context(), words)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"detail": "导入失败"})
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"added": added, "skipped": skipped})
|
||||
}
|
||||
|
||||
func (h *BannedWordsHandler) Delete(c *gin.Context) {
|
||||
n, err := h.words.Delete(c.Request.Context(), c.Param("id"))
|
||||
if err != nil {
|
||||
|
||||
@@ -142,6 +142,7 @@ func New(cfg *config.Config, auth *service.AuthService, handlers Handlers) *gin.
|
||||
authed.DELETE("/images", handlers.AdminRead.DeleteImage)
|
||||
authed.GET("/banned-words", handlers.BannedWords.List)
|
||||
authed.POST("/banned-words", handlers.BannedWords.Create)
|
||||
authed.POST("/banned-words/import", handlers.BannedWords.Import)
|
||||
authed.DELETE("/banned-words/:id", handlers.BannedWords.Delete)
|
||||
authed.GET("/refresh/profiles", handlers.ProviderAdmin.RefreshProfiles)
|
||||
authed.POST("/refresh/profiles/:profile_id/refresh-now", handlers.ProviderAdmin.RefreshNow)
|
||||
|
||||
@@ -21,7 +21,7 @@ func NewBannedWordRepository(db *gorm.DB) *BannedWordRepository {
|
||||
|
||||
func (r *BannedWordRepository) List(ctx context.Context) ([]model.BannedWord, error) {
|
||||
var items []model.BannedWord
|
||||
err := r.db.WithContext(ctx).Order("hits DESC, created_at DESC").Find(&items).Error
|
||||
err := r.db.WithContext(ctx).Order("created_at DESC, hits DESC").Find(&items).Error
|
||||
return items, err
|
||||
}
|
||||
|
||||
@@ -42,6 +42,43 @@ func (r *BannedWordRepository) Create(ctx context.Context, word string) (*model.
|
||||
return item, nil
|
||||
}
|
||||
|
||||
// BulkCreate inserts the given words, skipping blanks and ones already in the
|
||||
// table (case-insensitive). Returns how many were added vs skipped.
|
||||
func (r *BannedWordRepository) BulkCreate(ctx context.Context, words []string) (added, skipped int, err error) {
|
||||
existing, err := r.List(ctx)
|
||||
if err != nil {
|
||||
return 0, 0, err
|
||||
}
|
||||
seen := make(map[string]bool, len(existing))
|
||||
for _, w := range existing {
|
||||
seen[strings.ToLower(w.Word)] = true
|
||||
}
|
||||
for _, w := range words {
|
||||
w = strings.TrimSpace(w)
|
||||
if w == "" {
|
||||
continue
|
||||
}
|
||||
key := strings.ToLower(w)
|
||||
if seen[key] {
|
||||
skipped++
|
||||
continue
|
||||
}
|
||||
item := &model.BannedWord{
|
||||
ID: strings.ReplaceAll(uuid.NewString(), "-", "")[:32],
|
||||
Word: w,
|
||||
CreatedAt: time.Now(),
|
||||
UpdatedAt: time.Now(),
|
||||
}
|
||||
if e := r.db.WithContext(ctx).Create(item).Error; e != nil {
|
||||
skipped++
|
||||
continue
|
||||
}
|
||||
seen[key] = true
|
||||
added++
|
||||
}
|
||||
return added, skipped, nil
|
||||
}
|
||||
|
||||
func (r *BannedWordRepository) Delete(ctx context.Context, id string) (int64, error) {
|
||||
res := r.db.WithContext(ctx).Delete(&model.BannedWord{}, "id = ?", id)
|
||||
return res.RowsAffected, res.Error
|
||||
|
||||
Reference in New Issue
Block a user