-
-
Notifications
You must be signed in to change notification settings - Fork 1
/
api_service.go
137 lines (114 loc) · 3.1 KB
/
api_service.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
// Code generated by hertz generator.
package api
import (
"context"
"errors"
"time"
"github.com/L2ncE/CloudWeGo-101/api/biz/global"
"github.com/L2ncE/CloudWeGo-101/api/biz/middleware"
"github.com/L2ncE/CloudWeGo-101/api/biz/model/api"
"github.com/L2ncE/CloudWeGo-101/kitex_gen/article"
"github.com/L2ncE/CloudWeGo-101/kitex_gen/user"
"github.com/cloudwego/hertz/pkg/app"
"github.com/cloudwego/hertz/pkg/protocol/consts"
"github.com/dgrijalva/jwt-go"
)
// Register .
// @router /user/register/ [POST]
func Register(ctx context.Context, c *app.RequestContext) {
var err error
var req api.RegisterRequest
err = c.BindAndValidate(&req)
if err != nil {
c.String(consts.StatusBadRequest, err.Error())
return
}
resp := new(api.ResgisterResponse)
c.JSON(consts.StatusOK, resp)
}
// Login .
// @router /user/login/ [POST]
func Login(ctx context.Context, c *app.RequestContext) {
var err error
var req api.LoginRequest
err = c.BindAndValidate(&req)
if err != nil {
c.String(consts.StatusBadRequest, err.Error())
return
}
resp, err := global.GlobalUserClient.Login(ctx, &user.LoginRequest{
Username: req.Username,
Password: req.Password,
})
if err != nil {
c.String(consts.StatusBadRequest, err.Error())
return
}
j := middleware.NewJWT()
claims := middleware.CustomClaims{
ID: resp.Uid,
StandardClaims: jwt.StandardClaims{
NotBefore: time.Now().Unix(),
ExpiresAt: time.Now().Unix() + 60*60*24*30,
Issuer: "L2ncE",
},
}
token, err := j.CreateToken(claims)
c.JSON(consts.StatusOK, &api.LoginResponse{Token: token})
}
// GetArticleNum .
// @router /user/article [GET]
func GetArticleNum(ctx context.Context, c *app.RequestContext) {
aid, flag := c.Get("userId")
if !flag {
c.String(consts.StatusBadRequest, errors.New("wrong userid").Error())
return
}
resp, err := global.GlobalUserClient.GetArticleNum(ctx, &user.GetArticleNumRequest{UserId: aid.(int64)})
if err != nil {
c.String(consts.StatusBadRequest, err.Error())
return
}
c.JSON(consts.StatusOK, resp)
}
// AddArticleNum .
// @router /user/article [POST]
func AddArticleNum(ctx context.Context, c *app.RequestContext) {
aid, flag := c.Get("userId")
if !flag {
c.String(consts.StatusBadRequest, errors.New("wrong userid").Error())
return
}
resp, err := global.GlobalUserClient.AddArticleNum(ctx, &user.AddArticleNumRequest{UserId: aid.(int64)})
if err != nil {
c.String(consts.StatusBadRequest, err.Error())
return
}
c.JSON(consts.StatusOK, resp)
}
// PostArticle .
// @router article [POST]
func PostArticle(ctx context.Context, c *app.RequestContext) {
var err error
var req api.PostArticleRequest
err = c.BindAndValidate(&req)
if err != nil {
c.String(consts.StatusBadRequest, err.Error())
return
}
aid, flag := c.Get("userId")
if !flag {
c.String(consts.StatusBadRequest, errors.New("wrong userid").Error())
return
}
resp, err := global.GlobalArticleClient.PostArticle(ctx, &article.PostArticleRequest{
Title: req.Title,
Content: req.Content,
Uid: aid.(int64),
})
if err != nil {
c.String(consts.StatusBadRequest, err.Error())
return
}
c.JSON(consts.StatusOK, resp)
}