cory/zmkgC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

初始

Browse Source
This commit is contained in:
罗成
2025-07-07 20:11:59 +08:00
parent ab0fdbc447
commit 06e3aa2eb3
2009 changed files with 193082 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

327
internal/app/system/logic/manageTaskRecordResource/manage_task_record_resource.go Normal file
View File

@ -0,0 +1,327 @@
// ==========================================================================
// GFast自动生成logic操作代码。
// 生成日期：2024-05-30 17:07:01
// 生成路径: internal/app/system/logic/manage_task_record_resource.go
// 生成人：gfast
// desc:下发记录回传的媒体文件
// company:云南奇讯科技有限公司
// ==========================================================================
package logic
import (
"context"
"errors"
"fmt"
"github.com/gogf/gf/v2/frame/g"
"github.com/gogf/gf/v2/os/gtime"
"github.com/minio/minio-go/v7"
"github.com/tiger1103/gfast/v3/api/v1/common/coryCommon"
"github.com/tiger1103/gfast/v3/api/v1/system"
"github.com/tiger1103/gfast/v3/internal/app/system/consts"
"github.com/tiger1103/gfast/v3/internal/app/system/dao"
"github.com/tiger1103/gfast/v3/internal/app/system/model"
"github.com/tiger1103/gfast/v3/internal/app/system/model/do"
"github.com/tiger1103/gfast/v3/internal/app/system/service"
"github.com/tiger1103/gfast/v3/library/liberr"
"github.com/tiger1103/gfast/v3/third/plane/dj"
"strings"
"sync"
"time"
)
func init() {
service.RegisterManageTaskRecordResource(New())
}
func New() *sManageTaskRecordResource {
return &sManageTaskRecordResource{}
}
const UavRedisKey = "uav:flightId"
type sManageTaskRecordResource struct{}
func (s *sManageTaskRecordResource) List(ctx context.Context, req *system.ManageTaskRecordResourceSearchReq) (listRes *system.ManageTaskRecordResourceSearchRes, err error) {
listRes = new(system.ManageTaskRecordResourceSearchRes)
err = g.Try(ctx, func(ctx context.Context) {
m := dao.ManageTaskRecordResource.Ctx(ctx).WithAll()
if req.FlightId != "" {
m = m.Where(dao.ManageTaskRecordResource.Columns().FlightId+" = ?", req.FlightId)
}
//if req.FileName != "" {
// m = m.Where(dao.ManageTaskRecordResource.Columns().FileName+" like ?", "%"+req.FileName+"%")
//}
//if req.ObjectKey != "" {
// m = m.Where(dao.ManageTaskRecordResource.Columns().ObjectKey+" = ?", req.ObjectKey)
//}
//if req.Path != "" {
// m = m.Where(dao.ManageTaskRecordResource.Columns().Path+" = ?", req.Path)
//}
//if req.DroneModelKey != "" {
// m = m.Where(dao.ManageTaskRecordResource.Columns().DroneModelKey+" = ?", req.DroneModelKey)
//}
//if req.PayloadModelKey != "" {
// m = m.Where(dao.ManageTaskRecordResource.Columns().PayloadModelKey+" = ?", req.PayloadModelKey)
//}
//if req.CreateTime != "" {
// m = m.Where(dao.ManageTaskRecordResource.Columns().CreateTime+" = ?", gconv.Time(req.CreateTime))
//}
//if req.ShootPosition != "" {
// m = m.Where(dao.ManageTaskRecordResource.Columns().ShootPosition+" = ?", req.ShootPosition)
//}
//if req.Lat != "" {
// m = m.Where(dao.ManageTaskRecordResource.Columns().Lat+" = ?", req.Lat)
//}
//if req.Lng != "" {
// m = m.Where(dao.ManageTaskRecordResource.Columns().Lng+" = ?", req.Lng)
//}
//if req.Img != "" {
// m = m.Where(dao.ManageTaskRecordResource.Columns().Img+" = ?", req.Img)
//}
//if req.ImgSize != "" {
// m = m.Where(dao.ManageTaskRecordResource.Columns().ImgSize+" = ?", req.ImgSize)
//}
listRes.Total, err = m.Count()
liberr.ErrIsNil(ctx, err, "获取总行数失败")
if req.PageNum == 0 {
req.PageNum = 1
}
listRes.CurrentPage = req.PageNum
if req.PageSize == 0 {
req.PageSize = consts.PageSize
}
order := "flight_id desc"
if req.OrderBy != "" {
order = req.OrderBy
}
var res []*model.ManageTaskRecordResourceInfoRes
err = m.Fields(system.ManageTaskRecordResourceSearchRes{}).Page(req.PageNum, req.PageSize).Order(order).Scan(&res)
liberr.ErrIsNil(ctx, err, "获取数据失败")
listRes.List = make([]*model.ManageTaskRecordResourceListRes, len(res))
client, cerr := dj.PublicMinioClient()
liberr.ErrIsNil(ctx, cerr)
for k, v := range res {
listRes.List[k] = &model.ManageTaskRecordResourceListRes{
FlightId: v.FlightId,
FileName: v.FileName,
ObjectKey: v.ObjectKey,
Path: v.Path,
DroneModelKey: v.DroneModelKey,
PayloadModelKey: v.PayloadModelKey,
AbsoluteAltitude: v.AbsoluteAltitude,
RelativeAltitude: v.RelativeAltitude,
CreateTime: v.CreateTime,
ShootPosition: v.ShootPosition,
Lat: v.Lat,
Lng: v.Lng,
Img: v.Img,
Online: dj.GenerateAnOnlineScopedUrlBasedOnTheObject(ctx, client, dj.BucketName, v.ObjectKey, 3*24*time.Hour),
}
}
})
return
}
func (s *sManageTaskRecordResource) ObtainResourcesAccordingToFlightId(ctx context.Context, flightId string) (listRes *system.ObtainResourcesAccordingToFlightIdRes, err error) {
listRes = new(system.ObtainResourcesAccordingToFlightIdRes)
var str []string
err = g.Try(ctx, func(ctx context.Context) {
//默认第二代机场
m := dao.ManageTaskRecordResource.Ctx(ctx).WithAll()
if flightId != "" {
m = m.Where(dao.ManageTaskRecordResource.Columns().FlightId+" = ?", flightId)
}
order := "flight_id desc"
var res []*model.ManageTaskRecordResourceInfoRes
//默认第二代机场
err = m.Where("file_name like ?", "%W.jpeg").Order(order).Scan(&res)
liberr.ErrIsNil(ctx, err, "获取数据失败")
//三代机场的M4TD飞机
if len(res) == 0 {
err = m.Where("file_name like ?", "%V.jpeg").Order(order).Scan(&res)
liberr.ErrIsNil(ctx, err, "获取数据失败")
}
fmt.Println("1获取到的结果:", len(res))
split := strings.Split(coryCommon.Uav, "/")
fmt.Println("2获取到的结果:", split)
for _, v := range res {
str = append(str, coryCommon.GlobalFile+"/"+split[len(split)-2]+"/"+v.ObjectKey)
fmt.Println("拼接后的字符串:", coryCommon.GlobalFile+"/"+split[len(split)-2]+"/"+v.ObjectKey)
}
listRes.List = str
})
return
}
func (s *sManageTaskRecordResource) GetByFlightId(ctx context.Context, flightId string) (res *model.ManageTaskRecordResourceInfoRes, err error) {
err = g.Try(ctx, func(ctx context.Context) {
err = dao.ManageTaskRecordResource.Ctx(ctx).WithAll().Where(dao.ManageTaskRecordResource.Columns().FlightId, flightId).Scan(&res)
liberr.ErrIsNil(ctx, err, "获取信息失败")
})
return
}
func (s *sManageTaskRecordResource) Add(ctx context.Context, req *system.ManageTaskRecordResourceAddReq) (err error) {
err = g.Try(ctx, func(ctx context.Context) {
_, err = dao.ManageTaskRecordResource.Ctx(ctx).Insert(do.ManageTaskRecordResource{
FlightId: req.FlightId,
FileName: req.FileName,
ObjectKey: req.ObjectKey,
Path: req.Path,
DroneModelKey: req.DroneModelKey,
PayloadModelKey: req.PayloadModelKey,
AbsoluteAltitude: req.AbsoluteAltitude,
RelativeAltitude: req.RelativeAltitude,
CreateTime: req.CreateTime,
ShootPosition: req.ShootPosition,
Lat: req.Lat,
Lng: req.Lng,
Img: req.Img,
ImgSize: req.ImgSize,
})
liberr.ErrIsNil(ctx, err, "添加失败")
})
return
}
func (s *sManageTaskRecordResource) Edit(ctx context.Context, req *system.ManageTaskRecordResourceEditReq) (err error) {
err = g.Try(ctx, func(ctx context.Context) {
_, err = dao.ManageTaskRecordResource.Ctx(ctx).WherePri(req.FlightId).Update(do.ManageTaskRecordResource{
FileName: req.FileName,
ObjectKey: req.ObjectKey,
Path: req.Path,
DroneModelKey: req.DroneModelKey,
PayloadModelKey: req.PayloadModelKey,
AbsoluteAltitude: req.AbsoluteAltitude,
RelativeAltitude: req.RelativeAltitude,
CreateTime: req.CreateTime,
ShootPosition: req.ShootPosition,
Lat: req.Lat,
Lng: req.Lng,
Img: req.Img,
ImgSize: req.ImgSize,
})
liberr.ErrIsNil(ctx, err, "修改失败")
})
return
}
func (s *sManageTaskRecordResource) Delete(ctx context.Context, flightIds []string) (err error) {
err = g.Try(ctx, func(ctx context.Context) {
_, err = dao.ManageTaskRecordResource.Ctx(ctx).Delete(dao.ManageTaskRecordResource.Columns().FlightId+" in (?)", flightIds)
liberr.ErrIsNil(ctx, err, "删除失败")
})
return
}
func (s *sManageTaskRecordResource) DownloadAccordingToMysqlFunc(ctx context.Context, FlightId string) (err error) {
err = g.Try(ctx, func(ctx context.Context) {
//0、如果已经下载过就允许再次下载了
count, _ := dao.ManageTaskRecord.Ctx(ctx).Where("flight_id", FlightId).Where("is_resource_ready", "1").Count()
if count > 0 {
liberr.ErrIsNil(ctx, errors.New("已下载，不可重复下载！"))
return
}
//1、获取可操作的minio的对象
var listEntity []*model.ManageTaskRecordResourceListRes
err := dao.ManageTaskRecordResource.Ctx(ctx).Where("flight_id", FlightId).Where("file_name like ?", "%W.jpeg").Scan(&listEntity)
if len(listEntity) == 0 { //三代机场的M4TD飞机
err = dao.ManageTaskRecordResource.Ctx(ctx).Where("flight_id", FlightId).Where("file_name like ?", "%V.jpeg").Scan(&listEntity)
}
liberr.ErrIsNil(ctx, err, "执行失败！")
sumNumber := len(listEntity)
if sumNumber == 0 {
err = errors.New("没有可操作的对象！")
liberr.ErrIsNil(ctx, err)
return
}
//2、下载数据到本地
p := coryCommon.GetCWD() + coryCommon.Uav
minioClient, err := dj.PublicMinioClient()
if err != nil {
return
}
for i, data := range listEntity {
err = minioClient.FGetObject(ctx, dj.BucketName, data.ObjectKey, p+data.ObjectKey, minio.GetObjectOptions{})
if err != nil {
err = errors.New("下载minio资源失败！" + err.Error())
liberr.ErrIsNil(ctx, err)
return
}
//3、获取总数，计算百分比，存储到redis
percentage, err := coryCommon.PercentageFunc(100, float64(sumNumber), float64(i+1))
if err != nil {
liberr.ErrIsNil(ctx, err)
return
}
g.Redis().Set(ctx, UavRedisKey+data.FlightId, percentage)
}
})
return
}
var mu sync.Mutex
// StoresBackMediaFilesToTheDatabaseFunc 存储回传媒体文件到数据库
func StoresBackMediaFilesToTheDatabaseFunc(ctx context.Context, entity *dj.FileUploadCallbackEntity) {
if entity == nil {
return
}
mu.Lock()
defer mu.Unlock()
//1、组装数据
extension := entity.Data.File.Extension
metadata := entity.Data.File.Metadata
dateTime, err := gtime.StrToTime(metadata.CreateTime)
if err != nil {
fmt.Println("转换错误:", err)
return
}
name := entity.Data.File.Name
suffix := strings.Split(name, ".")[1]
// 判断后缀是否是jpeg,不是就跳过
if suffix == "jpeg" {
dataInsert := do.ManageTaskRecordResource{
FlightId: extension.FlightID,
FileName: name,
ObjectKey: entity.Data.File.ObjectKey,
Path: entity.Data.File.Path,
DroneModelKey: extension.DroneModelKey,
PayloadModelKey: extension.PayloadModelKey,
AbsoluteAltitude: metadata.AbsoluteAltitude,
RelativeAltitude: metadata.RelativeAltitude,
CreateTime: dateTime,
ShootPosition: metadata.ShootPosition,
Lat: metadata.ShootPosition.Latitude,
Lng: metadata.ShootPosition.Longitude,
//Img: metadata.Img,
}
//防止有重复数据进入
count, _ := dao.ManageTaskRecordResource.Ctx(ctx).
Where(dao.ManageTaskRecordResource.Columns().FlightId, extension.FlightID).
Where(dao.ManageTaskRecordResource.Columns().FileName, entity.Data.File.Name).Count()
if count > 0 {
return
}
//2、初始化客户端
client, err := dj.PublicMinioClient()
if err != nil {
fmt.Println("获取元数据错误:", err)
return
}
//3、获取对象元数据
info, err := dj.StatObjectFunc(ctx, client, dj.BucketName, entity.Data.File.ObjectKey)
if err != nil {
fmt.Println("获取元数据错误:", err)
return
}
//4、插入数据
dataInsert.ImgSize = info.Size
_, err = dao.ManageTaskRecordResource.Ctx(ctx).Insert(dataInsert)
if err != nil {
fmt.Println("插入数据失败:", err)
return
}
}
}