You've already forked VptPassiveAdapter
Jerry Yan
11f508342d
- 修改 config.yaml 中的 API 地址、存储配置及设备信息 - 新增 Viid 配置项支持新功能模块 - 在 go.mod 和 go.sum 中更新依赖包版本，引入新的第三方库 - 添加 model/custom_time.go 文件实现自定义时间类型的 JSON 序列化与反序列化 - 调整 DTO 结构体以适配新的配置项 - 升级 OTLP 导出器相关依赖并移除旧的标准输出导出器 - 引入 Gin 框架及相关中间件提升服务性能 - 更新 Protobuf 和 gRPC 相关依赖至最新版本 - 增加 zap 日志库和 lumberjack 日志轮转支持 - 添加 sonic、json-iterator 等高性能 JSON 处理库优化数据解析效率 - 引入 testify 断言库增强测试代码可读性 - 更新 sync 包版本提高并发安全性 - 添加 mock 工具支持单元测试模拟对象 - 引入 validator/v10 实现请求参数校验功能 - 更新 crypto、net、text 等标准库依赖确保安全性和兼容性 - 增加 mimetype 库用于文件类型识别 - 引入 quic-go 支持 HTTP/3 协议通信 - 添加 base64x
71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
package api
|
|
|
|
import (
|
|
"ZhenTuLocalPassiveAdapter/logger"
|
|
"context"
|
|
"fmt"
|
|
"go.uber.org/zap"
|
|
"golang.org/x/sync/errgroup"
|
|
)
|
|
|
|
// UploadFaceData implements the coordinated upload flow: Get Config -> Upload -> Notify
|
|
func UploadFaceData(ctx context.Context, scenicId int64, deviceNo string, faceImg, thumbImg, srcImg []byte, facePos FacePositionInfo) error {
|
|
// 1. Get Upload Config
|
|
uploadConfig, err := GetUploadConfig(ctx, scenicId, deviceNo)
|
|
if err != nil {
|
|
logger.Error("获取上传配置失败", zap.String("deviceNo", deviceNo), zap.Error(err))
|
|
return err
|
|
}
|
|
|
|
logger.Info("获取到VIID任务ID", zap.Int64("taskId", uploadConfig.TaskID))
|
|
|
|
// 2. Parallel Upload to OSS
|
|
g, subCtx := errgroup.WithContext(ctx)
|
|
|
|
// Upload Face Image
|
|
g.Go(func() error {
|
|
if len(faceImg) > 0 {
|
|
if err := UploadFileToOSS(subCtx, uploadConfig.FaceUploadURL, faceImg, "image/jpeg"); err != nil {
|
|
return fmt.Errorf("upload face image failed: %w", err)
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
|
|
// Upload Thumbnail Image
|
|
g.Go(func() error {
|
|
if len(thumbImg) > 0 {
|
|
if err := UploadFileToOSS(subCtx, uploadConfig.ThumbnailUploadURL, thumbImg, "image/jpeg"); err != nil {
|
|
return fmt.Errorf("upload thumbnail image failed: %w", err)
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
|
|
// Upload Source Image
|
|
g.Go(func() error {
|
|
if len(srcImg) > 0 {
|
|
if err := UploadFileToOSS(subCtx, uploadConfig.SourceUploadURL, srcImg, "image/jpeg"); err != nil {
|
|
return fmt.Errorf("upload source image failed: %w", err)
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
|
|
if err := g.Wait(); err != nil {
|
|
// Report failure
|
|
SubmitFailure(ctx, uploadConfig.TaskID, "UPLOAD_FAILED", err.Error())
|
|
logger.Error("文件上传失败", zap.Int64("taskId", uploadConfig.TaskID), zap.Error(err))
|
|
return err
|
|
}
|
|
|
|
// 3. Submit Result
|
|
if err := SubmitResult(ctx, uploadConfig.TaskID, facePos); err != nil {
|
|
logger.Error("提交结果失败", zap.Int64("taskId", uploadConfig.TaskID), zap.Error(err))
|
|
return err
|
|
}
|
|
|
|
logger.Info("VIID任务完成", zap.Int64("taskId", uploadConfig.TaskID))
|
|
return nil
|
|
}
|