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"
	model2 "cold-logistics/common/model"
	"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").Preload("CoolerBox").
		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 waybill model.Waybill
	err = e.Orm.Where("waybill_no = ?", c.WaybillNo).First(&waybill).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		err = errors.New("获取运单信息错误！")
		return
	}

	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
	}

	// 获取最后一个任务id
	var lastWaybillTask model.WaybillTask
	err = e.Orm.Model(&lastWaybillTask).Where("waybill_no = ?", c.WaybillNo).Last(&lastWaybillTask).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		err = errors.New("获取运单信息错误！")
		return
	}
	var afterStartTime bool
	var beforeEndTime bool
	if len(c.StartTime) > 0 {
		var st time.Time
		st, err = time.ParseInLocation("2006-01-02 15:04:05", c.StartTime, time.Local)
		if err != nil {
			err = errors.New("解析时间字符串出错！")
			return
		}
		fmt.Println("st", st)
		fmt.Println("task.StartTime", task.StartTime.Local())
		if task.StartTime.Local().After(st) || st.Truncate(time.Second).Equal(task.StartTime.Local().Truncate(time.Second)) {
			c.StartTime = task.StartTime.String()
		} else {
			afterStartTime = true
		}
	} else {
		c.StartTime = task.StartTime.String()
	}
	if len(c.EndTime) > 0 {
		var et time.Time
		et, err = time.ParseInLocation("2006-01-02 15:04:05", c.EndTime, time.Local)
		if err != nil {
			err = errors.New("解析时间字符串出错！")
			return
		}
		fmt.Println("et", et)
		fmt.Println("task.EndTime", task.EndTime.Local())
		if et.After(task.EndTime.Local()) || et.Truncate(time.Second).Equal(task.EndTime.Local().Truncate(time.Second)) {
			c.EndTime = task.EndTime.String()
		} else {
			beforeEndTime = true
		}
	} 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, 0, 9999)
	firstMap := map[int]nats_server.DeviceData_R{}
	lastMap := map[int]nats_server.DeviceData_R{}
	if count > 0 {
		for _, v := range c.T_ids {

			for i := 0; i < len(list); i++ {
				if afterStartTime {
					break
				}
				if v == list[i].T_id {
					if list[i].T_time != task.StartTime.String() {
						data := list[len(list)-1-i]
						data.T_time = task.StartTime.String()
						firstMap[v] = data
						count += 1
					}
					break
				}
			}

			for i := len(list) - 1; i >= 0; i-- {
				if beforeEndTime {
					break
				}
				if v == list[i].T_id {
					if list[i].T_time != task.EndTime.String() && !task.EndTime.Local().IsZero() {
						data := list[len(list)-1-i]
						data.T_time = task.EndTime.String()
						lastMap[v] = data
						count += 1
					}
					break
				}
			}

		}
	}

	for _, data := range firstMap {
		list = append(list, data)
	}

	for _, data := range lastMap {
		list = append(list, data)
	}

	sort.Slice(list, func(i, j int) bool {
		// 先按 T_time 字段排序，如果 T_time 相同则按 T_name 字段排序
		if list[i].T_time == list[j].T_time {
			return list[i].T_name < list[j].T_name
		}
		return list[i].T_time > list[j].T_time
	})

	// 计算总页数
	totalPages := (int(count) + c.PageSize - 1) / c.PageSize

	// 如果n超出范围，返回错误
	if c.Page > totalPages {
		return []nats_server.DeviceData_R{}, count, err
	}
	if c.Page < 1 {
		c.Page = 1
	}
	// 计算当前页的数据
	start := (c.Page - 1) * c.PageSize
	end := start + c.PageSize
	if end > int(count) {
		end = int(count)
	}
	currentPage := list[start:end]

	return currentPage, count, err
}
func (e *WaybillTask) GetPrintDataPage(c *dto.WaybillTaskGetDataPageReq) (list []nats_server.DeviceData_R, count int64, err error) {

	var waybill model.Waybill
	err = e.Orm.Where("waybill_no = ?", c.WaybillNo).First(&waybill).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		err = errors.New("获取运单信息错误！")
		return
	}

	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
	}

	// 获取最后一个任务id
	var lastWaybillTask model.WaybillTask
	err = e.Orm.Model(&lastWaybillTask).Where("waybill_no = ?", c.WaybillNo).Last(&lastWaybillTask).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		err = errors.New("获取运单信息错误！")
		return
	}
	var afterStartTime bool
	var beforeEndTime bool
	if len(c.StartTime) > 0 {
		var st time.Time
		st, err = time.ParseInLocation("2006-01-02 15:04:05", c.StartTime, time.Local)
		if err != nil {
			err = errors.New("解析时间字符串出错！")
			return
		}
		fmt.Println("st", st)
		fmt.Println("task.StartTime", task.StartTime.Local())
		if task.StartTime.Local().After(st) || st.Truncate(time.Second).Equal(task.StartTime.Local().Truncate(time.Second)) {
			c.StartTime = task.StartTime.String()
		} else {
			afterStartTime = true
		}
	} else {
		c.StartTime = task.StartTime.String()
	}
	if len(c.EndTime) > 0 {
		var et time.Time
		et, err = time.ParseInLocation("2006-01-02 15:04:05", c.EndTime, time.Local)
		if err != nil {
			err = errors.New("解析时间字符串出错！")
			return
		}
		fmt.Println("et", et)
		fmt.Println("task.EndTime", task.EndTime.Local())
		if et.After(task.EndTime.Local()) || et.Truncate(time.Second).Equal(task.EndTime.Local().Truncate(time.Second)) {
			c.EndTime = task.EndTime.String()
		} else {
			beforeEndTime = true
		}
	} 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, 0, 9999)
	firstMap := map[int]nats_server.DeviceData_R{}
	lastMap := map[int]nats_server.DeviceData_R{}
	if count > 0 {
		for _, v := range c.T_ids {

			for i := 0; i < len(list); i++ {
				if afterStartTime {
					break
				}
				if v == list[i].T_id {
					if list[i].T_time != task.StartTime.String() {
						data := list[len(list)-1-i]
						data.T_time = task.StartTime.String()
						firstMap[v] = data
						count += 1
					}
					break
				}
			}

			for i := len(list) - 1; i >= 0; i-- {
				if beforeEndTime {
					break
				}
				if v == list[i].T_id {
					if list[i].T_time != task.EndTime.String() && !task.EndTime.Local().IsZero() {
						data := list[len(list)-1-i]
						data.T_time = task.EndTime.String()
						lastMap[v] = data
						count += 1
					}
					break
				}
			}

		}
	}

	for _, data := range firstMap {
		list = append(list, data)
	}

	for _, data := range lastMap {
		list = append(list, data)
	}

	sort.Slice(list, func(i, j int) bool {
		// 先按 T_time 字段排序，如果 T_time 相同则按 T_name 字段排序
		if list[i].T_time == list[j].T_time {
			return list[i].T_name < list[j].T_name
		}
		return list[i].T_time > list[j].T_time
	})

	// 计算总页数
	totalPages := (int(count) + c.PageSize - 1) / c.PageSize

	// 如果n超出范围，返回错误
	if c.Page > totalPages {
		return []nats_server.DeviceData_R{}, count, err
	}
	if c.Page < 1 {
		c.Page = 1
	}
	// 计算当前页的数据
	start := (c.Page - 1) * c.PageSize
	end := start + c.PageSize
	if end > int(count) {
		end = int(count)
	}
	currentPage := list[start:end]

	return currentPage, count, err
}
func (e *WaybillTask) GetNewestDataPage(c *dto.WaybillTaskGetNewestDataPageReq) (data nats_server.DeviceData_R, err error) {
	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)
		err = errors.New("获取运单信息错误！")
		return
	}
	// 获取公司秘钥
	var company model.SysDept
	company, err = model.GetCompanyById(waybill.DeptId)
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		err = model.GetCompanyKeyErr
		return
	}

	var taskList []model.WaybillTask
	err = e.Orm.Where("waybill_no = ?", c.WaybillNo).Order("id desc").Find(&taskList).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		err = errors.New("获取运单信息任务失败！")
		return
	}
	if len(taskList) == 0 {
		err = errors.New("暂无温湿度记录信息！")
		return
	}
	var listAll []nats_server.DeviceData_R
	for _, task := range taskList {
		deviceSensorList, _, _ := nats_server.Cold_CompanyDeviceSensor_List_ByKey(task.Sn, company.ColdKey)
		if len(deviceSensorList) == 0 {
			continue
		}
		T_snid := fmt.Sprintf("%s,%d|", deviceSensorList[0].T_sn, deviceSensorList[0].T_id)
		if len(task.EndTime.String()) == 0 {
			task.EndTime = model2.Time(time.Now())
		}
		// 获取传感器信息
		list, _, _ := nats_server.Cold_ReadDeviceDataListBy_T_snid(T_snid, task.StartTime.String(), task.EndTime.String(), 0, 9999)
		if len(list) > 0 {
			listAll = append(listAll, list...)
			break
		}
	}

	if len(listAll) > 0 {
		sort.Slice(listAll, func(i, j int) bool {
			return listAll[i].T_time > listAll[j].T_time
		})
		data = listAll[0]
		return
	}
	err = errors.New("暂无温湿度记录信息！")
	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 or cooler_box_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
}