b_user_role.go 2.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. package internal
  2. import (
  3. "context"
  4. "github.com/gogf/gf/util/guid"
  5. "yx-dataset-server/app/errors"
  6. "yx-dataset-server/app/model"
  7. "yx-dataset-server/app/schema"
  8. )
  9. // NewUserRole 创建UserRole
  10. func NewUserRole(mUserRole model.IUserRole) *UserRole {
  11. return &UserRole{
  12. UserRoleModel: mUserRole,
  13. }
  14. }
  15. // UserRole 创建UserRole对象
  16. type UserRole struct {
  17. UserRoleModel model.IUserRole
  18. }
  19. // Query 查询数据
  20. func (a *UserRole) Query(ctx context.Context, params schema.UserRoleQueryParam, opts ...schema.UserRoleQueryOptions) (*schema.UserRoleQueryResult, error) {
  21. return a.UserRoleModel.Query(ctx, params, opts...)
  22. }
  23. // Get 查询指定数据
  24. func (a *UserRole) Get(ctx context.Context, recordID string, opts ...schema.UserRoleQueryOptions) (*schema.UserRole, error) {
  25. item, err := a.UserRoleModel.Get(ctx, recordID, opts...)
  26. if err != nil {
  27. return nil, err
  28. } else if item == nil {
  29. return nil, errors.ErrNotFound
  30. }
  31. return item, nil
  32. }
  33. func (a *UserRole) getUpdate(ctx context.Context, recordID string) (*schema.UserRole, error) {
  34. return a.Get(ctx, recordID)
  35. }
  36. // Create 创建数据
  37. func (a *UserRole) Create(ctx context.Context, item schema.UserRole) (*schema.UserRole, error) {
  38. item.RecordId = guid.S()
  39. err := a.UserRoleModel.Create(ctx, item)
  40. if err != nil {
  41. return nil, err
  42. }
  43. return a.getUpdate(ctx, item.RecordId)
  44. }
  45. // Update 更新数据
  46. func (a *UserRole) Update(ctx context.Context, recordID string, item schema.UserRole) (*schema.UserRole, error) {
  47. oldItem, err := a.UserRoleModel.Get(ctx, recordID)
  48. if err != nil {
  49. return nil, err
  50. } else if oldItem == nil {
  51. return nil, errors.ErrNotFound
  52. }
  53. err = a.UserRoleModel.Update(ctx, recordID, item)
  54. if err != nil {
  55. return nil, err
  56. }
  57. return a.getUpdate(ctx, recordID)
  58. }
  59. // Delete 删除数据
  60. func (a *UserRole) Delete(ctx context.Context, recordID string) error {
  61. oldItem, err := a.UserRoleModel.Get(ctx, recordID)
  62. if err != nil {
  63. return err
  64. } else if oldItem == nil {
  65. return errors.ErrNotFound
  66. }
  67. return a.UserRoleModel.Delete(ctx, recordID)
  68. }
  69. // UpdateStatus 更新状态
  70. func (a *UserRole) UpdateStatus(ctx context.Context, recordID string, status int) error {
  71. oldItem, err := a.UserRoleModel.Get(ctx, recordID)
  72. if err != nil {
  73. return err
  74. } else if oldItem == nil {
  75. return errors.ErrNotFound
  76. }
  77. return a.UserRoleModel.UpdateStatus(ctx, recordID, status)
  78. }