82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package callback
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"time"
|
|
|
|
"lone-services/pkg/utils"
|
|
"lone-services/services/analysis/internal/dao"
|
|
analysisoss "lone-services/services/analysis/internal/oss"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
func ProcessUpload(ctx context.Context, payload analysisoss.CallbackPayload) error {
|
|
if payload.TaskId == utils.StringEmpty || payload.Object == utils.StringEmpty {
|
|
return fmt.Errorf("callback missing task_id or object")
|
|
}
|
|
|
|
sess, err := analysisoss.LoadUploadSession(ctx, payload.TaskId)
|
|
if err != nil {
|
|
return fmt.Errorf("load upload session: %w", err)
|
|
}
|
|
if sess.DateDir == utils.StringEmpty {
|
|
return fmt.Errorf("upload session date_dir empty")
|
|
}
|
|
|
|
downloadURL, err := BuildDownloadURL(payload.Object)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
subTaskId := uuid.New().String()
|
|
subDirName := strconv.FormatInt(time.Now().UnixMilli(), 10)
|
|
targetDir := filepath.Join(utils.GetConfigString("file.analysis_path"), sess.DateDir, payload.TaskId)
|
|
if err = os.MkdirAll(targetDir, os.ModePerm); err != nil {
|
|
return fmt.Errorf("mkdir analysis path: %w", err)
|
|
}
|
|
zipPath := filepath.Join(targetDir, subDirName+".zip")
|
|
|
|
if err = DownloadFile(downloadURL, zipPath); err != nil {
|
|
return fmt.Errorf("download zip: %w", err)
|
|
}
|
|
|
|
pathRes, coverPath, err := HandleTaskZip(zipPath)
|
|
if err != nil {
|
|
return fmt.Errorf("handle zip: %w", err)
|
|
}
|
|
|
|
if err = CacheUploadedLights(payload.TaskId, pathRes.FinalDir); err != nil {
|
|
logx.WithContext(ctx).Errorf("cache uploaded lights failed: %v", err)
|
|
}
|
|
|
|
if utils.GetConfigInt("switch.facepp") == utils.NumberOne {
|
|
logx.WithContext(ctx).Infof("facepp skip task_id=%s", payload.TaskId)
|
|
}
|
|
|
|
if utils.GetConfigInt("switch.analysis") == utils.NumberOne {
|
|
if err = NotifyPythonService(payload.TaskId, subTaskId, pathRes.FinalDir); err != nil {
|
|
logx.WithContext(ctx).Errorf("notify python failed: %v", err)
|
|
}
|
|
}
|
|
|
|
ts := dao.TaskTimeStamp{
|
|
TaskId: payload.TaskId,
|
|
SubTaskId: subTaskId,
|
|
ProcessStartTs: strconv.FormatInt(time.Now().UnixMilli(), 10),
|
|
}
|
|
userId := sess.SkinUserId()
|
|
if userId < utils.NumberOne {
|
|
userId = sess.UserId
|
|
}
|
|
if err = SaveTask(ts, userId, pathRes, coverPath); err != nil {
|
|
return err
|
|
}
|
|
return SaveSubTask(ts, pathRes)
|
|
}
|