From 3cf0ee8441311e3318710aa4341c1e17660093e4 Mon Sep 17 00:00:00 2001
From: alber <alber_liu@qq.com>
Date: Sat, 18 Sep 2021 09:46:02 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/business/api/business_int.go | 20 ++------------------
 1 file changed, 2 insertions(+), 18 deletions(-)

diff --git a/internal/business/api/business_int.go b/internal/business/api/business_int.go
index 139dc58..e8549e7 100644
--- a/internal/business/api/business_int.go
+++ b/internal/business/api/business_int.go
@@ -17,15 +17,7 @@ func (*BusinessIntServer) GetUser(ctx context.Context, req *pb.GetUserReq) (*pb.
 	if err != nil {
 		return nil, err
 	}
-	return &pb.GetUserResp{User: &pb.User{
-		UserId:     user.Id,
-		Nickname:   user.Nickname,
-		Sex:        user.Sex,
-		AvatarUrl:  user.AvatarUrl,
-		Extra:      user.Extra,
-		CreateTime: user.CreateTime.Unix(),
-		UpdateTime: user.UpdateTime.Unix(),
-	}}, nil
+	return &pb.GetUserResp{User: user.ToProto()}, nil
 }
 
 func (*BusinessIntServer) GetUsers(ctx context.Context, req *pb.GetUsersReq) (*pb.GetUsersResp, error) {
@@ -41,15 +33,7 @@ func (*BusinessIntServer) GetUsers(ctx context.Context, req *pb.GetUsersReq) (*p
 
 	pbUsers := make(map[int64]*pb.User, len(users))
 	for i := range users {
-		pbUsers[users[i].Id] = &pb.User{
-			UserId:     users[i].Id,
-			Nickname:   users[i].Nickname,
-			Sex:        users[i].Sex,
-			AvatarUrl:  users[i].AvatarUrl,
-			Extra:      users[i].Extra,
-			CreateTime: users[i].CreateTime.Unix(),
-			UpdateTime: users[i].UpdateTime.Unix(),
-		}
+		pbUsers[users[i].Id] = users[i].ToProto()
 	}
 
 	return &pb.GetUsersResp{Users: pbUsers}, nil