init

2026-01-12 11:33:43 +08:00
commit f07062dbd7
38 changed files with 6805 additions and 0 deletions
--- a/internal/client/ocr_client.go
+++ b/internal/client/ocr_client.go
@@ -0,0 +1,113 @@
+package client
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"qd-sc/internal/config"
+)
+
+// OCRClient OCR服务客户端
+type OCRClient struct {
+	baseURL    string
+	httpClient *http.Client
+	logLevel   string
+}
+
+// OCRResponse OCR服务响应结构
+type OCRResponse struct {
+	Code       int     `json:"code"`
+	Data       string  `json:"data"`
+	CostTimeMs float64 `json:"cost_time_ms"`
+	Msg        string  `json:"msg,omitempty"`
+}
+
+// NewOCRClient 创建OCR客户端
+func NewOCRClient(cfg *config.Config) *OCRClient {
+	return &OCRClient{
+		baseURL:    cfg.OCR.BaseURL,
+		httpClient: NewHTTPClient(HTTPClientConfig{Timeout: cfg.OCR.Timeout, MaxIdleConns: 100, MaxIdleConnsPerHost: 50, MaxConnsPerHost: 0}),
+		logLevel:   cfg.Logging.Level,
+	}
+}
+
+// ParseURL 通过URL解析远程文件内容
+// 支持图片、PDF、Excel、PPT等格式
+func (c *OCRClient) ParseURL(fileURL string) (string, error) {
+	// 构建请求体
+	reqBody := map[string]string{
+		"url": fileURL,
+	}
+
+	reqData, err := json.Marshal(reqBody)
+	if err != nil {
+		return "", fmt.Errorf("序列化请求失败: %w", err)
+	}
+
+	// 打印文件解析请求体
+	log.Printf("OCR文件解析请求: URL=%s, 请求体=%s", c.baseURL+"/ocr/url", string(reqData))
+
+	// 创建HTTP请求
+	httpReq, err := http.NewRequest("POST", c.baseURL+"/ocr/url", bytes.NewReader(reqData))
+	if err != nil {
+		return "", fmt.Errorf("创建HTTP请求失败: %w", err)
+	}
+
+	httpReq.Header.Set("Content-Type", "application/json")
+
+	// 发送请求
+	resp, err := c.httpClient.Do(httpReq)
+	if err != nil {
+		return "", fmt.Errorf("HTTP请求失败: %w", err)
+	}
+	defer resp.Body.Close()
+
+	// 读取响应
+	respBody, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return "", fmt.Errorf("读取响应失败: %w", err)
+	}
+
+	// 打印原始响应（仅在debug级别时显示完整响应，否则显示摘要）
+	if c.logLevel == "debug" {
+		log.Printf("OCR原始响应: %s", string(respBody))
+	} else {
+		// 非debug模式下显示响应摘要
+		if len(respBody) > 200 {
+			log.Printf("OCR响应摘要: %s... (共%d字节)", string(respBody[:200]), len(respBody))
+		} else {
+			log.Printf("OCR响应: %s", string(respBody))
+		}
+	}
+
+	// 解析响应
+	var ocrResp OCRResponse
+	if err := json.Unmarshal(respBody, &ocrResp); err != nil {
+		return "", fmt.Errorf("解析响应失败: %w", err)
+	}
+
+	// 打印解析结果
+	log.Printf("OCR解析结果: Code=%d, CostTimeMs=%.2f, 数据长度=%d字节", ocrResp.Code, ocrResp.CostTimeMs, len(ocrResp.Data))
+	if c.logLevel == "debug" && ocrResp.Data != "" {
+		// debug模式下打印解析出的数据内容
+		if len(ocrResp.Data) > 500 {
+			log.Printf("OCR解析数据(前500字符): %s...", ocrResp.Data[:500])
+		} else {
+			log.Printf("OCR解析数据: %s", ocrResp.Data)
+		}
+	}
+
+	// 检查业务状态码
+	if ocrResp.Code != 200 {
+		errMsg := ocrResp.Msg
+		if errMsg == "" {
+			errMsg = "OCR解析失败"
+		}
+		return "", fmt.Errorf("OCR服务返回错误: %s", errMsg)
+	}
+
+	return ocrResp.Data, nil
+}