user.go 631 B

12345678910111213141516171819202122232425262728293031
  1. package db
  2. import (
  3. "context"
  4. "youngee_b_api/model/gorm_model"
  5. "gorm.io/gorm"
  6. )
  7. func CreateUser(ctx context.Context, user gorm_model.User) (*int, error) {
  8. db := GetReadDB(ctx)
  9. err := db.Create(&user).Error
  10. if err != nil {
  11. return nil, err
  12. }
  13. return &user.ID, nil
  14. }
  15. //GetUserByPhone 查不到返回空
  16. func GetUserByPhone(ctx context.Context, phone string) (*gorm_model.User, error) {
  17. db := GetReadDB(ctx)
  18. user := &gorm_model.User{}
  19. err := db.Model(user).Where("phone = ?", phone).First(user).Error
  20. if err != nil {
  21. if err == gorm.ErrRecordNotFound {
  22. return nil, nil
  23. }
  24. return nil, err
  25. }
  26. return user, nil
  27. }