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

package service

import (
	"cold-delivery/app/admin/model"
	"cold-delivery/app/admin/service/dto"
	"cold-delivery/common/actions"
	cDto "cold-delivery/common/dto"
	"cold-delivery/common/global"
	"cold-delivery/common/lib"
	model2 "cold-delivery/common/model"
	"cold-delivery/common/nats/nats_server"
	"cold-delivery/conf"
	"errors"
	"fmt"
	"go.uber.org/zap"
	"gogs.baozhida.cn/zoie/OAuth-core/pkg/sms"
	"gogs.baozhida.cn/zoie/OAuth-core/pkg/utils"
	"gogs.baozhida.cn/zoie/OAuth-core/service"
	"gorm.io/gorm"
	"net/url"
	"sort"
	"strconv"
	"strings"
	"time"
)

type Waybill struct {
	service.Service
}

// 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),
			).
			Preload("CoolerBox").
			Preload("Delivery").
			Preload("ReCheck").
			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("CoolerBox").
		Preload("Delivery").
		Preload("ReCheck").
		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) GetUserPage(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()),
			).
			Where("create_by = ?", p.UserId).
			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()),
		).
		Where("delivery_id = ? or re_check_id = ? or create_by = ?", p.UserId, p.UserId, p.UserId).
		Preload("CoolerBox").
		Preload("Delivery").
		Preload("ReCheck").
		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(
				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),
		).
		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.WaybillGetByWaybillPdfReq, waybillModel *model.Waybill, p *actions.DataPermission) error {
	err := e.Orm.
		Scopes(actions.Permission(waybillModel.TableName(), p)).
		Where("waybill_no = ?", d.WaybillNo).
		Preload("Dept").
		Preload("CoolerBox").
		Preload("Delivery").
		Preload("ReCheck").
		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

	// 保存收货地址
	//if c.SaveSenderAddress {
	//	// 查询地址是否存在
	//	address := model.Address{}
	//	err = e.Orm.Where("address = ? and phone = ? and name = ? and dept_id = ? and address_type = 'sender'",
	//		c.SenderAddressDetails, c.SenderAddressPhone, c.SenderAddressName, c.DeptId).First(&address).Error
	//	if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
	//		address.Address = c.SenderAddressDetails
	//		address.Phone = c.SenderAddressPhone
	//		address.Name = c.SenderAddressName
	//		address.DeptId = c.DeptId
	//		address.AddressType = "sender"
	//		err = tx.Create(&address).Error
	//		if err != nil {
	//			e.Log.Errorf("创建地址失败: %s", err)
	//			return global.CreateFailedErr
	//		}
	//	}
	//}

	// 保存发货地址
	if c.SaveConsigneeAddress {
		// 查询地址是否存在
		address := model.Address{}
		err = e.Orm.Where("address = ? and phone = ? and name = ? and dept_id = ? and address_type = 'consignee'",
			c.ConsigneeAddressDetails, c.ConsigneeAddressPhone, c.ConsigneeAddressName, c.DeptId).First(&address).Error
		if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
			address.Address = c.ConsigneeAddressDetails
			address.Phone = c.ConsigneeAddressPhone
			address.Name = c.ConsigneeAddressName
			address.DeptId = c.DeptId
			address.AddressType = "consignee"
			err = tx.Create(&address).Error
			if err != nil {
				e.Log.Errorf("创建地址失败: %s", err)
				return global.CreateFailedErr
			}
		}
	}

	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) UpdateStatus(c *dto.WaybillUpdateStatusReq, 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()
		}
	}()

	if !lib.StrListEqual(c.IceRaftCode, c.OldIceRaftCode) {
		// 获取删除的冰排列表
		//missingIceRaftCode = lib.FindStrListMissingData(c.OldIceRaftCode, c.IceRaftCode)

		//if len(missingIceRaftCode) > 0 {
		//	// 删除冰排出库记录
		//	IceRaftList := []model.IceRaft{}
		//	err = tx.Model(&model.IceRaft{}).Where("code in (?)", missingIceRaftCode).Find(&IceRaftList).Error
		//	if err != nil {
		//		e.Log.Errorf("db error: %s", err)
		//		return errors.New("获取冰排列表失败")
		//	}
		//	for _, iceRaft := range IceRaftList {
		//		// 获取冰排记录
		//		err = tx.Model(&model.IceRaftRecord{}).
		//			Where("ice_raft_id = ? and status = ?", iceRaft.Id, model.IceRaftRecordStatusUsing).
		//			Updates(map[string]interface{}{
		//				"status":           model.IceRaftRecordStatusWaitUse,
		//				"out_storage_time": nil,
		//				"cooler_box_id":    0,
		//			}).Error
		//		if err != nil {
		//			e.Log.Errorf("db error: %s", err)
		//			return errors.New(fmt.Sprintf("冰排【%s】返库失败", iceRaft.Code))
		//		}
		//	}
		//}
		// 获取新增的冰排列表
		extraData := lib.FindStrListExtraData(c.OldIceRaftCode, c.IceRaftCode)
		if len(extraData) > 0 {
			for _, code := range extraData {
				var iceRaft model.IceRaft
				err = tx.Model(&model.IceRaft{}).Where("code = ?", code).Preload("IceRaftRecord").First(&iceRaft).Error
				if err != nil {
					e.Log.Errorf("db error: %s", err)
					if errors.Is(err, gorm.ErrRecordNotFound) {
						return errors.New(fmt.Sprintf("冰排【%s】不存在", code))
					} else {
						return errors.New(fmt.Sprintf("获取冰排【%s】信息失败", code))
					}

				}
				if iceRaft.IceRaftRecord.Status != model.IceRaftRecordStatusWaitUse {
					err = errors.New(fmt.Sprintf("冰排【%s】状态为%s，出库保温箱失败！", code, model.GetIceRaftRecordStatus(iceRaft.IceRaftRecord.Status)))
					return err
				}
				// 获取冰排记录
				err = tx.Model(&model.IceRaftRecord{}).
					Where("ice_raft_id = ? and status = ?", iceRaft.Id, model.IceRaftRecordStatusWaitUse).
					Updates(map[string]interface{}{
						"status":           model.IceRaftRecordStatusUsing,
						"out_storage_time": time.Now(),
						"cooler_box_id":    c.CoolerBoxId,
					}).Error
				if err != nil {
					e.Log.Errorf("db error: %s", err)
					return errors.New(fmt.Sprintf("冰排【%s】出库失败", iceRaft.Code))
				}
			}
		}
	}

	var coolerBox = model.CoolerBox{}
	// 查询保温箱信息
	err = tx.Scopes(actions.Permission(coolerBox.TableName(), p)).
		First(&coolerBox, c.CoolerBoxId).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return errors.New("获取保温箱信息失败")
	}

	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
		}
		waybillModeStatus := waybillModel.Status
		if waybillModel.Status == model.WaybillStatusWaitDelivery ||
			(waybillModel.Status == model.WaybillStatusInDelivery && waybillModel.CoolerBoxId != c.CoolerBoxId) ||
			(waybillModel.Status == model.WaybillStatusInDelivery && !lib.StrListEqual(c.IceRaftCode, c.OldIceRaftCode)) {
			waybillModel.Status = model.WaybillStatusInDelivery
			waybillModel.CoolerBoxId = c.CoolerBoxId
			waybillModel.DeliveryTime = model2.Time(time.Now())
			waybillModel.IceRaftCode = c.IceRaftCode
			err = tx.Save(&waybillModel).Error
			if err != nil {
				e.Log.Errorf("db error: %s", err)
				return global.UpdateFailedErr
			}

			// 查询任务
			var logistics model.WaybillLogistics
			err = tx.Model(&logistics).Where("waybill_no = ? and status = ?", waybillModel.WaybillNo, model.WaybillStatusInDelivery).
				Last(&logistics).Error
			if err != nil {
				if errors.Is(err, gorm.ErrRecordNotFound) {
					// 添加物流
					logisticsObj := model.WaybillLogistics{
						WaybillNo:   waybillModel.WaybillNo,
						Status:      waybillModel.Status,
						CoolerBoxId: coolerBox.Id,
						ControlBy: model2.ControlBy{
							CreateBy: p.UserId,
						},
						DeptBy: model2.DeptBy{
							DeptId: p.DeptId,
						},
					}
					err = tx.Create(&logisticsObj).Error
					if err != nil {
						e.Log.Errorf("db error: %s", err)
						return errors.New(fmt.Sprintf("保存运单物流信息失败：%s", err))
					}
				} else {
					e.Log.Errorf("db error: %s", err)
					return errors.New(fmt.Sprintf("保存运单物流信息失败：%s", err))
				}
			}
			logistics.CoolerBoxId = coolerBox.Id
			err = tx.Save(&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 = ? ", waybillModel.WaybillNo).
				Last(&task).Error
			if err != nil {
				if errors.Is(err, gorm.ErrRecordNotFound) {
					// 添加任务
					taskObj := model.WaybillTask{
						WaybillNo:   waybillModel.WaybillNo,
						CoolerBoxId: coolerBox.Id,
						UserId:      p.UserId,
						Sn:          coolerBox.Sn,
						StartTime:   model2.Time(time.Now()),
						ControlBy: model2.ControlBy{
							CreateBy: p.UserId,
						},
						DeptBy: model2.DeptBy{
							DeptId: p.DeptId,
						},
					}
					err = tx.Create(&taskObj).Error
					if err != nil {
						e.Log.Errorf("db error: %s", err)
						return errors.New(fmt.Sprintf("保存运单任务信息失败：%s", err))
					}
				} else {
					e.Log.Errorf("db error: %s", err)
					return errors.New(fmt.Sprintf("查询运单任务信息失败：%s", err))
				}
			}

			task.StartTime = model2.Time(time.Now())
			task.UpdateBy = p.UserId
			task.CoolerBoxId = coolerBox.Id
			task.Sn = coolerBox.Sn
			err = tx.Save(&task).Error
			if err != nil {
				e.Log.Errorf("db error: %s", err)
				return errors.New(fmt.Sprintf("保存运单任务信息失败：%s", err))
			}

		}

		// 待派单状态下才发送短信
		if waybillModeStatus == model.WaybillStatusWaitDelivery {
			ss := sms.NewSMS(conf.ExtConfig.SubMail.Appid, conf.ExtConfig.SubMail.Signature)
			addr := conf.ExtConfig.Applet.WaybillUrl + url.QueryEscape(lib.AesEncryptCBC(waybillModel.WaybillNo, lib.AesKey))
			res, err1 := ss.SmsXSend(waybillModel.ConsigneeAddressPhone, addr)
			if err1 != nil || res.Status != sms.SUCCESS {
				e.Log.Errorf("派单短信发送失败", zap.Any("res", res), zap.Error(err1))
				err = errors.New("派单短信发送失败，请检查收件人电话是否正确！")
				return err
			}
			waybillModel.SendLog = model.WaybillSendLog{
				Phone:   waybillModel.ConsigneeAddressPhone,
				Content: "【冷链送药追溯平台】您的运单正在派送中，点击查看详情：" + addr,
			}
		}

	}

	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 {
		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) 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 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))
	}

	// 查询保温箱信息
	var coolerBox = model.CoolerBox{}
	err = tx.Scopes(actions.Permission(coolerBox.TableName(), p)).
		First(&coolerBox, waybillModel.CoolerBoxId).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return errors.New("获取保温箱信息失败")
	}
	if waybillModel.Status == model.WaybillStatusWaitDelivery {
		err = errors.New(fmt.Sprintf("运单状态为%s，禁止操作", model.WaybillStatusMap[waybillModel.Status]))
		return err
	}

	if waybillModel.Status == model.WaybillStatusReceipt || waybillModel.Status == model.WaybillStatusRejection {
		return nil
	}
	waybillModel.Status = c.Status
	waybillModel.RejectionReason = c.RejectionReason
	waybillModel.ReceiptTime = model2.Time(time.Now())
	waybillModel.ReceiptImg = c.ReceiptImg
	waybillModel.AssessContent = c.AssessContent
	waybillModel.AssessStar = c.AssessStar
	waybillModel.DeliveryDuration = int(time.Now().Sub(waybillModel.DeliveryTime.Local()).Minutes())
	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(waybillModel.DeptId, coolerBox.Sn, waybillModel.ReceiptTime.String())
	if err != nil {
		e.Log.Errorf("获取定位信息失败: %s", err)
		return err
	}

	// 查询任务
	var task model.WaybillTask
	err = tx.Model(&task).Where("waybill_no = ? and cooler_box_id = ?", c.WaybillNo, coolerBox.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.UserIds
		err = tx.Save(&task).Error
		if err != nil {
			e.Log.Errorf("db error: %s", err)
			return errors.New(fmt.Sprintf("保存运单任务信息失败：%s", err))
		}
	}
	var userId int
	if p != nil {
		userId = p.UserId
	}
	// 添加签收记录
	Logistics := model.WaybillLogistics{
		WaybillNo:   c.WaybillNo,
		Status:      c.Status,
		CoolerBoxId: coolerBox.Id,
		UserId:      userId,
		Lng:         lng,
		Lat:         lat,
		ControlBy: model2.ControlBy{
			CreateBy: userId,
		},
		DeptBy: model2.DeptBy{
			DeptId: waybillModel.DeptId,
		},
		ModelTime: model2.ModelTime{
			CreatedAt: waybillModel.ReceiptTime,
		},
	}
	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.WaybillStatusInDelivery).Count(&res.InDeliveryNum)
	// 已送达
	e.Orm.Model(&data).Scopes(actions.Permission(data.TableName(), p)).Where("status = ?", model.WaybillStatusReceipt).Count(&res.ReceiptNum)
	// 已取消
	e.Orm.Model(&data).Scopes(actions.Permission(data.TableName(), p)).Where("status = ?", model.WaybillStatusRejection).Count(&res.RejectionNum)
	// 获取本月，上月数据
	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.ParseInLocation("2006-01", c.Date, time.Local)
		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 GetDeliveryStatsScopes(userId int) func(db *gorm.DB) *gorm.DB {

	return func(db *gorm.DB) *gorm.DB {
		if userId == 0 {
			return db
		}
		return db.Where("delivery_id = ?", userId)
	}
}
func GetReCheckStatsScopes(userId int) func(db *gorm.DB) *gorm.DB {

	return func(db *gorm.DB) *gorm.DB {
		if userId == 0 {
			return db
		}
		return db.Where("re_check_id = ?", userId)
	}
}

