Merge branch 'demo' of github.com:wangzijian2002/LinkMe into demo

完善互动信息接口
Forrest-Tao · May 28, 2024 · 2a52293 · 2a52293
2 parents a74b20b + e2233d5
commit 2a52293
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/internal/repository/dao/user.go b/internal/repository/dao/user.go
@@ -19,6 +19,7 @@ type UserDAO interface {
 	CreateUser(ctx context.Context, u User) error
 	FindByID(ctx context.Context, id int64) (User, error)
 	FindByEmail(ctx context.Context, email string) (User, error)
+	FindByPhone(ctx context.Context, phone string) (User, error)
 }
 
 type userDAO struct {
@@ -76,3 +77,16 @@ func (ud *userDAO) FindByEmail(ctx context.Context, email string) (User, error)
 	}
 	return user, nil
 }
+
+// FindByPhone 根据phone查询用户信息
+func (ud *userDAO) FindByPhone(ctx context.Context, phone string) (User, error) {
+	var user User
+	err := ud.db.WithContext(ctx).Where("phone = ?", phone).First(&user).Error
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return User{}, ErrUserNotFound
+		}
+		return User{}, err
+	}
+	return user, nil
+}
diff --git a/internal/repository/user.go b/internal/repository/user.go
@@ -64,8 +64,8 @@ func (ur *userRepository) FindByID(ctx context.Context, id int64) (domain.User,
 
 // FindByPhone 通过电话查询用户
 func (ur *userRepository) FindByPhone(ctx context.Context, phone string) (domain.User, error) {
-	//TODO implement me
-	panic("implement me")
+	u, err := ur.dao.FindByEmail(ctx, phone)
+	return toDomainUser(u), err
 }
 
 // FindByEmail 通过Email查询用户