/
user.go
93 lines (79 loc) · 2.13 KB
/
user.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package service
import (
"context"
"github.com/jinzhu/copier"
"github.com/toomanysource/atreus/app/user/service/internal/biz"
"github.com/go-kratos/kratos/v2/log"
pb "github.com/toomanysource/atreus/api/user/service/v1"
)
const (
CodeSuccess = 0
CodeFailed = 300
)
type UserService struct {
pb.UnimplementedUserServiceServer
log *log.Helper
uc *biz.UserUsecase
}
func NewUserService(uc *biz.UserUsecase, logger log.Logger) *UserService {
return &UserService{uc: uc, log: log.NewHelper(logger)}
}
func (s *UserService) UserRegister(ctx context.Context, req *pb.UserRegisterRequest) (*pb.UserRegisterReply, error) {
user, err := s.uc.Register(ctx, req.Username, req.Password)
if err != nil {
return &pb.UserRegisterReply{
StatusCode: CodeFailed,
StatusMsg: err.Error(),
}, nil
}
reply := &pb.UserRegisterReply{
StatusCode: CodeSuccess,
StatusMsg: "success",
UserId: user.Id,
Token: user.Token,
}
return reply, nil
}
func (s *UserService) UserLogin(ctx context.Context, req *pb.UserLoginRequest) (*pb.UserLoginReply, error) {
user, err := s.uc.Login(ctx, req.Username, req.Password)
if err != nil {
return &pb.UserLoginReply{
StatusCode: CodeFailed,
StatusMsg: err.Error(),
}, nil
}
reply := &pb.UserLoginReply{
StatusCode: CodeSuccess,
StatusMsg: "success",
UserId: user.Id,
Token: user.Token,
}
return reply, nil
}
func (s *UserService) GetUserInfo(ctx context.Context, req *pb.UserInfoRequest) (*pb.UserInfoReply, error) {
user, err := s.uc.GetInfo(ctx, req.UserId)
if err != nil {
return &pb.UserInfoReply{
StatusCode: CodeFailed,
StatusMsg: err.Error(),
}, nil
}
reply := &pb.UserInfoReply{
StatusCode: CodeSuccess,
StatusMsg: "success",
User: new(pb.User),
}
copier.Copy(reply.User, user)
return reply, nil
}
func (s *UserService) GetUserInfos(ctx context.Context, req *pb.UserInfosRequest) (*pb.UserInfosReply, error) {
users, err := s.uc.GetInfos(ctx, req.UserId, req.UserIds)
if err != nil {
return nil, err
}
reply := &pb.UserInfosReply{
Users: make([]*pb.User, len(users)),
}
copier.Copy(&reply.Users, &users)
return reply, nil
}