// 获取配送员统计信息
func (e *Waybill) GetDeliveryStats(c *dto.WaybillUserStatsReq, p *actions.DataPermission) (res []dto.WaybillUserStats) {
	// 获取用户列表
	var userList []model.SysUser
	if err := e.Orm.Model(&userList).Scopes(actions.Permission(model.SysUser{}.TableName(), p), WaybillUserStatsUserIdScopes(c.UserIds)).
		Where("status = 2").Order("created_at desc").Find(&userList).Error; err != nil {
		return nil
	}
	for _, user := range userList {
		stats := e.GetUserStats(c, p, GetDeliveryStatsScopes(user.Id))
		stats.Name = user.NickName
		res = append(res, stats)
	}
	return
}
func (e *Waybill) GetReCheckStats(c *dto.WaybillUserStatsReq, p *actions.DataPermission) (res []dto.WaybillUserStats) {
	// 获取用户列表
	var userList []model.SysUser
	if err := e.Orm.Model(&userList).Scopes(actions.Permission(model.SysUser{}.TableName(), p), WaybillUserStatsUserIdScopes(c.UserIds)).
		Where("status = 2").Order("created_at desc").Find(&userList).Error; err != nil {
		return nil
	}
	for _, user := range userList {
		stats := e.GetUserStats(c, p, GetReCheckStatsScopes(user.Id))
		stats.Name = user.NickName
		res = append(res, stats)
	}
	return
}

