114 lines
2.9 KiB
Go
114 lines
2.9 KiB
Go
|
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
|
|||
|
|
}
|