原文首发于 ISLAND
上一章节我们留下了一个小尾巴,那就是权限校验中间件并灭有进行玩啥,那么今天就对它进行完善。
???? Cookies
接上回,如何来判断用户是否登录呢? cookie
就是一个很好的凭证,cookie
会保留到浏览器上,每次访问请求的时候就会携带上,后端通过对 cookie
的识别,来判断用户是否登录。
由此一来,首先我们是要在登录的时候保存 cookie
。修改 userHandler
中 UserLogin
代码
在判断密码正确的时候,对 cookie
进行添加。
context.SetCookie("user_cookie", string(u.Id), 1000, "/", "localhost", false, true)
复制代码
这里来介绍一下这些参数,第一个参数为 cookie 名;第二个参数为 cookie 值;第三个参数为 cookie 有效时长,当 cookie 存在的时间超过设定时间时,cookie 就会失效,它就不再是我们有效的 cookie;第四个参数为 cookie 所在的目录;第五个为所在域,表示我们的 cookie 作用范围;第六个表示是否只能通过 https 访问;第七个表示 cookie 是否可以通过 js代码进行操作。
启动我们的项目,进行登录操作,F12 打开我们的控制台,选择Application ,在侧边栏找到 cookie,并且选择我们的站点,此时就可以在我们的右侧看到我们刚刚设置的 cookie 信息
并且有 Expires/Max-Age 属性,该属性表示 cookie 的过期时间,一旦超过这个时间,cookie 就会自动消失。
此时,已经表明我们的 cookie 设置成功。
????通过中间件获取Cookie
既然 cookie 已经获取成功,那么就该对我们新写的 Auth 中间件进行修改了。
我们首先建立一个 401.tmpl 文件,用于展示我们的权限不足时的信息。
{{template "header"}}
{{template "nav"}}
请先登录:<a href="/">首页</a>
复制代码
当我们要访问需要权限的路由时,恰巧我们权限不足,就会转跳到该页面。
此时再完善我们的中间件。
func Auth() gin.HandlerFunc {
return func(context *gin.Context) {
_, e := context.Request.Cookie("user_cookie")
if e == nil {
context.Next()
} else {
context.Abort()
context.HTML(http.StatusUnauthorized, "401.tmpl", nil)
}
}
}
复制代码
在这里我们通过 context.Request.Cookie()
来获取指定的 cookie
。这里还有一个函数 context.Abort()
表示对当前的请求进行中止。
重新启动项目,直接去访问 http://localhost:8080/user/profile/?id=5
,会出现我们的401
页面。当我们登录后再进行访问,就可以正常访问。
此时此刻说明我们的 auth
中间件起到了作用。
????刷新 cookie
在上面的 cookie
设置中有一个参数是为了设置过期时间,那么过期时间到底设置多长呢?过期时间设置过长,那么可能留下安全隐患,如果设置过短,又给用户带来不方便。
所以我们要在 cookie
请求成功的时候自动刷新我们的 cookie
时间。
修改我们中间件代码。
首先是获取到原来的 cookie
,然后就是重新设置一个 cookie
。
func Auth() gin.HandlerFunc {
return func(context *gin.Context) {
cookie, e := context.Request.Cookie("user_cookie")
if e == nil {
context.SetCookie(cookie.Name, cookie.Value, 1000, cookie.Path, cookie.Domain, cookie.Secure, cookie.HttpOnly)
context.Next()
} else {
context.Abort()
context.HTML(http.StatusUnauthorized, "401.tmpl", nil)
}
}
}
复制代码
重新启动我们的项目,仍旧是打开我们项目的控制台,每次访问需要权限的接口时,都会看到cookie的过期时间在重新刷新。
✍总结
本章节主要讲述了 cookie
如何添加,如何获取,又如何通过中间件来校验 cookie
和刷新 cookie
????????本章节代码
推荐阅读
Gin(一):Hello
Gin(二):路由Router
Gin(三):模板tmpl
Gin(四):表单提交校验和模型绑定
Gin(五):连接MySQL
Gin(六):文件的上传
Gin(七):中间件的使用和定义
个人公众号
最新文章都会在公众号上进行分享,欢迎各位大佬关注
有疑问加站长微信联系(非本文作者)