func WaybillUserStatsUserIdScopes(userIds []int) func(db *gorm.DB) *gorm.DB {
	return func(db *gorm.DB) *gorm.DB {
		if len(userIds) == 0 {
			return db
		}
		if len(userIds) == 1 {
			return db.Where("id = ?", userIds[0])
		}
		return db.Where("id in (?)", userIds)
	}
}

func GetUserBasicsStatsScopes(userId int) func(db *gorm.DB) *gorm.DB {

	return func(db *gorm.DB) *gorm.DB {
		if userId == 0 {
			return db
		}
		return db.Where("delivery_id = ? or re_check_id = ? or create_by = ?", userId, userId, userId)
	}
}
func (e *Waybill) GetUserBasicsStats(c *dto.WaybillStatsReq, p *actions.DataPermission, scopes func(*gorm.DB) *gorm.DB) 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), scopes).
		Where("order_time between ? and ?", todayStartTime, todayEndTime).Count(&res.TodayNum)
	// 待派单
	e.Orm.Model(&data).Scopes(actions.Permission(data.TableName(), p), scopes).
		Where("status = ?", model.WaybillStatusWaitDelivery).Count(&res.WaitDeliveryNum)
	// 配送中
	e.Orm.Model(&data).Scopes(actions.Permission(data.TableName(), p), scopes).
		Where("status = ?", model.WaybillStatusInDelivery).Count(&res.InDeliveryNum)
	// 已送达
	e.Orm.Model(&data).Scopes(actions.Permission(data.TableName(), p), scopes).
		Where("status = ?", model.WaybillStatusReceipt).Count(&res.ReceiptNum)
	// 已取消
	e.Orm.Model(&data).Scopes(actions.Permission(data.TableName(), p), scopes).
		Where("status = ?", model.WaybillStatusRejection).Count(&res.RejectionNum)
	// 获取本月，上月数据
	e.Orm.Model(&data).Select("date_format(order_time,'%Y%m') date,count(1) as count ").Scopes(actions.Permission(data.TableName(), p), scopes).
		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), scopes).
		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), scopes).
			Where("order_time between ? and ?", firstDayOfyear, lastDayOfyear).Group("date").Find(&res.Stats)
	}

	if c.Type == "month" {
		// 获取上个月第一天
		month, _ := time.ParseInLocation("2006-01", c.Date, time.Local)
		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), scopes).
			Where("order_time between ? and ?", firstDayOfMonth, lastDayOfMonth).Group("date").Find(&res.Stats)
	}

	return res
}
func (e *Waybill) GetUserStats(c *dto.WaybillUserStatsReq, p *actions.DataPermission, scopes func(*gorm.DB) *gorm.DB) dto.WaybillUserStats {
	var res dto.WaybillUserStats
	var data model.Waybill
	type DateCount struct {
		Date  string
		Count int64
	}

	now := time.Now()

	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), scopes).
			Where("order_time between ? and ?", firstDayOfyear, lastDayOfyear).Group("date").Find(&res.Stats)
		// 获取这个月的所有日期
		allDates := getAllDatesInYear(c.Date)
		// 补齐缺失日期的数据
		res.Stats = completeMissingDates(allDates, res.Stats)
	}

	if c.Type == "month" {
		// 获取上个月第一天
		month, _ := time.ParseInLocation("2006-01", c.Date, time.Local)
		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), scopes).
			Where("order_time between ? and ?", firstDayOfMonth, lastDayOfMonth).Group("date").Find(&res.Stats)
		// 获取这个月的所有日期
		allDates := getAllDatesInMonth(c.Date)
		// 补齐缺失日期的数据
		res.Stats = completeMissingDates(allDates, res.Stats)
	}

	return res
}

