feat: 迁移plant
This commit is contained in:
@@ -5,6 +5,12 @@ package ocr
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strings"
|
||||
|
||||
"sundynix-micro-go/app/plant/api/internal/svc"
|
||||
"sundynix-micro-go/app/plant/api/internal/types"
|
||||
@@ -27,8 +33,60 @@ func NewOcrClassifyLogic(ctx context.Context, svcCtx *svc.ServiceContext) *OcrCl
|
||||
}
|
||||
}
|
||||
|
||||
func (l *OcrClassifyLogic) OcrClassify(req *types.OcrReq) error {
|
||||
// todo: add your logic here and delete this line
|
||||
|
||||
return nil
|
||||
// baiduTokenResp 百度 token 响应
|
||||
type baiduTokenResp struct {
|
||||
AccessToken string `json:"access_token"`
|
||||
}
|
||||
|
||||
// OcrClassify 调用百度 AI 植物识别接口
|
||||
func (l *OcrClassifyLogic) OcrClassify(req *types.OcrReq) (interface{}, error) {
|
||||
// 1. 获取百度 access_token
|
||||
apiKey := l.svcCtx.Config.BaiduOcr.ApiKey
|
||||
secretKey := l.svcCtx.Config.BaiduOcr.SecretKey
|
||||
if apiKey == "" || secretKey == "" {
|
||||
return nil, fmt.Errorf("百度 OCR 未配置 ApiKey/SecretKey")
|
||||
}
|
||||
|
||||
tokenURL := fmt.Sprintf(
|
||||
"https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=%s&client_secret=%s",
|
||||
apiKey, secretKey,
|
||||
)
|
||||
tokenResp, err := http.Post(tokenURL, "application/x-www-form-urlencoded", nil)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("获取百度 token 失败: %w", err)
|
||||
}
|
||||
defer tokenResp.Body.Close()
|
||||
tokenBody, _ := io.ReadAll(tokenResp.Body)
|
||||
var tokenObj baiduTokenResp
|
||||
if err = json.Unmarshal(tokenBody, &tokenObj); err != nil || tokenObj.AccessToken == "" {
|
||||
return nil, fmt.Errorf("解析百度 token 失败")
|
||||
}
|
||||
|
||||
// 2. 调用植物识别接口(传入图片 URL)
|
||||
apiURL := "https://aip.baidubce.com/rest/2.0/image-classify/v1/plant?access_token=" + tokenObj.AccessToken
|
||||
payload := strings.NewReader("url=" + url.QueryEscape(req.ImageUrl) + "&baike_num=1")
|
||||
classifyReq, err := http.NewRequest("POST", apiURL, payload)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("创建请求失败: %w", err)
|
||||
}
|
||||
classifyReq.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||
|
||||
client := &http.Client{}
|
||||
classifyResp, err := client.Do(classifyReq)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("调用百度植物识别接口失败: %w", err)
|
||||
}
|
||||
defer classifyResp.Body.Close()
|
||||
|
||||
body, err := io.ReadAll(classifyResp.Body)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("读取识别结果失败: %w", err)
|
||||
}
|
||||
|
||||
// 3. 直接返回百度原始结果(前端自行解析 result 字段)
|
||||
var result interface{}
|
||||
if err = json.Unmarshal(body, &result); err != nil {
|
||||
return nil, fmt.Errorf("解析识别结果失败: %w", err)
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user