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

package service

import (
	"cold-logistics/app/admin/model"
	"cold-logistics/app/admin/service/dto"
	"cold-logistics/common/actions"
	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/pkg/utils"
	"gogs.baozhida.cn/zoie/OAuth-core/service"
	"gorm.io/gorm"
	"sort"
	"strconv"
	"strings"
	"time"
)

type Waybill struct {
	service.Service
}

func WaybillCustomerStatusScopes(status int) func(db *gorm.DB) *gorm.DB {
	return func(db *gorm.DB) *gorm.DB {
		// 未发货
		if status == 1 {
			statusList := []int{
				model.WaybillStatusWaitDelivery,
				model.WaybillStatusTruck,
				model.WaybillStatusWaitStorage,
			}
			return db.Where("status in (?)", statusList)
		}
		// 已发货
		if status == 2 {
			statusList := []int{
				model.WaybillStatusTruck,
				model.WaybillStatusStorage,
				model.WaybillStatusTruckOut,
				model.WaybillStatusStorageOut,
			}
			return db.Where("status in (?)", statusList)
		}
		// 已签收
		if status == 3 {
			statusList := []int{
				model.WaybillStatusReceipt,
			}
			return db.Where("status in (?)", statusList)
		}
		return db
	}

}

// GetPage 获取Waybill列表
func (e *Waybill) GetPage(c *dto.WaybillGetPageReq, list *[]model.Waybill, count *int64, p *actions.DataPermission) error {
	var err error
	var data model.Waybill
	if c.PageSize == 9999 {
		err = e.Orm.Model(&data).
			Scopes(
				cDto.MakeCondition(c.GetNeedSearch()),
				actions.Permission(data.TableName(), p),
			).
			Find(list).Limit(-1).Offset(-1).
			Count(count).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return global.GetFailedErr
		}
		return nil
	}

	err = e.Orm.Model(&data).
		Scopes(
			cDto.MakeCondition(c.GetNeedSearch()),
			cDto.Paginate(c.GetPageSize(), c.GetPageIndex()),
			actions.Permission(data.TableName(), p),
		).
		Preload("PrintUser").
		Find(list).Limit(-1).Offset(-1).
		Count(count).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return global.GetFailedErr
	}
	return nil
}
func (e *Waybill) GetAppletPage(c *dto.WaybillGetAppletPageReq, list *[]model.Waybill, count *int64, p *actions.DataPermission) error {
	var err error
	//var data model.Waybill
	var logistics model.WaybillLogistics
	err = e.Orm.Table("waybill").
		Select("waybill.*,waybill_logistics.status as status").
		Scopes(
			cDto.MakeCondition(c.GetNeedSearch()),
			cDto.Paginate(c.GetPageSize(), c.GetPageIndex()),
			actions.Permission(logistics.TableName(), p)).
		Where("waybill_logistics.id in (SELECT MAX(id) FROM waybill_logistics where user_id = ? group by waybill_no )", p.UserId).
		Joins("left join waybill_logistics on waybill.waybill_no = waybill_logistics.waybill_no").
		Find(&list).Limit(-1).Offset(-1).Count(count).Error

	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return global.GetFailedErr
	}
	return nil
}

func (e *Waybill) GetAppletCount(list *[]model.Waybill, count *int64, p *actions.DataPermission) error {
	var err error
	//var data model.Waybill
	var logistics model.WaybillLogistics
	err = e.Orm.Table("waybill").
		Select("waybill.*,waybill_logistics.status as status").
		Scopes(
			actions.Permission(logistics.TableName(), p)).
		Where("waybill_logistics.id in (SELECT MAX(id) FROM waybill_logistics where user_id = ? group by waybill_no )", p.UserId).
		Joins("left join waybill_logistics on waybill.waybill_no = waybill_logistics.waybill_no").
		Find(list).Count(count).Error

	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return global.GetFailedErr
	}
	return nil
}

func (e *Waybill) GetCustomerPage(c *dto.WaybillGetCustomerPageReq, list *[]model.Waybill, count *int64, p *actions.DataPermission) error {
	var err error
	var data model.Waybill

	if c.PageSize == 9999 {
		err = e.Orm.Model(&data).
			Scopes(
				WaybillCustomerStatusScopes(c.Status),
				cDto.MakeCondition(c.GetNeedSearch()),
				actions.Permission(data.TableName(), p),
			).
			Find(list).Limit(-1).Offset(-1).
			Count(count).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return global.GetFailedErr
		}
		return nil
	}

	err = e.Orm.Model(&data).
		Scopes(
			WaybillCustomerStatusScopes(c.Status),
			cDto.MakeCondition(c.GetNeedSearch()),
			cDto.Paginate(c.GetPageSize(), c.GetPageIndex()),
			actions.Permission(data.TableName(), p),
		).
		Find(list).Limit(-1).Offset(-1).
		Count(count).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return global.GetFailedErr
	}
	return nil
}

// Get 获取Waybill对象
func (e *Waybill) Get(d *dto.WaybillGetReq, waybillModel *model.Waybill, p *actions.DataPermission) error {
	err := e.Orm.
		Scopes(actions.Permission(waybillModel.TableName(), p)).
		Preload("User").
		First(waybillModel, d.GetId()).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return global.GetNotFoundOrNoPermissionErr
		}
		return global.GetFailedErr
	}

	return nil
}
func (e *Waybill) GetByWaybillNo(d *dto.WaybillGetByWaybillNoReq, waybillModel *model.Waybill, p *actions.DataPermission) error {
	err := e.Orm.
		Scopes(actions.Permission(waybillModel.TableName(), p)).
		Where("waybill_no = ?", d.WaybillNo).
		First(waybillModel).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return global.GetNotFoundOrNoPermissionErr
		}
		return global.GetFailedErr
	}

	return nil
}