// 获取指定月份的所有日期
func getAllDatesInMonth(month string) []string {
	var dates []string
	layout := "2006-01-02"
	firstDay, _ := time.Parse(layout, month+"-01")
	if time.Now().Before(firstDay) {
		return dates
	}
	lastDay := firstDay.AddDate(0, 1, -1)
	if time.Now().Before(lastDay) {
		lastDay = time.Now()
	}

	for d := firstDay; !d.After(lastDay); d = d.AddDate(0, 0, 1) {
		dates = append(dates, d.Format(layout))
	}

	return dates
}

// 获取指定年份的所有日期
func getAllDatesInYear(year string) []string {
	var dates []string
	layout := "2006-01"
	firstDay, _ := time.Parse(layout, year+"-01")
	lastDay := firstDay.AddDate(1, 0, -1)
	if time.Now().Before(firstDay) {
		return dates
	}
	if time.Now().Before(lastDay) {
		lastDay = time.Now()
	}
	for d := firstDay; !d.After(lastDay); d = d.AddDate(0, 1, 0) {
		dates = append(dates, d.Format(layout))
	}
	return dates
}

// 补齐缺失日期的数据
func completeMissingDates(allDates []string, stats []dto.WaybillStats) []dto.WaybillStats {
	dateMap := make(map[string]int64)
	for _, stat := range stats {
		dateMap[stat.Date] = stat.Num
	}

	var completedStats []dto.WaybillStats
	for _, date := range allDates {
		num, exists := dateMap[date]
		if !exists {
			num = 0
		}
		completedStats = append(completedStats, dto.WaybillStats{Date: date, Num: num})
	}

	return completedStats
}

