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/nats/nats_server" "errors" "gogs.baozhida.cn/zoie/OAuth-core/service" "gorm.io/gorm" ) type CoolerBox struct { service.Service } // GetPage 获取CoolerBox列表 func (e *CoolerBox) GetPage(c *dto.CoolerBoxGetPageReq, list *[]model.CoolerBox, count *int64, p *actions.DataPermission) error { var err error var data model.CoolerBox 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 } if c.ShowTemp { // 获取公司秘钥 var company model.SysDept company, err = model.GetCompanyById(p.DeptId) if err != nil { e.Log.Errorf("db error: %s", err) return model.GetCompanyKeyErr } for i := 0; i < len(*list); i++ { deviceData, _, _ := nats_server.Cold_CompanyDeviceSensor_List_ByKey((*list)[i].Sn, company.ColdKey) if len(deviceData) > 0 { (*list)[i].DeviceData = deviceData[0].T_DeviceSensorData } } } return nil } // Get 获取CoolerBox对象 func (e *CoolerBox) Get(d *dto.CoolerBoxGetReq, CoolerBoxModel *model.CoolerBox, p *actions.DataPermission) error { err := e.Orm. Scopes(actions.Permission(CoolerBoxModel.TableName(), p)). First(CoolerBoxModel, 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 创建CoolerBox对象 func (e *CoolerBox) Insert(c *dto.CoolerBoxInsertReq, p *actions.DataPermission) error { var err error var data model.CoolerBox tx := e.Orm.Begin() defer func() { if err != nil { tx.Rollback() } else { tx.Commit() } }() var k int64 err = tx.Model(&data).Where("sn = ?", c.Sn).Count(&k).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.CreateFailedErr } if k > 0 { err = errors.New("该Sn已绑定!") e.Log.Errorf("db error: %s", err) return err } //var device nats_server.Device //var company model.SysDept //device, err = nats_server.Cold_ReadDeviceByT_sn(c.Sn) //if err != nil { // err = errors.New("获取SN信息失败,请检查SN是否正确!") // e.Log.Errorf("db error: %s", err) // return err //} //company, err = model.GetCompanyById(p.DeptId) //if err != nil { // e.Log.Errorf("db error: %s", err) // return err //} //if device.T_pid != company.Id { // err = errors.New("获取SN信息失败,请检查SN是否正确!") // e.Log.Errorf("获取SN信息失败,device.T_pid != company.Id") // return err //} // 添加保温箱 c.Generate(&data) data.HistorySn = []string{data.Sn} err = tx.Create(&data).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.CreateFailedErr } c.Id = data.Id return nil } func (e *CoolerBox) BatchInsert(c *dto.CoolerBoxBatchInsertReq) error { var err error tx := e.Orm.Begin() defer func() { if err != nil { tx.Rollback() } else { tx.Commit() } }() for _, coolerBox := range c.List { var data model.CoolerBox var k int64 err = tx.Model(&data).Where("sn = ?", coolerBox.Sn).Count(&k).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.CreateFailedErr } if k > 0 { //err = errors.New("该Sn已绑定!") continue } // 添加保温箱 coolerBox.CreateBy = c.CreateBy coolerBox.DeptId = c.DeptId coolerBox.Status = c.Status coolerBox.Generate(&data) data.HistorySn = []string{data.Sn} err = tx.Create(&data).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.CreateFailedErr } } return nil } // Update 修改CoolerBox对象 func (e *CoolerBox) Update(c *dto.CoolerBoxUpdateReq, 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.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 CoolerBoxModel.Sn != c.Sn && len(c.Sn) > 0 { var k int64 var data = model.CoolerBox{} err = tx.Model(&data).Where("sn = ?", c.Sn).Count(&k).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.CreateFailedErr } if k > 0 { err = errors.New("该Sn已绑定!") e.Log.Errorf("db error: %s", err) return err } //var device nats_server.Device //var company model.SysDept //device, err = nats_server.Cold_ReadDeviceByT_sn(c.Sn) //if err != nil { // err = errors.New("获取SN信息失败,请检查SN是否正确!") // e.Log.Errorf("db error: %s", err) // return err //} //company, err = model.GetCompanyById(p.DeptId) //if err != nil { // e.Log.Errorf("db error: %s", err) // return err //} //if device.T_pid != company.Id { // err = errors.New("获取SN信息失败,请检查SN是否正确!") // e.Log.Errorf("获取SN信息失败,device.T_pid != company.Id") // return err //} CoolerBoxModel.HistorySn = append(CoolerBoxModel.HistorySn, c.Sn) } c.Generate(&CoolerBoxModel) err = tx.Save(&CoolerBoxModel).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.UpdateFailedErr } c.Id = CoolerBoxModel.Id return nil } // Remove 删除CoolerBox func (e *CoolerBox) Remove(c *dto.CoolerBoxDeleteReq, 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 var count int64 err = e.Orm.Model(&iceRaftModel). Scopes( actions.Permission(iceRaftModel.TableName(), p), IceRaftNameRecordStatusScopes(model.IceRaftRecordStatusUsing), IceRaftNameRecordCoolerBoxIdScopes(c.Id), ). Where("ice_raft.status = '2'"). Joins("left join ice_raft_record on ice_raft.ice_raft_record_id = ice_raft_record.id"). Count(&count).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.DeleteFailedErr } if count > 0 { err = errors.New("该保温箱下有冰排正在使用中,无法删除!") return err } var CoolerBoxModel model.CoolerBox // 查询保温箱是否存在 err = e.Orm.Scopes(actions.Permission(CoolerBoxModel.TableName(), p)). First(&CoolerBoxModel, 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(&CoolerBoxModel) 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 }