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
package controllers
import (
"ai_developer_admin/libs/redis"
"ai_developer_admin/libs/utils"
"ai_developer_admin/models"
"errors"
beego "github.com/beego/beego/v2/server/web"
"strconv"
"time"
)
type MainController struct {
beego.Controller
}
func (server *MainController) Get() {
server.Data["Website"] = "beego.me"
server.Data["Email"] = "astaxie@gmail.com"
server.TplName = "index.tpl"
}
func (server *MainController) respond(code int, message string, data ...interface{}) {
status := 200
if code == 401 {
status = code
}
server.Ctx.Output.SetStatus(status)
var d interface{}
if len(data) > 0 {
d = data[0]
}
//reData := struct {
// Code int `json:"code"`
// Data interface{} `json:"data,omitempty"`
//}{
// Code: code,
// Data: d,
//}
server.Data["json"] = struct {
Code int `json:"code"`
Message string `json:"message"`
Data interface{} `json:"data,omitempty"`
}{
Code: code,
Message: message,
Data: d,
}
server.ServeJSON()
}
func (server *MainController) Check() (*models.JwtPayload, error) {
token := server.Ctx.Input.Header("Authorization")
if token == "" {
return nil, errors.New("error: login expire")
}
// token := "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6MSwianNvbiI6ImdhbmdhbiIsImV4cCI6MTU3ODU1NDMyOCwiaWF0IjoxNTc4NTUzNzI4LCJpc3MiOiJnYW5nYW4ifQ.jOlMlfLMFBJvyrJTLagrwQx2931LzM7Z0EVMFZ75xYI"
info, err := utils.ValidateToken(token)
if err != nil {
return nil, err
// this.respond(401, "登陆失效")
}
key := "token:user-" + strconv.Itoa(info.UserID)
session, err := redis.GetDataToString(key)
if err != nil {
redis.DeleteKey(key)
return nil, errors.New("error: login expire")
}
//session := c.GetSession(key)
if session == "" {
return nil, errors.New("error: login expire")
}
if !(token == session) {
//c.DelSession(info.UserID)
redis.DeleteKey(key)
return nil, errors.New("error: login expire")
}
if info.ExpiresAt <= time.Now().Unix() {
//c.DelSession(info.UserID)
redis.DeleteKey(key)
return nil, errors.New("error: login expire")
}
return info, err
}
func (server *MainController) StoreToken(name interface{}, value interface{}) error {
return server.SetSession("username", "liuxuzhong")
}