package service import ( "errors" "fmt" "gas-cylinder-api/app/admin/model" "gas-cylinder-api/app/admin/service/dto" "gas-cylinder-api/common/actions" cDto "gas-cylinder-api/common/dto" "gas-cylinder-api/common/global" common "gas-cylinder-api/common/model" "gogs.baozhida.cn/zoie/OAuth-core/service" "gorm.io/gorm" "time" ) type Order struct { service.Service } // GetPage 获取Order列表 func (e *Order) GetPage(c *dto.OrderGetPageReq, list *[]model.Order, count *int64, p *actions.DataPermission) error { var err error var data model.Order err = e.Orm.Model(&data). Scopes( cDto.MakeCondition(c.GetNeedSearch()), cDto.Paginate(c.GetPageSize(), c.GetPageIndex()), actions.Permission(data.TableName(), p), ). Preload("Customer").Preload("Store").Preload("User").Preload("Goods").Preload("Spec"). 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 *Order) GetDeliveryPage(c *dto.OrderGetDeliveryPageReq, list *[]model.Order, count *int64, p *actions.DataPermission) error { var err error var data model.Order err = e.Orm.Model(&data). Scopes( cDto.MakeCondition(c.GetNeedSearch()), cDto.Paginate(c.GetPageSize(), c.GetPageIndex()), ). Where("user_id = ?", p.UserId). Preload("Customer").Preload("Store").Preload("Goods").Preload("Spec"). 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 获取Order对象 func (e *Order) Get(d *dto.OrderGetReq, orderModel *model.Order, p *actions.DataPermission) error { err := e.Orm. Scopes(actions.Permission(orderModel.TableName(), p)). First(orderModel, 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 } // Insert 创建Order对象 func (e *Order) Insert(c *dto.OrderInsertReq) error { var err error var data model.Order tx := e.Orm.Begin() defer func() { if err != nil { tx.Rollback() } else { tx.Commit() } }() // 添加订单 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 } // Update 修改Order对象 func (e *Order) Update(c *dto.OrderUpdateReq, p *actions.DataPermission) error { var err error tx := e.Orm.Begin() defer func() { if err != nil { tx.Rollback() } else { tx.Commit() } }() var orderModel = model.Order{} // 查询订单是否存在 err = e.Orm.Scopes(actions.Permission(orderModel.TableName(), p)). First(&orderModel, 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(&orderModel) err = tx.Save(&orderModel).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.UpdateFailedErr } c.Id = orderModel.Id return nil } // Remove 删除Order func (e *Order) Remove(c *dto.OrderDeleteReq, p *actions.DataPermission) error { var err error tx := e.Orm.Begin() defer func() { if err != nil { tx.Rollback() } else { tx.Commit() } }() var orderModel model.Order // 查询订单是否存在 err = e.Orm.Scopes(actions.Permission(orderModel.TableName(), p)). First(&orderModel, c.GetId()).Error if err != nil { e.Log.Errorf("db error: %s", err) if errors.Is(err, gorm.ErrRecordNotFound) { return nil } return global.DeleteFailedErr } if orderModel.State != model.OrderStateOrder && orderModel.State != model.OrderStateCancel { err = errors.New(fmt.Sprintf("订单状态为%s,无法删除", model.OrderStateMap[orderModel.State])) return err } db := tx.Delete(&orderModel) 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 *Order) Cancel(c *dto.OrderCancelReq, p *actions.DataPermission) error { var err error tx := e.Orm.Begin() defer func() { if err != nil { tx.Rollback() } else { tx.Commit() } }() var orderModel model.Order // 查询订单是否存在 err = e.Orm.Scopes(actions.Permission(orderModel.TableName(), p)). First(&orderModel, 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 } if orderModel.State == model.OrderStateCancel { return nil } if orderModel.State != model.OrderStateOrder { return errors.New(fmt.Sprintf("订单状态为%s,禁止取消", model.OrderStateMap[orderModel.State])) } orderModel.State = model.OrderStateCancel orderModel.CancelTime = common.Time(time.Now()) db := tx.Save(&orderModel) if err = db.Error; err != nil { e.Log.Errorf("db error: %s", err) return global.UpdateFailedErr } if db.RowsAffected == 0 { return global.UpdateNotFoundOrNoPermissionErr } return nil } // Delivery 派单 func (e *Order) Delivery(c *dto.OrderDeliveryReq, p *actions.DataPermission) error { var err error var orderModel = model.Order{} // 查询订单是否存在 err = e.Orm.Scopes(actions.Permission(orderModel.TableName(), p)). First(&orderModel, 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(&orderModel) err = e.Orm.Save(&orderModel).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.UpdateFailedErr } c.Id = orderModel.Id return nil } // UpdateState 修改订单状态 func (e *Order) UpdateState(c *dto.OrderUpdateStateReq, p *actions.DataPermission) error { var err error var orderModel = model.Order{} // 查询订单是否存在 err = e.Orm.Scopes(actions.Permission(orderModel.TableName(), p)). Preload("User").Preload("Store").Preload("Dept"). First(&orderModel, 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 } tx := e.Orm.Begin() defer func() { if err != nil { tx.Rollback() } else { tx.Commit() } }() c.Generate(&orderModel) // 配送中 if c.State == model.OrderStateInDelivery { orderModel.DeliveryTime = common.Time(time.Now()) } // 已送达 if c.State == model.OrderStateArrive { orderModel.ArriveTime = common.Time(time.Now()) err = InsertOperationLogByOptType26Or36(tx, "26", c.ChipUid, orderModel) if err != nil { e.Log.Errorf("db error: %s", err) return global.UpdateFailedErr } } // 已取消 if c.State == model.OrderStateCancel && orderModel.State == model.OrderStateArrive { orderModel.CancelTime = common.Time(time.Now()) err = InsertOperationLogByOptType26Or36(tx, "36", c.ChipUid, orderModel) if err != nil { e.Log.Errorf("db error: %s", err) return global.UpdateFailedErr } } err = e.Orm.Save(&orderModel).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.UpdateFailedErr } c.Id = orderModel.Id return nil }