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 Address struct { service.Service } // GetPage 获取Address列表 func (e *Address) GetPage(c *dto.AddressGetPageReq, list *[]model.Address, count *int64) error { var err error var data model.Address err = e.Orm.Model(&data). Scopes( cDto.MakeCondition(c.GetNeedSearch()), cDto.Paginate(c.GetPageSize(), c.GetPageIndex()), ). 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 获取Address对象 func (e *Address) Get(d *dto.AddressGetReq, carInfoModel *model.Address, 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 创建Address对象 func (e *Address) Insert(c *dto.AddressInsertReq) error { var err error var data model.Address tx := e.Orm.Begin() defer func() { if err != nil { tx.Rollback() } else { tx.Commit() } }() // 查询是否有其他默认地址 var count int64 err = tx.Model(&model.Address{}).Where("is_default = ? and customer_id = ?", 1, c.CustomerId).Count(&count).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.CreateFailedErr } if count > 0 && c.IsDefault == true { err = tx.Model(&model.Address{}).Where("customer_id = ?", c.CustomerId). Updates(map[string]interface{}{ "is_default": 0, }).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.CreateFailedErr } } // 添加角色 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 修改Address对象 func (e *Address) Update(c *dto.AddressUpdateReq, p *actions.DataPermission) error { var err error tx := e.Orm.Begin() defer func() { if err != nil { tx.Rollback() } else { tx.Commit() } }() var carInfoModel = model.Address{} // 查询角色是否存在 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 } if c.IsDefault == true && c.IsDefault != carInfoModel.IsDefault { err = tx.Model(&model.Address{}).Where("customer_id = ?", carInfoModel.CustomerId). Updates(map[string]interface{}{ "is_default": 0, }).Error if err != nil { e.Log.Errorf("db error: %s", err) return global.CreateFailedErr } } 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 删除Address func (e *Address) Remove(c *dto.AddressDeleteReq, p *actions.DataPermission) error { var err error tx := e.Orm.Begin() defer func() { if err != nil { tx.Rollback() } else { tx.Commit() } }() var carInfoModel model.Address // 查询角色是否存在 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 }