// Insert 创建Waybill对象
func (e *Waybill) Insert(c *dto.WaybillInsertReq) error {
	var err error
	var data model.Waybill

	tx := e.Orm.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()

	var no string
	for {
		no = time.Now().Format("200601021504") + utils.GetRandString(6, "0123456789", 0)
		var j int64
		err = e.Orm.Model(&data).Where("waybill_no = ?", no).Count(&j).Error
		if err != nil {
			continue
		}
		if j == 0 {
			break
		}
	}

	// 添加运单
	data.WaybillNo = no
	c.Generate(&data)
	err = tx.Create(&data).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return global.CreateFailedErr
	}
	c.Id = data.Id

	return nil

}

// AppletInsert 员工添加运单
func (e *Waybill) AppletInsert(c *dto.WaybillInsertReq, p *actions.DataPermission) error {
	var err error
	var data model.Waybill

	tx := e.Orm.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()

	var userModel = model.SysUser{}
	// 查询运单是否存在
	err = tx.Scopes(actions.Permission(userModel.TableName(), p)).
		First(&userModel, p.UserId).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return global.GetNotFoundErr
		}
		return global.CreateFailedErr
	}
	if userModel.Type != model.SysUserTypeDriver && userModel.Type != model.SysUserTypeWarehouse {
		err = errors.New("无权添加！")
		return err
	}
	var status = model.WaybillStatusWaitDelivery
	var car = model.Car{}
	if userModel.Type == model.SysUserTypeDriver {
		status = model.WaybillStatusWaitTruck
		// 查询车辆库信息
		err = tx.Scopes(actions.Permission(car.TableName(), p)).
			Where("user_id = ?", p.UserId).
			First(&car).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New("获取车辆信息失败")
		}
	}

	var warehouse = model.Warehouse{}
	if userModel.Type == model.SysUserTypeWarehouse {
		status = model.WaybillStatusWaitStorage
		// 查询仓库信息
		err = tx.Scopes(actions.Permission(warehouse.TableName(), p)).
			Where("user_id = ?", p.UserId).
			First(&warehouse).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New("获取仓库信息失败")
		}
	}
	data.PrintUserId = p.UserId

	var no string
	for {
		no = time.Now().Format("200601021504") + utils.GetRandString(6, "0123456789", 0)
		var j int64
		err = e.Orm.Model(&data).Where("waybill_no = ?", no).Count(&j).Error
		if err != nil {
			continue
		}
		if j == 0 {
			break
		}
	}

	// 添加运单
	data.DeptId = p.DeptId
	data.CreateBy = p.UserId
	data.WaybillNo = no
	c.Generate(&data)
	data.Status = status
	err = tx.Create(&data).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return global.CreateFailedErr
	}
	c.Id = data.Id

	// 添加物流
	Logistics := model.WaybillLogistics{
		WaybillNo:   data.WaybillNo,
		Status:      data.Status,
		CarId:       car.Id,
		WarehouseId: warehouse.Id,
		UserId:      p.UserId,
		ControlBy: model2.ControlBy{
			CreateBy: p.UserId,
		},
		DeptBy: model2.DeptBy{
			DeptId: p.DeptId,
		},
	}
	err = tx.Create(&Logistics).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return errors.New(fmt.Sprintf("保存运单物流信息失败：%s", err))
	}

	return nil

}

