generated from CyanAsterisk/.github
/
chatservice.go
161 lines (142 loc) · 5.41 KB
/
chatservice.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
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
// Code generated by Kitex v0.4.4. DO NOT EDIT.
package chatservice
import (
"context"
chat "github.com/CyanAsterisk/TikGok/server/shared/kitex_gen/chat"
client "github.com/cloudwego/kitex/client"
kitex "github.com/cloudwego/kitex/pkg/serviceinfo"
)
func serviceInfo() *kitex.ServiceInfo {
return chatServiceServiceInfo
}
var chatServiceServiceInfo = NewServiceInfo()
func NewServiceInfo() *kitex.ServiceInfo {
serviceName := "ChatService"
handlerType := (*chat.ChatService)(nil)
methods := map[string]kitex.MethodInfo{
"GetChatHistory": kitex.NewMethodInfo(getChatHistoryHandler, newChatServiceGetChatHistoryArgs, newChatServiceGetChatHistoryResult, false),
"SentMessage": kitex.NewMethodInfo(sentMessageHandler, newChatServiceSentMessageArgs, newChatServiceSentMessageResult, false),
"GetLatestMessage": kitex.NewMethodInfo(getLatestMessageHandler, newChatServiceGetLatestMessageArgs, newChatServiceGetLatestMessageResult, false),
"BatchGetLatestMessage": kitex.NewMethodInfo(batchGetLatestMessageHandler, newChatServiceBatchGetLatestMessageArgs, newChatServiceBatchGetLatestMessageResult, false),
}
extra := map[string]interface{}{
"PackageName": "chat",
}
svcInfo := &kitex.ServiceInfo{
ServiceName: serviceName,
HandlerType: handlerType,
Methods: methods,
PayloadCodec: kitex.Thrift,
KiteXGenVersion: "v0.4.4",
Extra: extra,
}
return svcInfo
}
func getChatHistoryHandler(ctx context.Context, handler interface{}, arg, result interface{}) error {
realArg := arg.(*chat.ChatServiceGetChatHistoryArgs)
realResult := result.(*chat.ChatServiceGetChatHistoryResult)
success, err := handler.(chat.ChatService).GetChatHistory(ctx, realArg.Req)
if err != nil {
return err
}
realResult.Success = success
return nil
}
func newChatServiceGetChatHistoryArgs() interface{} {
return chat.NewChatServiceGetChatHistoryArgs()
}
func newChatServiceGetChatHistoryResult() interface{} {
return chat.NewChatServiceGetChatHistoryResult()
}
func sentMessageHandler(ctx context.Context, handler interface{}, arg, result interface{}) error {
realArg := arg.(*chat.ChatServiceSentMessageArgs)
realResult := result.(*chat.ChatServiceSentMessageResult)
success, err := handler.(chat.ChatService).SentMessage(ctx, realArg.Req)
if err != nil {
return err
}
realResult.Success = success
return nil
}
func newChatServiceSentMessageArgs() interface{} {
return chat.NewChatServiceSentMessageArgs()
}
func newChatServiceSentMessageResult() interface{} {
return chat.NewChatServiceSentMessageResult()
}
func getLatestMessageHandler(ctx context.Context, handler interface{}, arg, result interface{}) error {
realArg := arg.(*chat.ChatServiceGetLatestMessageArgs)
realResult := result.(*chat.ChatServiceGetLatestMessageResult)
success, err := handler.(chat.ChatService).GetLatestMessage(ctx, realArg.Req)
if err != nil {
return err
}
realResult.Success = success
return nil
}
func newChatServiceGetLatestMessageArgs() interface{} {
return chat.NewChatServiceGetLatestMessageArgs()
}
func newChatServiceGetLatestMessageResult() interface{} {
return chat.NewChatServiceGetLatestMessageResult()
}
func batchGetLatestMessageHandler(ctx context.Context, handler interface{}, arg, result interface{}) error {
realArg := arg.(*chat.ChatServiceBatchGetLatestMessageArgs)
realResult := result.(*chat.ChatServiceBatchGetLatestMessageResult)
success, err := handler.(chat.ChatService).BatchGetLatestMessage(ctx, realArg.Req)
if err != nil {
return err
}
realResult.Success = success
return nil
}
func newChatServiceBatchGetLatestMessageArgs() interface{} {
return chat.NewChatServiceBatchGetLatestMessageArgs()
}
func newChatServiceBatchGetLatestMessageResult() interface{} {
return chat.NewChatServiceBatchGetLatestMessageResult()
}
type kClient struct {
c client.Client
}
func newServiceClient(c client.Client) *kClient {
return &kClient{
c: c,
}
}
func (p *kClient) GetChatHistory(ctx context.Context, req *chat.DouyinMessageGetChatHistoryRequest) (r *chat.DouyinMessageGetChatHistoryResponse, err error) {
var _args chat.ChatServiceGetChatHistoryArgs
_args.Req = req
var _result chat.ChatServiceGetChatHistoryResult
if err = p.c.Call(ctx, "GetChatHistory", &_args, &_result); err != nil {
return
}
return _result.GetSuccess(), nil
}
func (p *kClient) SentMessage(ctx context.Context, req *chat.DouyinMessageActionRequest) (r *chat.DouyinMessageActionResponse, err error) {
var _args chat.ChatServiceSentMessageArgs
_args.Req = req
var _result chat.ChatServiceSentMessageResult
if err = p.c.Call(ctx, "SentMessage", &_args, &_result); err != nil {
return
}
return _result.GetSuccess(), nil
}
func (p *kClient) GetLatestMessage(ctx context.Context, req *chat.DouyinMessageGetLatestRequest) (r *chat.DouyinMessageGetLatestResponse, err error) {
var _args chat.ChatServiceGetLatestMessageArgs
_args.Req = req
var _result chat.ChatServiceGetLatestMessageResult
if err = p.c.Call(ctx, "GetLatestMessage", &_args, &_result); err != nil {
return
}
return _result.GetSuccess(), nil
}
func (p *kClient) BatchGetLatestMessage(ctx context.Context, req *chat.DouyinMessageBatchGetLatestRequest) (r *chat.DouyinMessageBatchGetLatestResponse, err error) {
var _args chat.ChatServiceBatchGetLatestMessageArgs
_args.Req = req
var _result chat.ChatServiceBatchGetLatestMessageResult
if err = p.c.Call(ctx, "BatchGetLatestMessage", &_args, &_result); err != nil {
return
}
return _result.GetSuccess(), nil
}