feat: 迁移plant

This commit is contained in:
Blizzard
2026-05-23 13:55:05 +08:00
parent a93477ea8e
commit ae6d03d351
228 changed files with 25296 additions and 917 deletions
@@ -0,0 +1,24 @@
package ocr
import (
"context"
"github.com/zeromicro/go-zero/core/logx"
"sundynix-micro-go/app/plant/api/internal/svc"
"sundynix-micro-go/app/plant/api/internal/types"
plantPb "sundynix-micro-go/app/plant/rpc/plant"
)
type DeleteClassifyLogLogic struct {
logx.Logger
ctx context.Context
svcCtx *svc.ServiceContext
}
func NewDeleteClassifyLogLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteClassifyLogLogic {
return &DeleteClassifyLogLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx}
}
func (l *DeleteClassifyLogLogic) DeleteClassifyLog(req *types.IdsReq) error {
_, err := l.svcCtx.PlantRpc.DeleteClassifyLog(l.ctx, &plantPb.IdsReq{Ids: req.Ids})
return err
}
@@ -0,0 +1,24 @@
package ocr
import (
"context"
"fmt"
"github.com/zeromicro/go-zero/core/logx"
"sundynix-micro-go/app/plant/api/internal/svc"
plantPb "sundynix-micro-go/app/plant/rpc/plant"
)
type GetMyClassifyLogLogic struct {
logx.Logger
ctx context.Context
svcCtx *svc.ServiceContext
}
func NewGetMyClassifyLogLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetMyClassifyLogLogic {
return &GetMyClassifyLogLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx}
}
func (l *GetMyClassifyLogLogic) GetMyClassifyLog() (*plantPb.ClassifyLogListResp, error) {
userId := fmt.Sprintf("%v", l.ctx.Value("userId"))
return l.svcCtx.PlantRpc.GetMyClassifyLog(l.ctx, &plantPb.GetProfileReq{UserId: userId})
}
@@ -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
}