usersmodelgen.go 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. // Code generated by goctl. DO NOT EDIT.
  2. // goctl 1.8.1
  3. package model
  4. import (
  5. "context"
  6. "github.com/zeromicro/go-zero/core/stores/mon"
  7. "go.mongodb.org/mongo-driver/bson"
  8. "go.mongodb.org/mongo-driver/bson/primitive"
  9. "go.mongodb.org/mongo-driver/mongo"
  10. )
  11. type usersModel interface {
  12. Insert(ctx context.Context, data *Users) error
  13. FindOne(ctx context.Context, id string) (*Users, error)
  14. Update(ctx context.Context, data *Users) (*mongo.UpdateResult, error)
  15. Delete(ctx context.Context, id string) (int64, error)
  16. }
  17. type defaultUsersModel struct {
  18. conn *mon.Model
  19. }
  20. func newDefaultUsersModel(conn *mon.Model) *defaultUsersModel {
  21. return &defaultUsersModel{conn: conn}
  22. }
  23. func (m *defaultUsersModel) Insert(ctx context.Context, data *Users) error {
  24. if data.ID.IsZero() {
  25. data.ID = primitive.NewObjectID()
  26. }
  27. _, err := m.conn.InsertOne(ctx, data)
  28. return err
  29. }
  30. func (m *defaultUsersModel) FindOne(ctx context.Context, id string) (*Users, error) {
  31. oid, err := primitive.ObjectIDFromHex(id)
  32. if err != nil {
  33. return nil, ErrInvalidObjectId
  34. }
  35. var data Users
  36. err = m.conn.FindOne(ctx, &data, bson.M{"_id": oid})
  37. switch err {
  38. case nil:
  39. return &data, nil
  40. case mon.ErrNotFound:
  41. return nil, ErrNotFound
  42. default:
  43. return nil, err
  44. }
  45. }
  46. func (m *defaultUsersModel) Update(ctx context.Context, data *Users) (*mongo.UpdateResult, error) {
  47. res, err := m.conn.UpdateOne(ctx, bson.M{"_id": data.ID}, bson.M{"$set": data})
  48. return res, err
  49. }
  50. func (m *defaultUsersModel) Delete(ctx context.Context, id string) (int64, error) {
  51. oid, err := primitive.ObjectIDFromHex(id)
  52. if err != nil {
  53. return 0, ErrInvalidObjectId
  54. }
  55. res, err := m.conn.DeleteOne(ctx, bson.M{"_id": oid})
  56. return res, err
  57. }