// Update 修改Waybill对象
func (e *Waybill) Update(c *dto.WaybillUpdateReq, p *actions.DataPermission) error {
	var err error

	tx := e.Orm.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()

	var waybillModel = model.Waybill{}
	// 查询运单是否存在
	err = e.Orm.Scopes(actions.Permission(waybillModel.TableName(), p)).
		First(&waybillModel, c.GetId()).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return global.UpdateNotFoundOrNoPermissionErr
		}
		return global.UpdateFailedErr
	}

	c.Generate(&waybillModel)
	err = tx.Save(&waybillModel).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return global.UpdateFailedErr
	}

	c.Id = waybillModel.Id
	return nil
}
func (e *Waybill) Delivery(c *dto.WaybillDeliveryReq, p *actions.DataPermission) error {
	var err error

	tx := e.Orm.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()

	for _, id := range c.WaybillIds {

		var waybillModel = model.Waybill{}
		// 查询运单是否存在
		err = tx.Scopes(actions.Permission(waybillModel.TableName(), p)).
			First(&waybillModel, id).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			if errors.Is(err, gorm.ErrRecordNotFound) {
				return global.UpdateNotFoundOrNoPermissionErr
			}
			return global.UpdateFailedErr
		}
		if waybillModel.Status != model.WaybillStatusWaitDelivery &&
			waybillModel.Status != model.WaybillStatusWaitTruck &&
			waybillModel.Status != model.WaybillStatusWaitStorage {
			err = errors.New(fmt.Sprintf("运单状态为%s，禁止操作！", model.WaybillStatusMap[waybillModel.Status]))
			return err
		}

		var car = model.Car{}
		if c.Type == model.SysUserTypeDriver {
			waybillModel.Status = model.WaybillStatusWaitTruck
			// 查询车辆库信息
			err = tx.Scopes(actions.Permission(car.TableName(), p)).
				Where("user_id = ?", c.PrintUserId).
				First(&car).Error
			if err != nil {
				e.Log.Errorf("db error: %s", err)
				return errors.New("获取车辆信息失败")
			}
		}

		var warehouse = model.Warehouse{}
		if c.Type == model.SysUserTypeWarehouse {
			waybillModel.Status = model.WaybillStatusWaitStorage
			// 查询仓库信息
			err = tx.Scopes(actions.Permission(warehouse.TableName(), p)).
				Where("user_id = ?", c.PrintUserId).
				First(&warehouse).Error
			if err != nil {
				e.Log.Errorf("db error: %s", err)
				return errors.New("获取仓库信息失败")
			}
		}
		waybillModel.PrintUserId = c.PrintUserId
		err = tx.Save(&waybillModel).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return global.UpdateFailedErr
		}

		// 添加物流
		Logistics := model.WaybillLogistics{
			WaybillNo:   waybillModel.WaybillNo,
			Status:      waybillModel.Status,
			CarId:       car.Id,
			WarehouseId: warehouse.Id,
			UserId:      c.PrintUserId,
			ControlBy: model2.ControlBy{
				CreateBy: p.UserId,
			},
			DeptBy: model2.DeptBy{
				DeptId: p.DeptId,
			},
		}
		err = tx.Create(&Logistics).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单物流信息失败：%s", err))
		}

	}

	return nil
}

// Remove 删除Waybill
func (e *Waybill) Remove(c *dto.WaybillDeleteReq, p *actions.DataPermission) error {
	var err error

	tx := e.Orm.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()

	var waybillModel model.Waybill

	// 查询运单是否存在
	err = e.Orm.Scopes(actions.Permission(waybillModel.TableName(), p)).
		First(&waybillModel, c.GetId()).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return global.DeleteNotFoundOrNoPermissionErr
		}
		return global.DeleteFailedErr
	}
	if waybillModel.Status != model.WaybillStatusWaitDelivery && waybillModel.Status != model.WaybillStatusWaitTruck && waybillModel.Status != model.WaybillStatusWaitStorage {
		return errors.New(fmt.Sprintf("运单状态为%s，禁止删除", model.WaybillStatusMap[waybillModel.Status]))
	}

	db := tx.Delete(&waybillModel)

	if err = db.Error; err != nil {
		e.Log.Errorf("db error: %s", err)
		return global.DeleteFailedErr
	}
	if db.RowsAffected == 0 {
		return global.DeleteNotFoundOrNoPermissionErr
	}

	return nil
}