// 获取运单所有温湿度素具
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
	}

	// 获取最后一个任务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 dataList, waybillPDF, err
	}

	// 创建名称到权重的映射
	orderMap := make(map[string]int)
	for i, v := range taskList {
		orderMap[v.Sn] = i
	}

	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)
		}
		var count int64
		list, count, 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
		}

		firstMap := map[int]nats_server.DeviceData_R{}
		lastMap := map[int]nats_server.DeviceData_R{}
		if count > 0 {
			for _, v := range deviceSensorList {

				for j := 0; j < len(list); j++ {
					if v.T_id == list[j].T_id {
						if list[j].T_time != taskList[i].StartTime.String() {
							firstData := list[j]
							firstData.T_time = taskList[i].StartTime.String()
							firstMap[v.T_id] = firstData
							count += 1
						}
						break
					}
				}
				if waybill.Status == model.WaybillStatusReceipt {
					for k := len(list) - 1; k >= 0; k-- {
						if v.T_id == list[k].T_id {
							if taskList[i].Id == lastWaybillTask.Id && list[k].T_time != taskList[i].EndTime.String() && !time.Time(taskList[i].EndTime).IsZero() {
								lastData := list[k]
								lastData.T_time = taskList[i].EndTime.String()
								lastMap[v.T_id] = lastData
								count += 1
							}
							break
						}
					}
				}
			}
		}

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

		for _, v := range lastMap {
			list = append(list, v)
		}
		sort.Slice(list, func(x, y int) bool {
			if list[x].T_time == list[y].T_time {
				// 如果时间相同，则按预设顺序排序
				return orderMap[list[x].T_sn] < orderMap[list[y].T_sn]
			}
			return list[x].T_time > list[y].T_time
		})

		dataList = append(dataList, list...)
		waybillPDF = append(waybillPDF, WaybillPDF{
			Data:             list,
			DeviceSensorList: deviceSensorList,
			Task:             taskList[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"`
}

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

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

	var company model.SysDept

	var i int64
	err = e.Orm.Model(&company).Where("name = ?", c.CompanyName).Count(&i).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return global.CreateFailedErr
	}
	if i == 0 {
		err = errors.New("公司名称不存在！")
		e.Log.Errorf("db error: %s", err)
		return err
	}

	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
	data.OrderNo = c.OrderId
	data.DeptId = company.Id
	data.Status = 1
	data.OrderTime = model2.Time(time.Now())
	err = tx.Create(&data).Error
	if err != nil {
		e.Log.Errorf("db error: %s", err)
		return global.CreateFailedErr
	}
	c.WaybillNo = data.WaybillNo

	return nil

}