update package

liuxiaolong

2020-05-27 95f8034c4ae22a73b12b320a9f24da64a767f1d7

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
package auth
 
import (
    "fmt"
    "github.com/gin-gonic/gin"
    "github.com/gomodule/redigo/redis"
    "net/http"
    "strings"
    "shop-common/code"
    "shop-common/util"
    "shop-common/cache"
)
 
const (
    TokenKey = "abc123456789"
)
 
type Auth interface {
    Check(c *gin.Context)bool
    Decode(token string) (bool,map[string]interface{})
    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
}
 
func SetOutUser(userId string) {
    c := cache.Get()
    defer c.Close()
    r, err := c.Do("SET", userId, userId)
    if err != nil {
        fmt.Println("SetOutUser err:", err, "reply:", r)
    }
}
 
func OutUser(userId string) bool {
    c := cache.Get()
    defer c.Close()
    b, err := redis.Bool(c.Do("EXISTS", userId))
    if err != nil {
        fmt.Println("OutUser err:", err)
        return false
    }
    return b
}
 
func RemoveOutUser(userId string) {
    c := cache.Get()
    defer c.Close()
    reply, err := c.Do("DEL", userId)
    if err != nil {
        fmt.Println("removeOutUser err:", err, "reply:", reply)
    }
}
 
func AuthHandler() gin.HandlerFunc {
    return func(c *gin.Context) {
        urlPath := c.Request.URL.Path
 
        if strings.Contains(urlPath,"/data/api-") && !filterUrls(urlPath){
            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 filterUrls(urlPath string) bool {
    freeArr := []string{
        "login",
        "makeVerifyCode",
        "/data/api-u/user/register",
        "/data/api-v/license",
        "/data/api-v/info/",
        "/data/api-u/area/findAreaByParentId",
        "/data/api-u/dic/findDicByType",
        "/data/api-u/user/verifyCode",
    }
    b := false
    for _,url := range freeArr {
        if strings.Contains(urlPath, url) {
            b = true
            break
        }
    }
    return b
}
 
 
func GetCurUser(c *gin.Context)map[string]interface{}{
    return (*GenerateAuthDriver()).User(c)
}