package service import ( "errors" "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" "gogs.baozhida.cn/zoie/OAuth-core/service" "gorm.io/gorm" ) type Goods struct { service.Service } func GoodsStationIdScopes(isShow int) func(db *gorm.DB) *gorm.DB { return func(db *gorm.DB) *gorm.DB { if isShow == 1 { return db.Where("is_show = 1") } if isShow == 2 { return db.Where("is_show = 0") } return db } } // GetPage 获取Goods列表 func (e *Goods) GetPage(c *dto.GoodsGetPageReq, list *[]model.Goods, count *int64, p *actions.DataPermission) error { var err error var data model.Goods err = e.Orm.Model(&data). Scopes( cDto.MakeCondition(c.GetNeedSearch()), cDto.Paginate(c.GetPageSize(), c.GetPageIndex()), actions.Permission(data.TableName(), p), GoodsStationIdScopes(c.IsShow), ). 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 获取Goods对象 func (e *Goods) Get(d *dto.GoodsGetReq, carInfoModel *model.Goods, p *actions.DataPermission) error { err := e.Orm. Scopes(actions.Permission(carInfoModel.TableName(), p)). First(carInfoModel, 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 创建Goods对象 func (e *Goods) Insert(c *dto.GoodsInsertReq) error { var err error var data model.Goods 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 修改Goods对象 func (e *Goods) Update(c *dto.GoodsUpdateReq, p *actions.DataPermission) error { var err error tx := e.Orm.Begin() defer func() { if err != nil { tx.Rollback() } else { tx.Commit() } }() var carInfoModel = model.Goods{} // 查询角色是否存在 err = e.Orm.Scopes(actions.Permission(carInfoModel.TableName(), p)). First(&carInfoModel, 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(&carInfoModel) err = tx.Save(&carInfoModel).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.UpdateFailedErr } c.Id = carInfoModel.Id return nil } // Remove 删除Goods func (e *Goods) Remove(c *dto.GoodsDeleteReq, p *actions.DataPermission) error { var err error tx := e.Orm.Begin() defer func() { if err != nil { tx.Rollback() } else { tx.Commit() } }() var carInfoModel model.Goods // 查询角色是否存在 err = e.Orm.Scopes(actions.Permission(carInfoModel.TableName(), p)). First(&carInfoModel, 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(&carInfoModel) 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 *Goods) GetAppletPage(c *dto.AppletGoodsGetPageReq, list *[]model.Goods, count *int64) error { var err error var data model.Goods err = e.Orm.Model(&data). Scopes( cDto.MakeCondition(c.GetNeedSearch()), cDto.Paginate(c.GetPageSize(), c.GetPageIndex()), GoodsStationIdScopes(c.IsShow), ). Find(list).Limit(-1).Offset(-1). Count(count).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.GetFailedErr } return nil }