You've already forked VptPassiveAdapter
112 lines
3.5 KiB
Go
112 lines
3.5 KiB
Go
package main
|
|
|
|
import (
|
|
"ZhenTuLocalPassiveAdapter/api"
|
|
"ZhenTuLocalPassiveAdapter/config"
|
|
"ZhenTuLocalPassiveAdapter/core"
|
|
"ZhenTuLocalPassiveAdapter/dto"
|
|
"ZhenTuLocalPassiveAdapter/telemetry"
|
|
"context"
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/codes"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
var tracer = otel.Tracer("vpt")
|
|
|
|
func startTask(device config.DeviceMapping, task dto.Task) {
|
|
ctx, span := tracer.Start(context.Background(), "startTask")
|
|
defer span.End()
|
|
span.SetAttributes(attribute.String("deviceNo", device.DeviceNo))
|
|
span.SetAttributes(attribute.String("taskId", task.TaskID))
|
|
span.SetAttributes(attribute.String("scenicId", task.ScenicID))
|
|
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")))
|
|
fo, err := core.HandleTask(ctx, device, task)
|
|
if err != nil {
|
|
span.SetStatus(codes.Error, "处理任务失败")
|
|
log.Printf("处理任务失败, TaskID：【%s】, DeviceNo: %s, 错误: %v\n", task.TaskID, task.DeviceNo, err)
|
|
api.ReportTaskFailure(ctx, task.TaskID)
|
|
return
|
|
}
|
|
span.SetAttributes(attribute.String("fileUrl", fo.URL))
|
|
log.Printf("处理任务成功, TaskID：【%s】, DeviceNo: %s\n", task.TaskID, task.DeviceNo)
|
|
err = api.UploadTaskFile(ctx, task, *fo)
|
|
if err != nil {
|
|
span.SetStatus(codes.Error, "上传文件失败")
|
|
log.Printf("上传文件失败, TaskID：【%s】, DeviceNo: %s, 错误: %v\n", task.TaskID, task.DeviceNo, err)
|
|
api.ReportTaskFailure(ctx, task.TaskID)
|
|
return
|
|
}
|
|
result := api.ReportTaskSuccess(ctx, task.TaskID, fo)
|
|
if !result {
|
|
span.SetStatus(codes.Error, "上报任务成功失败")
|
|
log.Printf("上报任务成功失败, TaskID：【%s】, DeviceNo: %s, 错误: %v\n", task.TaskID, task.DeviceNo, err)
|
|
return
|
|
}
|
|
span.SetStatus(codes.Ok, "上传文件成功")
|
|
log.Printf("上传文件成功, TaskID：【%s】, DeviceNo: %s\n", task.TaskID, task.DeviceNo)
|
|
}
|
|
|
|
func main() {
|
|
err := config.LoadConfig()
|
|
if err != nil {
|
|
log.Println("加载配置文件失败:", err)
|
|
return
|
|
}
|
|
// 日志文件路径
|
|
logFilePath := "app.log"
|
|
ctx := context.Background()
|
|
shutdown, err := telemetry.InitTelemetry(ctx)
|
|
if err != nil {
|
|
log.Fatalf("Failed to initialize telemetry: %v", err)
|
|
return
|
|
}
|
|
defer shutdown(ctx)
|
|
// 创建或打开日志文件
|
|
logFile, err := os.OpenFile(logFilePath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
|
|
if err != nil {
|
|
log.Fatalf("Failed to open log file: %v", err)
|
|
}
|
|
defer logFile.Close()
|
|
|
|
// 设置日志输出到文件
|
|
log.SetOutput(logFile)
|
|
if config.Config.Record.Storage.Type == "local" {
|
|
_, err = os.Stat(config.Config.Record.Storage.Path)
|
|
if err != nil {
|
|
log.Println("录像文件夹配置失败", err)
|
|
return
|
|
} else {
|
|
log.Println("录像文件夹配置有效")
|
|
}
|
|
} else {
|
|
log.Println("录像文件夹配置为OSS")
|
|
}
|
|
// 每两秒定时执行
|
|
for {
|
|
// 执行任务
|
|
tasks, err := api.SyncTask()
|
|
if err == nil {
|
|
for _, task := range tasks {
|
|
log.Printf("开始处理任务, TaskID：【%s】，DeviceNo: %s，开始时间: %s，结束时间: %s\n", task.TaskID, task.DeviceNo, task.StartTime, task.EndTime)
|
|
// 处理任务
|
|
for _, device := range config.Config.Devices {
|
|
if device.DeviceNo == task.DeviceNo {
|
|
// 处理任务
|
|
go startTask(device, task)
|
|
break // 提前返回，避免不必要的循环
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
log.Println("同步任务失败:", err)
|
|
}
|
|
// 等待两秒
|
|
<-time.After(2 * time.Second)
|
|
}
|
|
}
|