func (e *Waybill) WarehouseIn(c *dto.WaybillInOutReq, p *actions.DataPermission) error {
	var err error

	tx := e.Orm.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()
	// 查询仓库信息
	var warehouse = model.Warehouse{}
	// 查询运单是否存在
	err = tx.Scopes(actions.Permission(warehouse.TableName(), p)).
		Where("user_id = ?", p.UserId).
		First(&warehouse).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return errors.New("获取仓库信息失败")
	}

	for _, waybillNo := range c.WaybillNoList {
		var waybillModel = model.Waybill{}
		// 查询运单是否存在
		err = tx.Scopes(actions.Permission(waybillModel.TableName(), p)).
			Where("waybill_no = ?", waybillNo).
			First(&waybillModel).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			if errors.Is(err, gorm.ErrRecordNotFound) {
				return errors.New(fmt.Sprintf("运单号%s不存在", waybillNo))
			}
			return errors.New(fmt.Sprintf("运单号%s查询失败", waybillNo))
		}
		if waybillModel.WarehouseId == warehouse.Id && waybillModel.Status == model.WaybillStatusStorage {
			continue
		}
		if waybillModel.WarehouseId != warehouse.Id && waybillModel.Status == model.WaybillStatusStorage {
			err = errors.New(fmt.Sprintf("运单号%s状态为%s，无法入库！", waybillNo, model.WaybillStatusMap[waybillModel.Status]))
			return err
		}
		if waybillModel.Status == model.WaybillStatusWaitTruck {
			// 待装车
			err = errors.New(fmt.Sprintf("运单号%s状态为%s，无法入库！", waybillNo, model.WaybillStatusMap[waybillModel.Status]))
			return err
		}
		// 验证时间
		var logistics model.WaybillLogistics
		err = e.Orm.Where("waybill_no = ?", waybillNo).Last(&logistics).Error
		if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			e.Log.Errorf("获取运单日志失败: %s", err)
			return err
		}
		if time.Time(logistics.CreatedAt).After(time.Time(c.StartTime)) {
			err = errors.New(fmt.Sprintf("运单号%s装车时间[%s]与上一阶段时间[%s]冲突，无法入库！", waybillNo, c.StartTime.String(), logistics.CreatedAt.String()))
			return err
		}

		var lng, lat string
		lng, lat, err = e.GetSite(p.DeptId, warehouse.Sn, c.StartTime.String())
		if err != nil {
			e.Log.Errorf("获取定位信息失败: %s", err)
			return err
		}

		if waybillModel.Status == model.WaybillStatusTruck {
			// 已装车
			var car model.Car
			err = e.Orm.First(&car, waybillModel.CarId).Error
			if err != nil {
				e.Log.Errorf("db error: %s", err)
				return errors.New(fmt.Sprintf("查询仓库信息失败：%s", err))
			}
			// 添加物流
			Logistics := model.WaybillLogistics{
				WaybillNo: waybillNo,
				Status:    model.WaybillStatusTruckOut,
				CarId:     waybillModel.CarId,
				UserId:    car.UserId,
				Lng:       lng,
				Lat:       lat,
				ControlBy: model2.ControlBy{
					CreateBy: p.UserId,
				},
				DeptBy: model2.DeptBy{
					DeptId: car.DeptId,
				},
				ModelTime: model2.ModelTime{
					CreatedAt: c.StartTime,
				},
			}
			err = tx.Create(&Logistics).Error
			if err != nil {
				e.Log.Errorf("db error: %s", err)
				return errors.New(fmt.Sprintf("保存运单物流信息失败：%s", err))
			}

			// 查询任务
			var task model.WaybillTask
			err = tx.Model(&task).Where("waybill_no = ? and car_id = ?", waybillNo, waybillModel.CarId).
				Last(&task).Error
			if err != nil {
				e.Log.Errorf("db error: %s", err)
				return errors.New(fmt.Sprintf("查询运单任务信息失败：%s", err))
			}
			task.EndTime = c.StartTime
			task.UpdateBy = p.UserId
			err = tx.Save(&task).Error
			if err != nil {
				e.Log.Errorf("db error: %s", err)
				return errors.New(fmt.Sprintf("保存运单任务信息失败：%s", err))
			}
		}
		if waybillModel.Status == model.WaybillStatusWaitStorage {
			waybillModel.DeliveryTime = c.StartTime
		}
		waybillModel.Status = model.WaybillStatusStorage
		waybillModel.WarehouseId = warehouse.Id
		err = tx.Save(&waybillModel).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单信息失败：%s", err))
		}

		// 添加物流
		Logistics := model.WaybillLogistics{
			WaybillNo:   waybillNo,
			Status:      model.WaybillStatusStorage,
			WarehouseId: warehouse.Id,
			UserId:      p.UserId,
			Lng:         lng,
			Lat:         lat,
			ControlBy: model2.ControlBy{
				CreateBy: p.UserId,
			},
			DeptBy: model2.DeptBy{
				DeptId: p.DeptId,
			},
			ModelTime: model2.ModelTime{
				CreatedAt: c.StartTime,
			},
		}
		err = tx.Create(&Logistics).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单物流信息失败：%s", err))
		}

		// 添加任务
		task := model.WaybillTask{
			WaybillNo:   waybillNo,
			WarehouseId: warehouse.Id,
			UserId:      p.UserId,
			Sn:          warehouse.Sn,
			StartTime:   c.StartTime,
			ControlBy: model2.ControlBy{
				CreateBy: p.UserId,
			},
			DeptBy: model2.DeptBy{
				DeptId: p.DeptId,
			},
		}
		err = tx.Create(&task).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单任务信息失败：%s", err))
		}
	}

	return nil
}
func (e *Waybill) WarehouseOut(c *dto.WaybillInOutReq, p *actions.DataPermission) error {
	var err error

	tx := e.Orm.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()
	// 查询仓库信息
	var warehouse = model.Warehouse{}
	// 查询运单是否存在
	err = tx.Scopes(actions.Permission(warehouse.TableName(), p)).
		Where("user_id = ?", p.UserId).
		First(&warehouse).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return errors.New("获取仓库信息失败")
	}

	for _, waybillNo := range c.WaybillNoList {
		var waybillModel = model.Waybill{}
		// 查询运单是否存在
		err = tx.Scopes(actions.Permission(waybillModel.TableName(), p)).
			Where("waybill_no = ?", waybillNo).
			First(&waybillModel).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			if errors.Is(err, gorm.ErrRecordNotFound) {
				return errors.New(fmt.Sprintf("运单号%s不存在", waybillNo))
			}
			return errors.New(fmt.Sprintf("运单号%s查询失败", waybillNo))
		}

		if waybillModel.WarehouseId != warehouse.Id {
			return errors.New("请选择正确仓库出库！")
		}

		if waybillModel.WarehouseId == warehouse.Id && waybillModel.Status == model.WaybillStatusStorageOut {
			continue
		}
		if waybillModel.Status != model.WaybillStatusStorage {
			err = errors.New(fmt.Sprintf("运单号%s状态为%s，无法出库！", waybillNo, model.WaybillStatusMap[waybillModel.Status]))
			return err
		}

		waybillModel.Status = model.WaybillStatusStorageOut
		waybillModel.WarehouseId = warehouse.Id
		err = tx.Save(&waybillModel).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单信息失败：%s", err))
		}

		var lng, lat string
		lng, lat, err = e.GetSite(p.DeptId, warehouse.Sn, c.StartTime.String())
		if err != nil {
			e.Log.Errorf("获取定位信息失败: %s", err)
			return err
		}

		// 添加物流
		Logistics := model.WaybillLogistics{
			WaybillNo:   waybillNo,
			Status:      model.WaybillStatusStorageOut,
			WarehouseId: warehouse.Id,
			UserId:      p.UserId,
			Lng:         lng,
			Lat:         lat,
			ControlBy: model2.ControlBy{
				CreateBy: p.UserId,
			},
			DeptBy: model2.DeptBy{
				DeptId: p.DeptId,
			},
			ModelTime: model2.ModelTime{
				CreatedAt: c.StartTime,
			},
		}
		err = tx.Create(&Logistics).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单物流信息失败：%s", err))
		}

		// 查询任务
		var task model.WaybillTask
		err = tx.Model(&task).Where("waybill_no = ? and warehouse_id = ?", waybillNo, warehouse.Id).
			First(&task).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("查询运单任务信息失败：%s", err))
		}
		task.EndTime = c.StartTime
		task.UpdateBy = p.UserId
		err = tx.Save(&task).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单任务信息失败：%s", err))
		}
	}

	return nil
}
func (e *Waybill) CarIn(c *dto.WaybillInOutReq, p *actions.DataPermission) error {
	var err error

	tx := e.Orm.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()
	// 查询车辆信息
	var car = model.Car{}
	// 查询运单是否存在
	err = tx.Scopes(actions.Permission(car.TableName(), p)).
		Where("user_id = ?", p.UserId).
		First(&car).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return errors.New("获取车辆绑定信息失败")
	}

	for _, waybillNo := range c.WaybillNoList {
		var waybillModel = model.Waybill{}
		// 查询运单是否存在
		err = tx.Scopes(actions.Permission(waybillModel.TableName(), p)).
			Where("waybill_no = ?", waybillNo).
			First(&waybillModel).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			if errors.Is(err, gorm.ErrRecordNotFound) {
				return errors.New(fmt.Sprintf("运单号%s不存在", waybillNo))
			}
			return errors.New(fmt.Sprintf("运单号%s查询失败", waybillNo))
		}
		if waybillModel.CarId == car.Id && waybillModel.Status == model.WaybillStatusTruck {
			continue
		}
		if waybillModel.CarId != car.Id && waybillModel.Status == model.WaybillStatusTruck {
			err = errors.New(fmt.Sprintf("运单号%s状态为%s，无法装车！", waybillNo, model.WaybillStatusMap[waybillModel.Status]))
			return err
		}
		if waybillModel.Status == model.WaybillStatusWaitStorage {
			err = errors.New(fmt.Sprintf("运单号%s状态为%s，无法装车！", waybillNo, model.WaybillStatusMap[waybillModel.Status]))
			return err
		}

		// 验证时间
		var logistics model.WaybillLogistics
		err = e.Orm.Where("waybill_no = ?", waybillNo).Last(&logistics).Error
		if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			e.Log.Errorf("获取运单日志失败: %s", err)
			return err
		}
		if time.Time(logistics.CreatedAt).After(time.Time(c.StartTime)) {
			err = errors.New(fmt.Sprintf("运单号%s入库时间[%s]与上一阶段时间[%s]冲突，无法入库！", waybillNo, c.StartTime.String(), logistics.CreatedAt.String()))
			return err
		}

		if waybillModel.Status == model.WaybillStatusWaitTruck {
			waybillModel.DeliveryTime = c.StartTime
		}

		var lng, lat string
		lng, lat, err = e.GetSite(p.DeptId, car.Sn, c.StartTime.String())
		if err != nil {
			e.Log.Errorf("获取定位信息失败: %s", err)
			return err
		}
		// 如果上一个状态为入库，则自动出库
		if waybillModel.Status == model.WaybillStatusStorage {
			var warehouse model.Warehouse
			err = e.Orm.First(&warehouse, waybillModel.WarehouseId).Error
			if err != nil {
				e.Log.Errorf("db error: %s", err)
				return errors.New(fmt.Sprintf("查询仓库信息失败：%s", err))
			}
			// 添加物流
			Logistics := model.WaybillLogistics{
				WaybillNo:   waybillNo,
				Status:      model.WaybillStatusStorageOut,
				WarehouseId: waybillModel.WarehouseId,
				UserId:      warehouse.UserId,
				Lng:         lng,
				Lat:         lat,
				ControlBy: model2.ControlBy{
					CreateBy: p.UserId,
				},
				DeptBy: model2.DeptBy{
					DeptId: warehouse.DeptId,
				},
				ModelTime: model2.ModelTime{
					CreatedAt: c.StartTime,
				},
			}
			err = tx.Create(&Logistics).Error
			if err != nil {
				e.Log.Errorf("db error: %s", err)
				return errors.New(fmt.Sprintf("保存运单物流信息失败：%s", err))
			}

			// 查询任务
			var task model.WaybillTask
			err = tx.Model(&task).Where("waybill_no = ? and warehouse_id = ?", waybillNo, waybillModel.WarehouseId).
				Last(&task).Error
			if err != nil {
				e.Log.Errorf("db error: %s", err)
				return errors.New(fmt.Sprintf("查询运单任务信息失败：%s", err))
			}
			task.EndTime = c.StartTime
			task.UpdateBy = p.UserId
			err = tx.Save(&task).Error
			if err != nil {
				e.Log.Errorf("db error: %s", err)
				return errors.New(fmt.Sprintf("保存运单任务信息失败：%s", err))
			}
		}

		waybillModel.Status = model.WaybillStatusTruck
		waybillModel.CarId = car.Id

		err = tx.Save(&waybillModel).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单信息失败：%s", err))
		}

		// 添加物流
		Logistics := model.WaybillLogistics{
			WaybillNo: waybillNo,
			Status:    model.WaybillStatusTruck,
			CarId:     car.Id,
			UserId:    p.UserId,
			Lng:       lng,
			Lat:       lat,
			ControlBy: model2.ControlBy{
				CreateBy: p.UserId,
			},
			DeptBy: model2.DeptBy{
				DeptId: p.DeptId,
			},
			ModelTime: model2.ModelTime{
				CreatedAt: c.StartTime,
			},
		}
		err = tx.Create(&Logistics).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单物流信息失败：%s", err))
		}

		// 添加任务
		task := model.WaybillTask{
			WaybillNo: waybillNo,
			CarId:     car.Id,
			UserId:    p.UserId,
			Sn:        car.Sn,
			StartTime: c.StartTime,
			ControlBy: model2.ControlBy{
				CreateBy: p.UserId,
			},
			DeptBy: model2.DeptBy{
				DeptId: p.DeptId,
			},
		}
		err = tx.Create(&task).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单任务信息失败：%s", err))
		}
	}

	return nil
}
func (e *Waybill) CarOut(c *dto.WaybillInOutReq, p *actions.DataPermission) error {
	var err error

	tx := e.Orm.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()
	// 查询仓库信息
	var car = model.Car{}
	// 查询运单是否存在
	err = tx.Scopes(actions.Permission(car.TableName(), p)).
		Where("user_id = ?", p.UserId).
		First(&car).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return errors.New("获取车辆绑定信息失败")
	}

	for _, waybillNo := range c.WaybillNoList {
		var waybillModel = model.Waybill{}
		// 查询运单是否存在
		err = tx.Scopes(actions.Permission(waybillModel.TableName(), p)).
			Where("waybill_no = ?", waybillNo).
			First(&waybillModel).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			if errors.Is(err, gorm.ErrRecordNotFound) {
				return errors.New(fmt.Sprintf("运单号%s不存在", waybillNo))
			}
			return errors.New(fmt.Sprintf("运单号%s查询失败", waybillNo))
		}

		if waybillModel.CarId != car.Id {
			return errors.New("请选择正确车辆下车！")
		}

		if waybillModel.CarId == car.Id && waybillModel.Status == model.WaybillStatusTruckOut {
			continue
		}
		if waybillModel.Status != model.WaybillStatusTruck {
			err = errors.New(fmt.Sprintf("运单号%s状态为%s，无法下车！", waybillNo, model.WaybillStatusMap[waybillModel.Status]))
			return err
		}
		waybillModel.Status = model.WaybillStatusTruckOut
		waybillModel.CarId = car.Id
		err = tx.Save(&waybillModel).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单信息失败：%s", err))
		}

		var lng, lat string
		lng, lat, err = e.GetSite(p.DeptId, car.Sn, c.StartTime.String())
		if err != nil {
			e.Log.Errorf("获取定位信息失败: %s", err)
			return err
		}

		// 添加物流
		Logistics := model.WaybillLogistics{
			WaybillNo: waybillNo,
			Status:    model.WaybillStatusTruckOut,
			CarId:     car.Id,
			UserId:    p.UserId,
			Lng:       lng,
			Lat:       lat,
			ControlBy: model2.ControlBy{
				CreateBy: p.UserId,
			},
			DeptBy: model2.DeptBy{
				DeptId: p.DeptId,
			},
			ModelTime: model2.ModelTime{
				CreatedAt: c.StartTime,
			},
		}
		err = tx.Create(&Logistics).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单物流信息失败：%s", err))
		}

		// 查询任务
		var task model.WaybillTask
		err = tx.Model(&task).Where("waybill_no = ? and car_id = ?", waybillNo, car.Id).
			First(&task).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("查询运单任务信息失败：%s", err))
		}
		task.EndTime = c.StartTime
		task.UpdateBy = p.UserId
		err = tx.Save(&task).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单任务信息失败：%s", err))
		}
	}

	return nil
}
func (e *Waybill) Receipt(c *dto.WaybillReceiptReq, p *actions.DataPermission) error {
	var err error

	tx := e.Orm.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()
	// 查询仓库信息
	var car = model.Car{}
	// 查询运单是否存在
	err = tx.Scopes(actions.Permission(car.TableName(), p)).
		Where("user_id = ?", p.UserId).
		First(&car).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return errors.New("获取车辆信息失败")
	}

	var waybillModel = model.Waybill{}
	// 查询运单是否存在
	err = tx.Scopes(actions.Permission(waybillModel.TableName(), p)).
		Where("waybill_no = ?", c.WaybillNo).
		First(&waybillModel).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return errors.New(fmt.Sprintf("运单号%s不存在", c.WaybillNo))
		}
		return errors.New(fmt.Sprintf("运单号%s查询失败", c.WaybillNo))
	}

	if waybillModel.Status == model.WaybillStatusReceipt {
		return nil
	}
	waybillModel.Status = model.WaybillStatusReceipt
	waybillModel.CarId = car.Id
	waybillModel.ReceiptTime = c.StartTime
	waybillModel.ReceiptImg = c.ReceiptImg
	err = tx.Save(&waybillModel).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return errors.New(fmt.Sprintf("保存运单信息失败：%s", err))
	}

	var lng, lat string
	lng, lat, err = e.GetSite(p.DeptId, car.Sn, c.StartTime.String())
	if err != nil {
		e.Log.Errorf("获取定位信息失败: %s", err)
		return err
	}

	// 查询任务
	var task model.WaybillTask
	err = tx.Model(&task).Where("waybill_no = ? and car_id = ?", c.WaybillNo, car.Id).
		Last(&task).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return errors.New(fmt.Sprintf("查询运单任务信息失败：%s", err))
	}
	// 未下车 直接点签收
	if time.Time(task.EndTime).IsZero() {
		task.EndTime = model2.Time(time.Now())
		task.UpdateBy = p.UserId
		err = tx.Save(&task).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单任务信息失败：%s", err))
		}
		// 添加下车物流记录
		Logistics := model.WaybillLogistics{
			WaybillNo: c.WaybillNo,
			Status:    model.WaybillStatusTruckOut,
			CarId:     car.Id,
			UserId:    p.UserId,
			Lng:       lng,
			Lat:       lat,
			ControlBy: model2.ControlBy{
				CreateBy: p.UserId,
			},
			DeptBy: model2.DeptBy{
				DeptId: p.DeptId,
			},
			ModelTime: model2.ModelTime{
				CreatedAt: c.StartTime,
			},
		}
		err = tx.Create(&Logistics).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单物流信息失败：%s", err))
		}
	}

	// 添加签收记录
	Logistics := model.WaybillLogistics{
		WaybillNo: c.WaybillNo,
		Status:    model.WaybillStatusReceipt,
		CarId:     car.Id,
		UserId:    p.UserId,
		Lng:       lng,
		Lat:       lat,
		ControlBy: model2.ControlBy{
			CreateBy: p.UserId,
		},
		DeptBy: model2.DeptBy{
			DeptId: p.DeptId,
		},
		ModelTime: model2.ModelTime{
			CreatedAt: c.StartTime,
		},
	}
	err = tx.Create(&Logistics).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return errors.New(fmt.Sprintf("保存运单物流信息失败：%s", err))
	}

	return nil
}

