-
Notifications
You must be signed in to change notification settings - Fork 1
/
favorite_api.go
124 lines (107 loc) · 3.28 KB
/
favorite_api.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
// Code generated by hertz generator.
package api
import (
"context"
"github.com/bytecamp-galaxy/mini-tiktok/cmd/api/biz/jwt"
"github.com/bytecamp-galaxy/mini-tiktok/cmd/api/biz/model/api"
"github.com/bytecamp-galaxy/mini-tiktok/cmd/api/biz/pack"
"github.com/bytecamp-galaxy/mini-tiktok/internal/convert"
"github.com/bytecamp-galaxy/mini-tiktok/internal/rpc"
"github.com/bytecamp-galaxy/mini-tiktok/kitex_gen/favorite"
"github.com/bytecamp-galaxy/mini-tiktok/pkg/conf"
"github.com/bytecamp-galaxy/mini-tiktok/pkg/errno"
"github.com/bytecamp-galaxy/mini-tiktok/pkg/utils"
"github.com/cloudwego/hertz/pkg/app"
"github.com/cloudwego/hertz/pkg/protocol/consts"
"github.com/cloudwego/kitex/pkg/kerrors"
"github.com/marmotedu/errors"
)
// FavoriteAction .
// @router /douyin/favorite/action/ [POST]
func FavoriteAction(ctx context.Context, c *app.RequestContext) {
var err error
var req api.FavoriteActionRequest
err = c.BindAndValidate(&req)
if err != nil {
pack.Error(c, errors.WithCode(errno.ErrBindAndValidation, err.Error()))
return
}
uid, ok := c.Get(jwt.IdentityKey)
if !ok {
pack.Error(c, errors.WithCode(errno.ErrParseToken, ""))
return
}
reqRPC := &favorite.FavoriteActionRequest{
UserId: uid.(int64),
VideoId: req.VideoId,
ActionType: req.ActionType,
}
v := conf.Init()
cli, err := rpc.InitFavoriteClient(v.GetString("api-server.name"))
if err != nil {
pack.Error(c, errors.WithCode(errno.ErrClientRPCInit, err.Error()))
return
}
_, err = (*cli).FavoriteAction(ctx, reqRPC)
if err != nil {
if bizErr, ok := kerrors.FromBizStatusError(err); ok {
pack.Error(c, errors.WithCode(int(bizErr.BizStatusCode()), bizErr.BizMessage()))
return
} else {
pack.Error(c, errors.WithCode(errno.ErrRPCLink, err.Error()))
return
}
}
resp := &api.FavoriteActionResponse{
StatusCode: errno.ErrSuccess,
StatusMsg: utils.String(pack.SuccessStatusMessage),
}
c.JSON(consts.StatusOK, resp)
}
// FavoriteList .
// @router /douyin/favorite/list/ [GET]
func FavoriteList(ctx context.Context, c *app.RequestContext) {
var err error
var req api.FavoriteListRequest
err = c.BindAndValidate(&req)
if err != nil {
pack.Error(c, errors.WithCode(errno.ErrBindAndValidation, err.Error()))
return
}
userViewId, ok := c.Get(jwt.IdentityKey)
if !ok {
pack.Error(c, errors.WithCode(errno.ErrParseToken, ""))
return
}
// set up connection with comment server
v := conf.Init()
cli, err := rpc.InitFavoriteClient(v.GetString("api-server.name"))
if err != nil {
pack.Error(c, errors.WithCode(errno.ErrClientRPCInit, err.Error()))
return
}
reqRPC := favorite.FavoriteListRequest{
UserId: req.UserId,
UserViewId: userViewId.(int64),
}
respRPC, err := (*cli).FavoriteList(ctx, &reqRPC)
if err != nil {
if bizErr, ok := kerrors.FromBizStatusError(err); ok {
pack.Error(c, errors.WithCode(int(bizErr.BizStatusCode()), bizErr.BizMessage()))
return
} else {
pack.Error(c, errors.WithCode(errno.ErrRPCLink, err.Error()))
return
}
}
list := make([]*api.Video, len(respRPC.VideoList))
for i, video := range respRPC.VideoList {
list[i] = convert.VideoConverterAPI(video)
}
resp := &api.FavoriteListResponse{
StatusCode: errno.ErrSuccess,
StatusMsg: utils.String(pack.SuccessStatusMessage),
VideoList: list,
}
c.JSON(consts.StatusOK, resp)
}