完善二维码池 OCR 配置与识别

This commit is contained in:
yml
2026-06-18 15:22:46 +08:00
parent e8bdf574f4
commit 35f2d93fdf
16 changed files with 1025 additions and 48 deletions
@@ -1,6 +1,7 @@
package chat
import (
"errors"
"net/http"
"strconv"
@@ -78,6 +79,41 @@ func (h *Handler) GetQrCodeStatsHandler(c *gin.Context) {
c.JSON(http.StatusOK, gin.H{"data": stats})
}
// RecognizeQrCodeGroupNameHandler 调用 PaddleOCR API 识别二维码图片中的企业微信群名
func (h *Handler) RecognizeQrCodeGroupNameHandler(c *gin.Context) {
file, header, err := c.Request.FormFile("file")
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "请上传二维码图片"})
return
}
defer file.Close()
result, err := h.service.repo.RecognizeQrCodeGroupName(
c.Request.Context(),
header.Filename,
header.Header.Get("Content-Type"),
file,
)
if err != nil {
if errors.Is(err, ErrQrCodeOCRNotConfigured) {
c.JSON(http.StatusBadRequest, gin.H{"error": "请先在系统配置中填写 PaddleOCR API Token"})
return
}
if errors.Is(err, ErrQrCodeOCRInvalidFile) {
c.JSON(http.StatusBadRequest, gin.H{"error": "无效的二维码图片"})
return
}
if errors.Is(err, ErrQrCodeOCRUnavailable) {
c.JSON(http.StatusBadGateway, gin.H{"error": "PaddleOCR 服务暂时不可用"})
return
}
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return
}
c.JSON(http.StatusOK, gin.H{"data": result})
}
// UpdateQrCodeHandler 更新二维码
func (h *Handler) UpdateQrCodeHandler(c *gin.Context) {
id, err := strconv.ParseUint(c.Param("id"), 10, 64)