-
Notifications
You must be signed in to change notification settings - Fork 0
/
userservice.go
132 lines (116 loc) · 3.98 KB
/
userservice.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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
// Code generated by Kitex v0.4.4. DO NOT EDIT.
package userservice
import (
"context"
user "github.com/PICOF/simple-tiktok/kitex_gen/user"
client "github.com/cloudwego/kitex/client"
kitex "github.com/cloudwego/kitex/pkg/serviceinfo"
)
func serviceInfo() *kitex.ServiceInfo {
return userServiceServiceInfo
}
var userServiceServiceInfo = NewServiceInfo()
func NewServiceInfo() *kitex.ServiceInfo {
serviceName := "UserService"
handlerType := (*user.UserService)(nil)
methods := map[string]kitex.MethodInfo{
"UserRegister": kitex.NewMethodInfo(userRegisterHandler, newUserServiceUserRegisterArgs, newUserServiceUserRegisterResult, false),
"UserLogin": kitex.NewMethodInfo(userLoginHandler, newUserServiceUserLoginArgs, newUserServiceUserLoginResult, false),
"GetUserInfo": kitex.NewMethodInfo(getUserInfoHandler, newUserServiceGetUserInfoArgs, newUserServiceGetUserInfoResult, false),
}
extra := map[string]interface{}{
"PackageName": "user",
}
svcInfo := &kitex.ServiceInfo{
ServiceName: serviceName,
HandlerType: handlerType,
Methods: methods,
PayloadCodec: kitex.Thrift,
KiteXGenVersion: "v0.4.4",
Extra: extra,
}
return svcInfo
}
func userRegisterHandler(ctx context.Context, handler interface{}, arg, result interface{}) error {
realArg := arg.(*user.UserServiceUserRegisterArgs)
realResult := result.(*user.UserServiceUserRegisterResult)
success, err := handler.(user.UserService).UserRegister(ctx, realArg.Request)
if err != nil {
return err
}
realResult.Success = success
return nil
}
func newUserServiceUserRegisterArgs() interface{} {
return user.NewUserServiceUserRegisterArgs()
}
func newUserServiceUserRegisterResult() interface{} {
return user.NewUserServiceUserRegisterResult()
}
func userLoginHandler(ctx context.Context, handler interface{}, arg, result interface{}) error {
realArg := arg.(*user.UserServiceUserLoginArgs)
realResult := result.(*user.UserServiceUserLoginResult)
success, err := handler.(user.UserService).UserLogin(ctx, realArg.Request)
if err != nil {
return err
}
realResult.Success = success
return nil
}
func newUserServiceUserLoginArgs() interface{} {
return user.NewUserServiceUserLoginArgs()
}
func newUserServiceUserLoginResult() interface{} {
return user.NewUserServiceUserLoginResult()
}
func getUserInfoHandler(ctx context.Context, handler interface{}, arg, result interface{}) error {
realArg := arg.(*user.UserServiceGetUserInfoArgs)
realResult := result.(*user.UserServiceGetUserInfoResult)
success, err := handler.(user.UserService).GetUserInfo(ctx, realArg.Request)
if err != nil {
return err
}
realResult.Success = success
return nil
}
func newUserServiceGetUserInfoArgs() interface{} {
return user.NewUserServiceGetUserInfoArgs()
}
func newUserServiceGetUserInfoResult() interface{} {
return user.NewUserServiceGetUserInfoResult()
}
type kClient struct {
c client.Client
}
func newServiceClient(c client.Client) *kClient {
return &kClient{
c: c,
}
}
func (p *kClient) UserRegister(ctx context.Context, request *user.RegisterRequest) (r *user.RegisterResponse, err error) {
var _args user.UserServiceUserRegisterArgs
_args.Request = request
var _result user.UserServiceUserRegisterResult
if err = p.c.Call(ctx, "UserRegister", &_args, &_result); err != nil {
return
}
return _result.GetSuccess(), nil
}
func (p *kClient) UserLogin(ctx context.Context, request *user.LoginRequest) (r *user.LoginResponse, err error) {
var _args user.UserServiceUserLoginArgs
_args.Request = request
var _result user.UserServiceUserLoginResult
if err = p.c.Call(ctx, "UserLogin", &_args, &_result); err != nil {
return
}
return _result.GetSuccess(), nil
}
func (p *kClient) GetUserInfo(ctx context.Context, request *user.UserInfoRequest) (r *user.UserInfoResponse, err error) {
var _args user.UserServiceGetUserInfoArgs
_args.Request = request
var _result user.UserServiceGetUserInfoResult
if err = p.c.Call(ctx, "GetUserInfo", &_args, &_result); err != nil {
return
}
return _result.GetSuccess(), nil
}