79 lines
3.0 KiB
Go
79 lines
3.0 KiB
Go
package core
|
|
|
|
import (
|
|
"ZhenTuLocalPassiveAdapter/config"
|
|
"ZhenTuLocalPassiveAdapter/dto"
|
|
"ZhenTuLocalPassiveAdapter/fs"
|
|
"ZhenTuLocalPassiveAdapter/util"
|
|
"context"
|
|
"fmt"
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/codes"
|
|
"log"
|
|
"os"
|
|
"path"
|
|
)
|
|
|
|
var tracer = otel.Tracer("task")
|
|
|
|
func HandleTask(ctx context.Context, device config.DeviceMapping, task dto.Task) (*dto.FileObject, error) {
|
|
subCtx, span := tracer.Start(ctx, "startTask")
|
|
defer span.End()
|
|
adapter := fs.GetAdapter()
|
|
span.SetAttributes(attribute.String("taskId", task.TaskID))
|
|
span.SetAttributes(attribute.String("startTime", task.StartTime.Format("2006-01-02 15:04:05")))
|
|
span.SetAttributes(attribute.String("endTime", task.EndTime.Format("2006-01-02 15:04:05")))
|
|
span.SetAttributes(attribute.String("device.no", device.DeviceNo))
|
|
span.SetAttributes(attribute.String("device.name", device.Name))
|
|
fileList, err := adapter.GetFileList(
|
|
subCtx,
|
|
path.Join(device.Name, task.StartTime.Format("2006"+config.Config.FileName.DateSeparator+"01"+config.Config.FileName.DateSeparator+"02")),
|
|
task.StartTime,
|
|
)
|
|
if err != nil {
|
|
span.SetAttributes(attribute.String("error", err.Error()))
|
|
span.SetStatus(codes.Error, "获取文件列表失败")
|
|
log.Printf("获取文件列表失败, DeviceNo: %s, 错误: %v\n", device.DeviceNo, err)
|
|
return nil, err
|
|
}
|
|
files := util.FilterAndSortFiles(fileList, task.StartTime, task.EndTime)
|
|
if len(files) == 0 {
|
|
span.SetStatus(codes.Error, "没有找到文件")
|
|
return nil, fmt.Errorf("没有找到文件")
|
|
}
|
|
span.SetAttributes(attribute.Int("fileCount", len(files)))
|
|
constructTask, err := util.CheckFileCoverageAndConstructTask(subCtx, files, task.StartTime, task.EndTime, task)
|
|
if err != nil {
|
|
span.SetAttributes(attribute.String("error", err.Error()))
|
|
span.SetStatus(codes.Error, "文件片段检查失败")
|
|
log.Printf("文件片段检查失败, DeviceNo: %s, 错误: %v\n", device.DeviceNo, err)
|
|
return nil, err
|
|
}
|
|
ok := util.RunFfmpegTask(subCtx, constructTask)
|
|
if !ok {
|
|
span.SetAttributes(attribute.String("error", "ffmpeg任务执行失败"))
|
|
span.SetStatus(codes.Error, "ffmpeg任务执行失败")
|
|
return nil, fmt.Errorf("ffmpeg任务执行失败")
|
|
}
|
|
outfile, err := os.Stat(constructTask.OutputFile)
|
|
if err != nil {
|
|
span.SetAttributes(attribute.String("error", err.Error()))
|
|
span.SetStatus(codes.Error, "文件不存在")
|
|
return nil, fmt.Errorf("文件不存在：%s", constructTask.OutputFile)
|
|
}
|
|
span.SetAttributes(attribute.String("file.name", outfile.Name()))
|
|
span.SetAttributes(attribute.Int64("file.size", outfile.Size()))
|
|
if outfile.Size() < 4096 {
|
|
span.SetAttributes(attribute.String("error", "文件大小过小"))
|
|
span.SetStatus(codes.Error, "文件大小过小")
|
|
return nil, fmt.Errorf("文件大小过小：%s", constructTask.OutputFile)
|
|
}
|
|
return &dto.FileObject{
|
|
CreateTime: task.EndTime,
|
|
EndTime: task.EndTime,
|
|
NeedDownload: true,
|
|
URL: constructTask.OutputFile,
|
|
}, nil
|
|
}
|