add refresh token to resp

liuxiaolong

2020-06-05 7c811247ecf143e08c576986a884bedadc57dd66

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
package auth
 
import (
    "github.com/gin-gonic/gin"
    "net/http"
    "strings"
    "sync"
    "webserver/extend/code"
    "webserver/extend/util"
)
 
const (
    TokenKey = "abc123456789"
)
 
type Auth interface {
    Check(c *gin.Context)bool
    User(c *gin.Context)map[string]interface{}
    Login(http *http.Request,w http.ResponseWriter,user map[string]interface{}) (bool, string, string)
    Logout(http *http.Request,w http.ResponseWriter) bool
    RefreshToken(tokenStr string) (bool, string, string)
}
 
func GenerateAuthDriver() *Auth {
    var authDriver Auth
    authDriver = NewJwtAuthDriver()
    return &authDriver
}
 
var outUserM = make(map[string]string,0)
var lock sync.RWMutex
 
func SetOutUser(userId string) {
    lock.Lock()
    defer lock.Unlock()
    outUserM[userId] = userId
}
 
func OutUser(userId string) bool {
    lock.Lock()
    defer lock.Unlock()
    if _,ok := outUserM[userId];ok{
        return true
    }
    return false
}
 
func RemoveOutUser(userId string) {
    lock.Lock()
    defer lock.Unlock()
    if _,ok := outUserM[userId];ok{
        delete(outUserM,userId)
    }
}
 
func AuthHandler() gin.HandlerFunc {
    return func(c *gin.Context) {
        urlPath := c.Request.URL.Path
 
        if strings.Contains(urlPath,"/data/api-") && !strings.Contains(urlPath,"login") && !strings.Contains(urlPath, "/data/api-v/license") && !strings.Contains(urlPath, "/data/api-v/info/"){
            jwtDriver :=NewJwtAuthDriver()
            if !jwtDriver.Check(c) {
                util.ResponseFormat(c,code.TokenNotFound,"尚未登录，请登录")
                c.Abort()
                return
            }
            userM := (*jwtDriver).User(c)
            if userM == nil {
                util.ResponseFormat(c,code.TokenNotFound,"尚未登录，请登录")
                c.Abort()
                return
            }
            userId := userM["id"].(string)
            if OutUser(userId) {
                util.ResponseFormat(c,code.TokenNotFound,"尚未登录，请登录")
                c.Abort()
                return
            }
            c.Next()
        } else {
            c.Next()
        }
    }
}
 
func GetCurUser(c *gin.Context)map[string]interface{}{
    return (*GenerateAuthDriver()).User(c)
}