cold-logistics/app/admin/service/waybill_task.go

package service

import (
	"cold-logistics/app/admin/model"
	"cold-logistics/app/admin/service/dto"
	cDto "cold-logistics/common/dto"
	"cold-logistics/common/global"
	"cold-logistics/common/nats/nats_server"
	"errors"
	"fmt"
	"gogs.baozhida.cn/zoie/OAuth-core/service"
	"gorm.io/gorm"
	"sort"
	"time"
)

type WaybillTask struct {
	service.Service
}

func WaybillTaskTimeScopes(startTime, endTime string) func(db *gorm.DB) *gorm.DB {
	return func(db *gorm.DB) *gorm.DB {
		if len(startTime) == 0 && len(endTime) == 0 {
			return db
		}
		return db.Where("(start_time between ? and ?) or (end_time between ? and ?)",
			startTime, endTime, startTime, endTime)
	}
}

// GetPage 获取WaybillTask列表
func (e *WaybillTask) GetPage(c *dto.WaybillTaskGetPageReq, list *[]model.WaybillTask, count *int64) error {
	var err error
	var data model.WaybillTask
	var waybill model.Waybill
	err = e.Orm.Model(&waybill).Where("waybill_no = ?", c.WaybillNo).First(&waybill).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return errors.New("获取运单信息失败")
	}

	err = e.Orm.Model(&data).
		Scopes(
			cDto.MakeCondition(c.GetNeedSearch()),
		).
		Preload("Warehouse").Preload("Car").
		Find(list).Limit(-1).Offset(-1).
		Count(count).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return global.GetFailedErr
	}
	// 获取公司秘钥
	var company model.SysDept
	company, err = model.GetCompanyById(waybill.DeptId)
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return model.GetCompanyKeyErr
	}
	for i := 0; i < len(*list); i++ {
		// 获取传感器信息
		deviceSensorList, _, _ := nats_server.Cold_CompanyDeviceSensor_List_ByKey((*list)[i].Sn, company.ColdKey)
		(*list)[i].DeviceSensorList = deviceSensorList
	}
	return nil
}

// GetPage 获取WaybillTask列表
func (e *WaybillTask) GetDataPage(c *dto.WaybillTaskGetDataPageReq) (list []nats_server.DeviceData_R, count int64, err error) {
	var task model.WaybillTask
	err = e.Orm.Where("id = ? and waybill_no = ?", c.TaskId, c.WaybillNo).First(&task).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		err = errors.New("获取运单信息错误！")
		return
	}

	if len(c.StartTime) > 0 {
		var st time.Time
		st, err = time.Parse("2006-01-02 15:04:05", c.StartTime)
		if err != nil {
			err = errors.New("解析时间字符串出错！")
			return
		}
		if st.Before(time.Time(task.StartTime)) {
			c.StartTime = task.StartTime.String()
		}
	} else {
		c.StartTime = task.StartTime.String()
	}
	if len(c.EndTime) > 0 {
		var et time.Time
		et, err = time.Parse("2006-01-02 15:04:05", c.EndTime)
		if err != nil {
			err = errors.New("解析时间字符串出错！")
			return
		}
		if et.After(time.Time(task.EndTime)) {
			c.EndTime = task.EndTime.String()
		}
	} else {
		c.EndTime = task.EndTime.String()
	}

	T_snid := ""
	for _, id := range c.T_ids {
		T_snid += fmt.Sprintf("%s,%d|", task.Sn, id)
	}
	// 获取传感器信息
	list, count, err = nats_server.Cold_ReadDeviceDataListBy_T_snid(T_snid, c.StartTime, c.EndTime, c.Page, c.PageSize)
	return
}
func (e *WaybillTask) GetLocus(c *dto.WaybillGetLocusReq) ([]nats_server.DeviceData_R2, error) {
	var err error
	var data model.WaybillTask
	var waybill model.Waybill
	var taskList []model.WaybillTask
	locusList := make([]nats_server.DeviceData_R2, 0)

	err = e.Orm.Model(&waybill).Where("waybill_no = ?", c.WaybillNo).First(&waybill).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return locusList, errors.New("获取运单信息失败")
	}
	// 获取公司秘钥
	var company model.SysDept
	company, err = model.GetCompanyById(waybill.DeptId)
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return locusList, model.GetCompanyKeyErr
	}

	err = e.Orm.Model(&data).
		Scopes(
			cDto.MakeCondition(c.GetNeedSearch()),
		).Where("car_id > 0").
		Find(&taskList).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return locusList, global.GetFailedErr
	}

	for i := 0; i < len(taskList); i++ {
		// 获取传感器信息
		deviceSensorList, _, _ := nats_server.Cold_CompanyDeviceSensor_List_ByKey(taskList[i].Sn, company.ColdKey)
		if len(deviceSensorList) > 0 {
			T_snid := fmt.Sprintf("%s,%d|", taskList[i].Sn, deviceSensorList[0].T_id)
			dataList, _, err := nats_server.Cold_ReadDeviceDataListBy_T_snidForLocus(T_snid, taskList[i].StartTime.String(), taskList[i].EndTime.String(), 0, 9999)
			if err != nil {
				e.Log.Errorf("nats 获取轨迹信息失败: %s", err)
				return locusList, global.GetFailedErr
			}

			// 倒序
			sort.Slice(dataList, func(i, j int) bool {
				if dataList[i].T_time < dataList[j].T_time {
					return true
				}
				return false
			})
			locusList = append(locusList, dataList...)

		}

	}
	return locusList, nil
}