123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880 |
- 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"
- model2 "cold-delivery/common/model"
- "errors"
- "fmt"
- "gogs.baozhida.cn/zoie/OAuth-core/service"
- "gorm.io/gorm"
- "sort"
- "time"
- )
- type IceRaft struct {
- service.Service
- }
- // GetPage 获取IceRaft列表
- func (e *IceRaft) GetPage(c *dto.IceRaftGetPageReq, list *[]model.IceRaft, count *int64, p *actions.DataPermission) error {
- var err error
- var data model.IceRaft
- if c.PageSize == 9999 {
- err = e.Orm.Model(&data).
- Scopes(
- actions.Permission(data.TableName(), p),
- ).
- Find(list).Limit(-1).Offset(-1).
- Count(count).Error
- } else {
- 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
- }
- func IceRaftNameRecordStatusScopes(status string) func(db *gorm.DB) *gorm.DB {
- return func(db *gorm.DB) *gorm.DB {
- if len(status) == 0 {
- return db
- }
- if status == "0" {
- return db.Where("ice_raft_record.status is null")
- }
- return db.Where("ice_raft_record.status = ?", status)
- }
- }
- func IceRaftNameRecordIceLockerScopes(status string) func(db *gorm.DB) *gorm.DB {
- return func(db *gorm.DB) *gorm.DB {
- if len(status) == 0 {
- return db.Where("ice_raft_record.status = 1 or ice_raft_record.status =2")
- }
- if status == "0" {
- return db.Where("ice_raft_record.status is null")
- }
- return db.Where("ice_raft_record.status = ?", status)
- }
- }
- func IceRaftNameRecordIceLockerIdScopes(iceLockerId int) func(db *gorm.DB) *gorm.DB {
- return func(db *gorm.DB) *gorm.DB {
- if iceLockerId == 0 {
- return db
- }
- return db.Where("ice_raft_record.ice_locker_id = ? and ice_raft_record.cooler_box_id = 0", iceLockerId)
- }
- }
- func IceRaftNameRecordCoolerBoxIdScopes(coolerBoxId int) func(db *gorm.DB) *gorm.DB {
- return func(db *gorm.DB) *gorm.DB {
- if coolerBoxId == 0 {
- return db
- }
- return db.Where("ice_raft_record.cooler_box_id = ?", coolerBoxId)
- }
- }
- func IceRaftStartTimeAndEndTimeScopes(startTime, endtime string) func(db *gorm.DB) *gorm.DB {
- return func(db *gorm.DB) *gorm.DB {
- if len(startTime) == 0 {
- return db
- }
- return db.Where("ice_raft_record.ice_use_time >= ?", startTime).Where("ice_raft_record.ice_use_time <= ?", endtime)
- }
- }
- func CollerStartTimeAndEndTimeScopes(startTime, endtime string) func(db *gorm.DB) *gorm.DB {
- return func(db *gorm.DB) *gorm.DB {
- if len(startTime) == 0 {
- return db
- }
- return db.Where("cooler_box_record.cooler_box_star_time >= ?", startTime).Where("cooler_box_record.cooler_box_star_time <= ?", endtime)
- }
- }
- func (e *IceRaft) GetRecordPage(c *dto.IceRaftGetNewestRecordPageReq, list *[]model.IceRaft, count *int64, p *actions.DataPermission) error {
- var err error
- var data model.IceRaft
- if c.PageSize == 9999 {
- err = e.Orm.Model(&data).
- Scopes(
- actions.Permission(data.TableName(), p),
- ).
- Where("ice_raft.status = '2'").
- Joins("left join ice_raft_record on ice_raft.ice_raft_record_id = ice_raft_record.id").
- Preload("IceRaftRecord.IceLocker").
- Preload("IceRaftRecord.CoolerBox").
- Preload("IceRaftRecord").
- Order("FIELD(ice_raft_record.status,'2','','1','3','4')").
- Find(list).Limit(-1).Offset(-1).
- Count(count).Error
- } else {
- err = e.Orm.Model(&data).
- Scopes(
- cDto.MakeCondition(c.GetNeedSearch()),
- cDto.Paginate(c.GetPageSize(), c.GetPageIndex()),
- actions.Permission(data.TableName(), p),
- IceRaftNameRecordStatusScopes(c.Status),
- IceRaftNameRecordIceLockerIdScopes(c.IceLockerId),
- IceRaftNameRecordCoolerBoxIdScopes(c.CoolerBoxId),
- ).
- Where("ice_raft.status = '2'").
- Joins("left join ice_raft_record on ice_raft.ice_raft_record_id = ice_raft_record.id").
- Preload("IceRaftRecord.IceLocker").
- Preload("IceRaftRecord.CoolerBox").
- Preload("IceRaftRecord").
- Order("FIELD(ice_raft_record.status,'2','','1','3','4')").
- 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 获取IceRaft对象
- func (e *IceRaft) Get(d *dto.IceRaftGetReq, iceRaftModel *model.IceRaft, p *actions.DataPermission) error {
- err := e.Orm.
- Scopes(actions.Permission(iceRaftModel.TableName(), p)).
- First(iceRaftModel, 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 *IceRaft) GetByCode(d *dto.IceRaftGetByCodeReq, iceRaftModel *model.IceRaft, p *actions.DataPermission) error {
- err := e.Orm.
- Scopes(actions.Permission(iceRaftModel.TableName(), p)).
- Where("code = ?", d.Code).
- First(iceRaftModel).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- if errors.Is(err, gorm.ErrRecordNotFound) {
- return errors.New(fmt.Sprintf("冰排编号【%s】不存在!", d.Code))
- }
- return global.GetFailedErr
- }
- return nil
- }
- // Insert 创建IceRaft对象
- func (e *IceRaft) Insert(c *dto.IceRaftInsertReq) error {
- var err error
- var data model.IceRaft
- tx := e.Orm.Begin()
- defer func() {
- if err != nil {
- tx.Rollback()
- } else {
- tx.Commit()
- }
- }()
- for _, code := range c.CodeList {
- var k int64
- err = tx.Model(&data).Where("code = ? and dept_id = ?", code, c.DeptId).Count(&k).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.CreateFailedErr
- }
- // 冰排编号已存在
- if k > 0 {
- err = errors.New("该编号已存在!")
- return err
- }
- // 添加冰排
- data.Id = 0
- c.Generate(&data, code)
- err = tx.Create(&data).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.CreateFailedErr
- }
- }
- return nil
- }
- // Update 修改IceRaft对象
- func (e *IceRaft) Update(c *dto.IceRaftUpdateReq, p *actions.DataPermission) error {
- var err error
- tx := e.Orm.Begin()
- defer func() {
- if err != nil {
- tx.Rollback()
- } else {
- tx.Commit()
- }
- }()
- var iceRaftModel = model.IceRaft{}
- // 查询冰排是否存在
- err = e.Orm.Scopes(actions.Permission(iceRaftModel.TableName(), p)).
- First(&iceRaftModel, 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 iceRaftModel.Code != c.Code {
- var k int64
- var data = model.IceRaft{}
- err = tx.Model(&data).Where("code = ? and dept_id = ?", c.Code, iceRaftModel.DeptId).Count(&k).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.CreateFailedErr
- }
- if k > 0 {
- err = errors.New("冰排编号已存在!")
- e.Log.Errorf("db error: %s", err)
- return err
- }
- }
- c.Generate(&iceRaftModel)
- err = tx.Save(&iceRaftModel).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.UpdateFailedErr
- }
- // 查询冰排记录是否存在
- var iceRaftRecordModel model.IceRaftRecord
- var count int64
- e.Orm.Table(iceRaftRecordModel.TableName()).Where("ice_raft_id = ?", c.Id).Count(&count)
- if count > 0 {
- iceRaftRecordModel.Id = iceRaftModel.IceRaftRecordId
- iceRaftRecordModel.FreezeClaim = iceRaftModel.FreezeClaim
- iceRaftRecordModel.Label = iceRaftModel.Label
- err = tx.Updates(&iceRaftRecordModel).Error
- if err != nil {
- e.Log.Errorf("db error: %s 该冰排未使用", err)
- return errors.New("修改冰排记录失败")
- }
- }
- c.Id = iceRaftModel.Id
- return nil
- }
- // Remove 删除IceRaft
- func (e *IceRaft) Remove(c *dto.IceRaftDeleteReq, p *actions.DataPermission) error {
- var err error
- tx := e.Orm.Begin()
- defer func() {
- if err != nil {
- tx.Rollback()
- } else {
- tx.Commit()
- }
- }()
- var iceRaftModel model.IceRaft
- // 查询冰排是否存在
- err = e.Orm.Scopes(actions.Permission(iceRaftModel.TableName(), p)).
- First(&iceRaftModel, 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
- }
- db := tx.Delete(&iceRaftModel)
- 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 *IceRaft) InStorage(c *dto.IceRaftInStorageReq, p *actions.DataPermission) error {
- var err error
- tx := e.Orm.Begin()
- defer func() {
- if err != nil {
- tx.Rollback()
- } else {
- tx.Commit()
- }
- }()
- // 查询冷冻柜信息
- var iceLockerModel model.IceLocker
- err = e.Orm.Scopes(actions.Permission(iceLockerModel.TableName(), p)).
- First(&iceLockerModel, c.IceLockerId).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return errors.New("获取冷冻柜信息失败")
- }
- if iceLockerModel.Status != model.IceLockerStatusNormal {
- err = errors.New("冷冻柜状态异常")
- return err
- }
- for _, code := range c.Code {
- // 查询冰排是否存在
- var iceRaftModel model.IceRaft
- err = e.Orm.Scopes(actions.Permission(iceRaftModel.TableName(), p)).
- Where("code = ?", code).
- Preload("IceRaftRecord").
- First(&iceRaftModel).Error
- if err != nil {
- // 冰排编号不存在,添加
- //if errors.Is(err, gorm.ErrRecordNotFound) {
- // iceRaftModel.Code = code
- // iceRaftModel.Status = model.IceRaftStatusNormal
- // iceRaftModel.DeptId = p.DeptId
- // iceRaftModel.CreateBy = p.UserId
- // err = tx.Create(&iceRaftModel).Error
- // if err != nil {
- // e.Log.Errorf("db error: %s", err)
- // return errors.New(fmt.Sprintf("添加冰排【%s】信息失败", code))
- // }
- //} else {
- // e.Log.Errorf("db error: %s", err)
- // return errors.New(fmt.Sprintf("获取冰排【%s】信息失败",code))
- //}
- e.Log.Errorf("db error: %s", err)
- return errors.New(fmt.Sprintf("获取冰排【%s】信息失败", code))
- }
- if iceRaftModel.Status != model.IceRaftStatusNormal {
- err = errors.New(fmt.Sprintf("冰排【%s】状态异常", code))
- return err
- }
- if iceRaftModel.IceRaftRecord.Status == model.IceRaftRecordStatusFreezing || iceRaftModel.IceRaftRecord.Status == model.IceRaftRecordStatusWaitUse {
- err = errors.New(fmt.Sprintf("冰排【%s】状态为%s,入库失败!", code, model.GetIceRaftRecordStatus(iceRaftModel.IceRaftRecord.Status)))
- return err
- }
- // 更新冰排记录状态 使用中 -> 已完成
- err = tx.Model(&model.IceRaftRecord{}).
- Where("ice_raft_id = ? and status = ?", iceRaftModel.Id, model.IceRaftRecordStatusUsing).
- Updates(map[string]interface{}{
- "status": model.IceRaftRecordStatusFinish,
- }).Error
- // 添加冰排记录
- var iceRaftRecordModel = model.IceRaftRecord{
- IceLockerId: iceLockerModel.Id,
- IceRaftId: iceRaftModel.Id,
- Status: model.IceRaftRecordStatusFreezing,
- Label: iceRaftModel.Label,
- Code: code,
- InStorageTime: model2.Time(time.Now()),
- //FreezeClaim: iceRaftModel.FreezeClaim,
- DeptBy: model2.DeptBy{
- DeptId: p.DeptId,
- },
- }
- iceRaftRecordModel.FreezeClaim = append(iceRaftRecordModel.FreezeClaim, iceRaftModel.FreezeClaim...)
- err = tx.Create(&iceRaftRecordModel).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return errors.New("添加冰排记录失败")
- }
- // 更新冰排使用记录
- err = tx.Model(&iceRaftModel).
- Where("id = ?", iceRaftModel.Id).
- Updates(map[string]interface{}{
- "ice_raft_record_id": iceRaftRecordModel.Id,
- }).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return errors.New("关联冰排记录失败")
- }
- }
- return nil
- }
- func (e *IceRaft) OutStorage(c *dto.IceRaftOutStorageReq, p *actions.DataPermission) error {
- var err error
- tx := e.Orm.Begin()
- defer func() {
- if err != nil {
- tx.Rollback()
- } else {
- tx.Commit()
- }
- }()
- // 查询保温箱信息
- var coolerBoxModel model.CoolerBox
- err = e.Orm.Scopes(actions.Permission(coolerBoxModel.TableName(), p)).
- First(&coolerBoxModel, c.CoolerBoxId).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return errors.New("获取保温箱信息失败")
- }
- if coolerBoxModel.Status != model.CoolerBoxStatusNormal {
- err = errors.New("保温箱状态异常")
- return err
- }
- var user model.SysUser
- first := e.Orm.Model(&model.SysUser{}).Where("id = ?", p.UserId).First(&user)
- if first.Error != nil {
- return global.GetFailedErr
- }
- for _, code := range c.Code {
- // 查询冰排是否存在
- var iceRaftModel model.IceRaft
- err = e.Orm.Scopes(actions.Permission(iceRaftModel.TableName(), p)).
- Where("code = ?", code).
- Preload("IceRaftRecord").
- First(&iceRaftModel).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return errors.New("获取冰排信息失败")
- }
- if iceRaftModel.Status != model.IceRaftStatusNormal {
- err = errors.New(fmt.Sprintf("冰排【%s】状态异常", code))
- return err
- }
- // 获取冰排记录
- var iceRaftRecordModel model.IceRaftRecord
- err = e.Orm.Scopes(actions.Permission(iceRaftRecordModel.TableName(), p)).
- Where("ice_raft_id = ?", iceRaftModel.Id).
- First(&iceRaftRecordModel, iceRaftModel.IceRaftRecordId).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return errors.New("获取冰排记录失败,请确认冰排是否入库!")
- }
- if len(iceRaftRecordModel.OutStorageTime.String()) > 0 {
- err = errors.New(fmt.Sprintf("冰排【%s】已出库,禁止出库!", code))
- return err
- }
- inTime := iceRaftRecordModel.InStorageTime.Local()
- sort.Slice(iceRaftRecordModel.FreezeClaim, func(i, j int) bool {
- return iceRaftRecordModel.FreezeClaim[i] < iceRaftRecordModel.FreezeClaim[j]
- })
- err, sysUser := GetUserProfile(e.Orm, p.UserId)
- if err != nil {
- err = errors.New("获取用户信息失败")
- return err
- }
- if sysUser.Dept.IsOutStorage {
- for _, v := range iceRaftRecordModel.FreezeClaim {
- if inTime.Add(time.Hour * time.Duration(v)).After(time.Now()) {
- str := fmt.Sprintf("冰排【%s】未达到冷冻时长【%v】,禁止出库", code, v)
- err = errors.New(str)
- return err
- }
- }
- }
- //if inTime.Add(time.Hour * time.Duration(iceRaftRecordModel.FreezeClaim[0])).After(time.Now()) {
- // err = errors.New(fmt.Sprintf("冰排【%s】未达到冷冻时长,禁止出库", code))
- // return err
- //}
- iceRaftRecordModel.Status = model.IceRaftRecordStatusUsing
- iceRaftRecordModel.CoolerBoxId = coolerBoxModel.Id
- iceRaftRecordModel.OutStorageTime = model2.Time(time.Now())
- iceRaftRecordModel.IceUseUsers = user.NickName
- iceRaftRecordModel.IceUseTime = model2.Time(time.Now())
- //data, err := nats_server.Read_DeviceTask_List_By_Condition(coolerBoxModel.Sn)
- //if data.T_State != 1 {
- // e.Log.Errorf("该保温箱未启动: %s", err)
- // return errors.New("该保温箱未启动,请启动保温箱监控")
- //}
- //if err != nil {
- // e.Log.Errorf("获取设备任务列表失败: %s", err)
- // return errors.New("该保温箱未绑定平台")
- //}
- //iceRaftRecordModel.CoolerBoxStarTime = model2.Time(data.T_Ut_start)
- err = tx.Save(&iceRaftRecordModel).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return errors.New("保存冰排记录失败")
- }
- }
- //添加保温箱记录
- var cooler model.CoolerBoxRecord
- cooler.CoolerBoxId = coolerBoxModel.Id
- cooler.CoolerUserTime = model2.Time(time.Now())
- cooler.Sn = coolerBoxModel.Sn
- cooler.Status = "1" //保温箱使用状态变为使用中
- cooler.DeptId = p.DeptId
- cooler.Name = coolerBoxModel.Name
- cooler.HistoryCode = append(cooler.HistoryCode, c.Code...)
- cooler.CoolerUseUsers = user.NickName
- //查询是否有正在使用中的保温箱记录
- var count int64
- err = e.Orm.Model(&model.CoolerBoxRecord{}).Where("id = ?", coolerBoxModel.Id).Where("status = ?", "1").Count(&count).Error
- if err != nil {
- e.Log.Errorf("获取保温箱信息失败: %s", err)
- return errors.New("获取保温箱信息失败")
- }
- //如果有正在使用中的保温箱记录,就更新
- if count > 0 {
- err = e.Orm.Updates(&cooler).Error
- if err != nil {
- e.Log.Errorf("db 更新保温箱记录失败: %s", err)
- return errors.New("更新保温箱记录失败")
- }
- } else {
- err = e.Orm.Create(&cooler).Error
- if err != nil {
- e.Log.Errorf("db 创建保温箱记录失败: %s", err)
- return errors.New("创建保温箱记录失败")
- }
- }
- return nil
- }
- // IsOutStorage 判断冰排是否达到冷冻时长
- func (e *IceRaft) IsOutStorage(c *dto.IceOutStorageReq, p *actions.DataPermission) (error, string) {
- var err error
- var str string
- // 获取冰排记录
- for _, v := range c.Code {
- var iceRaftRecordModel model.IceRaftRecord
- err = e.Orm.Scopes(actions.Permission(iceRaftRecordModel.TableName(), p)).
- Where("code = ?", v).
- Where("status = ?", model.IceRaftRecordStatusFreezing).Or("status = ?", model.IceRaftRecordStatusWaitUse).
- First(&iceRaftRecordModel).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return errors.New("获取冰排记录失败,请确认冰排是否入库!"), ""
- }
- if len(iceRaftRecordModel.OutStorageTime.String()) > 0 {
- err = errors.New(fmt.Sprintf("冰排【%s】已出库,禁止出库!", v))
- return err, ""
- }
- inTime := iceRaftRecordModel.InStorageTime.Local()
- sort.Slice(iceRaftRecordModel.FreezeClaim, func(i, j int) bool {
- return iceRaftRecordModel.FreezeClaim[i] < iceRaftRecordModel.FreezeClaim[j]
- })
- for _, vs := range iceRaftRecordModel.FreezeClaim {
- if inTime.Add(time.Hour * time.Duration(vs)).After(time.Now()) {
- str = fmt.Sprintf("冰排编号【%s】未达到冷冻时长【%v】小时,已达到冷冻时长【%v】,是否继续出库", v, vs, time.Now().Sub(inTime))
- break
- }
- }
- }
- return nil, str
- }
- // 查询保温箱关联的冰排
- func (e *IceRaft) GetPageByCoolerBoxId(coolerBoxId int, list *[]string, p *actions.DataPermission) error {
- var data model.IceRaft
- err := e.Orm.Model(&data).
- Select("ice_raft.code").
- Scopes(
- actions.Permission(data.TableName(), p),
- IceRaftNameRecordStatusScopes(model.IceRaftRecordStatusUsing),
- IceRaftNameRecordCoolerBoxIdScopes(coolerBoxId),
- ).
- Joins("left join ice_raft_record on ice_raft.ice_raft_record_id = ice_raft_record.id").
- Find(list).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.GetFailedErr
- }
- return nil
- }
- type IceRaftRecord struct {
- service.Service
- }
- // GetPage 获取IceRaft列表
- func (e *IceRaftRecord) GetPage(c *dto.IceRaftRecordGetPageReq, list *[]model.IceRaftRecord, count *int64, p *actions.DataPermission) error {
- var err error
- var data model.IceRaftRecord
- if c.PageSize == 9999 {
- err = e.Orm.Model(&data).
- Scopes(
- actions.Permission(data.TableName(), p),
- ).
- Preload("IceLocker").
- Preload("CoolerBox").
- Find(list).Limit(-1).Offset(-1).
- Count(count).Error
- } else {
- err = e.Orm.Model(&data).
- Scopes(
- cDto.MakeCondition(c.GetNeedSearch()),
- cDto.Paginate(c.GetPageSize(), c.GetPageIndex()),
- actions.Permission(data.TableName(), p),
- ).
- Preload("IceLocker").
- Preload("CoolerBox").
- Find(list).Limit(-1).Offset(-1).
- Count(count).Error
- }
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.GetFailedErr
- }
- return nil
- }
- // Update 修改IceRaft对象
- func (e *IceRaftRecord) Update(c *dto.IceRaftRecordUpdateReq, p *actions.DataPermission) error {
- var err error
- tx := e.Orm.Begin()
- defer func() {
- if err != nil {
- tx.Rollback()
- } else {
- tx.Commit()
- }
- }()
- var iceRaftRecordModel = model.IceRaftRecord{}
- // 查询冰排是否存在
- err = e.Orm.Scopes(actions.Permission(iceRaftRecordModel.TableName(), p)).
- First(&iceRaftRecordModel, 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
- }
- iceRaftRecordModel.IceLockerId = c.IceLockerId
- iceRaftRecordModel.FreezeClaim = c.FreezeClaim
- if len(iceRaftRecordModel.OutStorageTime.String()) > 0 {
- iceRaftRecordModel.CoolerBoxId = c.CoolerBoxId
- }
- err = tx.Save(&iceRaftRecordModel).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.UpdateFailedErr
- }
- c.Id = iceRaftRecordModel.Id
- return nil
- }
- // Remove 删除IceRaft
- func (e *IceRaftRecord) Remove(c *dto.IceRaftRecordDeleteReq, p *actions.DataPermission) error {
- var err error
- tx := e.Orm.Begin()
- defer func() {
- if err != nil {
- tx.Rollback()
- } else {
- tx.Commit()
- }
- }()
- var iceRaftRecordModel model.IceRaftRecord
- // 查询冰排是否存在
- err = e.Orm.Scopes(actions.Permission(iceRaftRecordModel.TableName(), p)).
- First(&iceRaftRecordModel, 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
- }
- db := tx.Delete(&iceRaftRecordModel)
- if err = db.Error; err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.DeleteFailedErr
- }
- if db.RowsAffected == 0 {
- return global.DeleteNotFoundOrNoPermissionErr
- }
- // 更改冰排关联状态
- var iceRaftModel model.IceRaft
- var secondIceRaftRecordModel model.IceRaftRecord
- err = e.Orm.Scopes(actions.Permission(iceRaftModel.TableName(), p)).
- First(&iceRaftModel, iceRaftRecordModel.IceRaftId).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.DeleteFailedErr
- }
- err = e.Orm.Scopes(actions.Permission(secondIceRaftRecordModel.TableName(), p)).
- Where("ice_raft_id = ? and id != ?", iceRaftRecordModel.IceRaftId, iceRaftRecordModel.Id).
- First(&secondIceRaftRecordModel).Error
- if err != nil {
- if errors.Is(err, gorm.ErrRecordNotFound) {
- iceRaftModel.IceRaftRecordId = 0
- }
- }
- iceRaftModel.IceRaftRecordId = secondIceRaftRecordModel.Id
- err = tx.Save(&iceRaftModel).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.UpdateFailedErr
- }
- return nil
- }
- func (e *IceRaftRecord) EndForCold(c *dto.IceRaftRecordEedReq, p *actions.DataPermission) error {
- var err error
- tx := e.Orm.Begin()
- defer func() {
- if err != nil {
- tx.Rollback()
- } else {
- tx.Commit()
- }
- }()
- // 查询冰排是否存在
- for _, v := range c.Id {
- var iceRaftRecordModel model.IceRaftRecord
- err = e.Orm.Scopes(actions.Permission(iceRaftRecordModel.TableName(), p)).
- First(&iceRaftRecordModel, v).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- if errors.Is(err, gorm.ErrRecordNotFound) {
- return global.UpdateNotFoundOrNoPermissionErr
- }
- return global.UpdateFailedErr
- }
- iceRaftRecordModel.IsSuitableForCold = 1 // 是否适冷
- iceRaftRecordModel.EndForColdTime = model2.Time(time.Now())
- iceRaftRecordModel.Id = v
- updates := tx.Where("id = ?", iceRaftRecordModel.Id).Updates(&iceRaftRecordModel)
- if err = updates.Error; err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.UpdateFailedErr
- }
- }
- return nil
- }
- // StartForCold 冰排开始释冷
- func (e *IceRaftRecord) StartForCold(c *dto.IceStartbleForColfTimReq, p *actions.DataPermission) error {
- var err error
- tx := e.Orm.Begin()
- defer func() {
- if err != nil {
- tx.Rollback()
- } else {
- tx.Commit()
- }
- }()
- var user model.SysUser
- first := e.Orm.Model(&model.SysUser{}).Where("id = ?", p.UserId).First(&user)
- if first.Error != nil {
- return global.GetFailedErr
- }
- //查询冰排是否存在
- for _, v := range c.Id {
- var iceRaftRecordModel model.IceRaftRecord
- var iceRaft model.IceRaft
- err = e.Orm.Scopes(actions.Permission(iceRaftRecordModel.TableName(), p)).
- First(&iceRaftRecordModel, v).Error
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- if errors.Is(err, gorm.ErrRecordNotFound) {
- return global.UpdateNotFoundOrNoPermissionErr
- }
- return global.UpdateFailedErr
- }
- err = e.Orm.Model(&iceRaft).Where("id = ?", iceRaftRecordModel.IceRaftId).Error
- if err != nil {
- return global.GetFailedErr
- }
- iceRaftRecordModel.StartIceColdTime = model2.Time(time.Now())
- iceRaftRecordModel.Status = model.IceRaftRecordReleasedInTheCold //释冷中
- iceRaftRecordModel.IceColdUsers = user.NickName
- iceRaftRecordModel.IceColdAddress = iceRaft.IceColdAddress
- iceRaftRecordModel.SuitableForCold = iceRaft.SuitableForCold
- updates := tx.Where("id = ?", iceRaftRecordModel.Id).Updates(&iceRaftRecordModel)
- if err = updates.Error; err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.UpdateFailedErr
- }
- }
- return nil
- }
- // IceRaftRecordRecording 获取冰排的历史记录
- func (e *IceRaft) IceRaftRecordRecording(c *dto.IceRaftRecordRecording, list *[]model.IceRaftRecord, count *int64, p *actions.DataPermission) error {
- var err error
- var data model.IceRaftRecord
- if c.PageSize == 9999 {
- err = e.Orm.Model(&data).
- Scopes(
- cDto.MakeCondition(c.GetNeedSearch()),
- actions.Permission(data.TableName(), p),
- IceRaftStartTimeAndEndTimeScopes(c.StartTime, c.EndTime),
- ).
- Where("status =?", model.IceRaftRecordStatusFinish).
- Preload("IceLocker").
- Preload("CoolerBox").
- Find(list).Limit(-1).Offset(-1).
- Count(count).Error
- } else {
- err = e.Orm.Model(&data).
- Scopes(
- cDto.MakeCondition(c.GetNeedSearch()),
- cDto.Paginate(c.GetPageSize(), c.GetPageIndex()),
- actions.Permission(data.TableName(), p),
- IceRaftStartTimeAndEndTimeScopes(c.StartTime, c.EndTime),
- ).
- Where("status =?", model.IceRaftRecordStatusFinish).
- Preload("IceLocker").
- Preload("CoolerBox").
- Find(list).Limit(-1).Offset(-1).
- Count(count).Error
- }
- if err != nil {
- e.Log.Errorf("db error: %s", err)
- return global.GetFailedErr
- }
- return nil
- }
|