register.go 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. package service
  2. import (
  3. "context"
  4. "fmt"
  5. "time"
  6. "youngee_m_api/consts"
  7. "youngee_m_api/db"
  8. "youngee_m_api/redis"
  9. )
  10. var Register *register
  11. type register struct {
  12. sessionTTL time.Duration
  13. }
  14. func (r *register) CodeJudge(ctx context.Context, phone string, code string) (string, error) {
  15. vcode, err := r.getSession(ctx, phone)
  16. if err != nil {
  17. return "", err
  18. }
  19. // 校验验证码
  20. if *vcode != code {
  21. return "验证码有误", nil
  22. }
  23. return "", nil
  24. }
  25. func (r *register) AuthRegister(ctx context.Context, phone string, code string) (string, error) {
  26. // 获取验证码
  27. vcode, err := r.getSession(ctx, phone)
  28. if err != nil {
  29. return "", err
  30. }
  31. // 校验验证码
  32. fmt.Println("*vcode", *vcode)
  33. if *vcode != code {
  34. return "验证码有误", nil
  35. }
  36. // 获取用户信息
  37. user, err := db.GetUserByPhoneNumber(ctx, phone)
  38. if err != nil {
  39. return "", err
  40. }
  41. // 检查用户是否已存在
  42. if user != nil {
  43. return "手机号已存在", nil
  44. }
  45. return "", nil
  46. }
  47. func (r *register) getSession(ctx context.Context, phone string) (*string, error) {
  48. value, err := redis.Get(ctx, r.getRedisKey(phone))
  49. if err != nil {
  50. if err == consts.RedisNil {
  51. return nil, fmt.Errorf("not found in redis,phone:%+v", phone)
  52. }
  53. return nil, err
  54. }
  55. return &value, nil
  56. }
  57. func (r *register) getRedisKey(key string) string {
  58. return fmt.Sprintf("%s%s", consts.SessionRedisPrefix, key)
  59. }