// 获取入库 出库 上车 下车 定位信息
func (e *Waybill) GetSite(companyId int, sn string, time string) (lng, lat string, err error) {
	// 获取公司秘钥
	var company model.SysDept
	company, err = model.GetCompanyById(companyId)
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return lng, lat, model.GetCompanyKeyErr
	}
	// 获取传感器信息
	var deviceSensorList = []nats_server.DeviceSensor_R{}
	var count int64
	deviceSensorList, count, err = nats_server.Cold_CompanyDeviceSensor_List_ByKey(sn, company.ColdKey)
	if err != nil || count == 0 {
		err = errors.New("查询设备定位信息失败")
		return lng, lat, err
	}
	var deviceData nats_server.DeviceData_
	deviceData, err = nats_server.Cold_ReadDeviceDataBy_T_snid_T_time(deviceSensorList[0].T_sn, deviceSensorList[0].T_id, time)
	if err != nil {
		err = errors.New("查询设备定位信息失败")
		return lng, lat, err
	}
	if len(deviceData.T_site) > 0 {
		site := strings.Split(deviceSensorList[0].T_DeviceSensorData.T_site, ",")
		if len(site) == 2 {
			lng = site[0]
			lat = site[1]
		}
	}
	return lng, lat, nil
}

// 获取今日运单数 未派单 未装车 未入库 运送中
// 本月运单总数 上月运单总数 本年运单总数 上年运单总数
func (e *Waybill) GetBasicsStats(c *dto.WaybillStatsReq, p *actions.DataPermission) dto.WaybillStatsRes {
	var res dto.WaybillStatsRes
	var data model.Waybill
	type DateCount struct {
		Date  string
		Count int64
	}
	yearCount := make([]DateCount, 0)
	monthCount := make([]DateCount, 0)
	now := time.Now()
	todayStartTime := now.Format("2006-01-02") + " 00:00:00"
	todayEndTime := now.Format("2006-01-02") + " 23:59:59"

	// 获取上个月第一天
	firstDayOfLastMonth := time.Date(now.Year(), now.Month()-1, 1, 0, 0, 0, 0, now.Location())

	// 获取去年的第一天
	firstDayOfLastYear := time.Date(now.Year()-1, time.January, 1, 0, 0, 0, 0, now.Location())
	monthStartTime := firstDayOfLastMonth.Format("2006-01-02") + " 00:00:00"
	yearStartTime := firstDayOfLastYear.Format("2006-01-02") + " 00:00:00"

	// 今日总运单数
	e.Orm.Model(&data).Scopes(actions.Permission(data.TableName(), p)).Where("order_time between ? and ?", todayStartTime, todayEndTime).Count(&res.TodayNum)
	// 待派单
	e.Orm.Model(&data).Scopes(actions.Permission(data.TableName(), p)).Where("status = ?", model.WaybillStatusWaitDelivery).Count(&res.WaitDeliveryNum)
	// 未装车
	e.Orm.Model(&data).Scopes(actions.Permission(data.TableName(), p)).Where("status = ?", model.WaybillStatusWaitTruck).Count(&res.WaitTruckNum)
	// 未入库
	e.Orm.Model(&data).Scopes(actions.Permission(data.TableName(), p)).Where("status = ?", model.WaybillStatusWaitStorage).Count(&res.WaitStorageNum)
	// 配送中
	e.Orm.Model(&data).Scopes(actions.Permission(data.TableName(), p)).Where("status in (?)", []int{model.WaybillStatusTruck, model.WaybillStatusStorage, model.WaybillStatusTruckOut, model.WaybillStatusStorageOut}).Count(&res.InDeliveryNum)
	// 获取本月，上月数据
	e.Orm.Model(&data).Select("date_format(order_time,'%Y%m') date,count(1) as count ").Scopes(actions.Permission(data.TableName(), p)).
		Where("order_time between ? and ?", monthStartTime, now).Group("date").Find(&monthCount)

	for _, month := range monthCount {
		if month.Date == now.Format("200601") {
			res.ThisMonthNum = month.Count
		}
		if month.Date == firstDayOfLastMonth.Format("200601") {
			res.LastMonthNum = month.Count
		}
	}

	// 获取本年，上年数据
	e.Orm.Model(&data).Select("date_format(order_time,'%Y') date,count(1) as count ").Scopes(actions.Permission(data.TableName(), p)).
		Where("order_time between ? and ?", yearStartTime, now).Group("date").Find(&yearCount)

	for _, month := range yearCount {
		if month.Date == now.Format("2006") {
			res.ThisYearNum = month.Count
		}
		if month.Date == firstDayOfLastYear.Format("2006") {
			res.LastYearNum = month.Count
		}
	}

	if c.Type == "year" {
		// 获取上个月第一天
		year, _ := strconv.Atoi(c.Date)
		firstDayOfyear := time.Date(year, 1, 1, 0, 0, 0, 0, now.Location())
		lastDayOfyear := time.Date(year, 12, 31, 23, 59, 59, 0, now.Location())

		// 年度数据统计
		e.Orm.Model(&data).Select("date_format(order_time,'%Y-%m') date,count(1) as num ").Scopes(actions.Permission(data.TableName(), p)).
			Where("order_time between ? and ?", firstDayOfyear, lastDayOfyear).Group("date").Find(&res.Stats)
	}

	if c.Type == "month" {
		// 获取上个月第一天
		month, _ := time.Parse("2006-01", c.Date)
		firstDayOfMonth := time.Date(month.Year(), month.Month(), 1, 0, 0, 0, 0, now.Location())
		lastDayOfMonth := time.Date(month.Year(), month.Month()+1, 1, 23, 59, 59, 0, now.Location()).Add(-time.Hour * 24)
		e.Orm.Model(&data).Select("date_format(order_time,'%Y-%m-%d') date,count(1) as num ").Scopes(actions.Permission(data.TableName(), p)).
			Where("order_time between ? and ?", firstDayOfMonth, lastDayOfMonth).Group("date").Find(&res.Stats)
	}

	return res
}

