register.go 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package service
  2. import (
  3. "context"
  4. "fmt"
  5. "time"
  6. "youngee_b_api/consts"
  7. "youngee_b_api/db"
  8. "youngee_b_api/redis"
  9. )
  10. var Register *register
  11. type register struct {
  12. sessionTTL time.Duration
  13. }
  14. func (r *register) Register(ctx context.Context, phone string, vcode string) (*int64, error) {
  15. user, err := db.GetUserByPhone(ctx, phone)
  16. if err != nil {
  17. return nil, err
  18. } else if user != nil {
  19. return &user.ID, nil
  20. } else {
  21. code, err := r.getSession(ctx, phone)
  22. if err != nil {
  23. return nil, err
  24. } else {
  25. fmt.Printf("%+v", code)
  26. if *code != vcode {
  27. return &user.ID, nil
  28. } else {
  29. return nil, nil
  30. }
  31. }
  32. }
  33. }
  34. func (r *register) getSession(ctx context.Context, phone string) (*string, error) {
  35. value, err := redis.Get(ctx, r.getRedisKey(phone))
  36. if err != nil {
  37. if err == consts.RedisNil {
  38. return nil, fmt.Errorf("not found in redis,phone:%+v", phone)
  39. }
  40. return nil, err
  41. }
  42. return &value, nil
  43. }
  44. func (r *register) getRedisKey(key string) string {
  45. return fmt.Sprintf("%s%s", consts.SessionRedisPrefix, key)
  46. }