// 获取运单所有温湿度素具
func (e *Waybill) GetAllData(c *dto.WaybillGetByWaybillNoReq) ([]nats_server.DeviceData_R, []WaybillPDF, error) {
	var err error
	var data model.WaybillTask
	var waybill model.Waybill
	var taskList []model.WaybillTask
	var waybillPDF []WaybillPDF
	dataList := make([]nats_server.DeviceData_R, 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 dataList, waybillPDF, errors.New("获取运单信息失败")
	}

	// 未签收，不返回数据
	if waybill.Status != model.WaybillStatusReceipt {
		return dataList, waybillPDF, nil
	}

	// 获取公司秘钥
	var company model.SysDept
	company, err = model.GetCompanyById(waybill.DeptId)
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return dataList, waybillPDF, model.GetCompanyKeyErr
	}

	err = e.Orm.Model(&data).
		Where("waybill_no = ?", c.WaybillNo).
		Order("id desc").
		Find(&taskList).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return dataList, waybillPDF, global.GetFailedErr
	}

	for i := 0; i < len(taskList); i++ {
		// 获取传感器信息
		deviceSensorList, _, _ := nats_server.Cold_CompanyDeviceSensor_List_ByKey(taskList[i].Sn, company.ColdKey)
		var T_snid string
		var list []nats_server.DeviceData_R
		for _, r := range deviceSensorList {
			T_snid += fmt.Sprintf("%s,%d|", r.T_sn, r.T_id)
		}
		list, _, err = nats_server.Cold_ReadDeviceDataListBy_T_snid(T_snid, taskList[i].StartTime.String(), taskList[i].EndTime.String(), 0, 9999)
		if err != nil {
			e.Log.Errorf("nats 获取温湿度信息失败: %s", err)
			return dataList, waybillPDF, global.GetFailedErr
		}
		dataList = append(dataList, list...)
		waybillPDF = append(waybillPDF, WaybillPDF{
			Data:             list,
			DeviceSensorList: deviceSensorList,
			Task:             taskList[i],
		})
	}
	// 创建名称到权重的映射
	orderMap := make(map[string]int)
	for i, v := range taskList {
		orderMap[v.Sn] = i
	}
	sort.Slice(dataList, func(i, j int) bool {
		if dataList[i].T_time == dataList[j].T_time {
			// 如果时间相同，则按预设顺序排序
			return orderMap[dataList[i].T_sn] < orderMap[dataList[j].T_sn]
		}
		return dataList[i].T_time < dataList[j].T_time
	})

	return dataList, waybillPDF, nil
}

type WaybillPDF struct {
	Data             []nats_server.DeviceData_R   `json:"data"`
	DeviceSensorList []nats_server.DeviceSensor_R `json:"deviceSensorList"`
	Task             model.WaybillTask            `json:"task"`
}