虽然现在的接口大部分都使用token保持状态,但session和cookie作为经典方案,我们还是有必须要了解下如何使用。golang原生路由和gin框架路由都没有实现自动开启cookie,需要我们手动去配置
项目开始阶段,我们先使用gin的模板渲染,在常规html页面实现聊天效果,最后再向vue过渡
创建模板文件
在根目录创建views目录,在目录中创建index.html和login.html文件
login.html
<!DOCTYPE html>
<html>
<head>
<meta charset="utf-8">
<title></title>
</head>
<body class="body">
<form method="post" action="/user/login" target="_self">
<input type="text" name="phone" value=""/>
<input type="text" name="password" value=""/>
<button type="submit" >登录</button>
</form>
</body>
</html>
index.html
<!DOCTYPE html>
<html>
<head>
<meta charset="utf-8">
<title></title>
</head>
<body class="body">
<p>欢迎{{.name}}</p>
<div id="content" style="border:1px solid chartreuse;width:300px;height:400px;margin: auto;">
</div>
<br/>
<input type="text" id="send-text" />
<br/>
<br/>
<button id="send">发送</button>
<input type="hidden" id="username" value="{{.name}}"/>
<input type="hidden" id="userid" value="{{.id}}"/>
<div id="msg">
</div>
</body>
</html>
改造路由文件
修改后的 routes/route.go 文件如下
package routes
import (
"fmt"
"github.com/gin-contrib/sessions"
"github.com/gin-contrib/sessions/cookie"
"github.com/gin-gonic/gin"
"project1/api"
)
func NewRouter() *gin.Engine{
r := gin.Default()
//创建随机的加密字符串作为发送给客户端的cookie
//cookie-secret是一个加盐参数,可以随意填写
store := cookie.NewStore([]byte("cookie-secret"))
//使用中间件,设置cookie名称,之后session值都基于该cookie设置和获取
r.Use(sessions.Sessions("cookie-name",store))
r.POST("/user/register", api.Register)
r.POST("/user/login", api.Login)
r.POST("/index", func(c *gin.Context){
// 声明模板文件所在目录
r.LoadHTMLGlob("views/*")
//通过session,判断登录状态
session := sessions.Default(c)
// userInfo在登录或注册成功时,写入session中
userInfoInterface := session.Get("userInfo")
//获取到用户数据
if userInfoInterface != nil{
//从session中获取的值,被包裹在一个接口类型中,需要知道值的原始类型,然后使用接口的断言获取到值
userInfo := userInfoInterface.(models.User)
c.HTML(200,"index.html",gin.H{"id":userInfo.ID,"name":userInfo.Name})
//c.JSON(200,gin.H{"status":200,"userInfo":userInfo,"logs":list})
}else{
// 未获取到用户数据,说明未登录,渲染登录页面
c.HTML(200,"login.html",gin.H{})
//c.JSON(402,gin.H{"status":402,"msg":"未登录:"})
}
})
return r
}
修改后,请求接口,发现header中已经携带了cookie信息
使用session
我们在第2弹:数据库链接和操作中,实现了注册和登录接口。现在我们可以把登录后的信息保存到seesion中,以保持用户的登录状态
api/user.go 中增加如下代码
import(
...
"github.com/gin-contrib/sessions"
"encoding/gob"
)
//保存信息到session中
func SaveUserToSession(name string ,data interface{},c *gin.Context) error{
session := sessions.Default(c)
//配置cookie的过期时间和作用域,paht一定要配置为/,否则path会被指定到url的目录中
//比如我访问uer/login,path被指定为/user,我在请求login时,死活获取不到cookie,耽误了很长时间
option := sessions.Options{MaxAge: 3600,Path:"/"}
session.Options(option)
// 设置session相当于是把对象序列化了,当对象是高级类型,如自定义的struct或map[string]interface{}{}时,需要先声明
// 此处我们的user是个自定义的struct,我们需要使用gob.register进行声明后,在设置session
//gob.Register()放在mian.go中执行一次即可,放在此处为了方便说明
gob.Register(models.User{})
session.Set(name,user)
err := session.Save()
return err
}
func Login(c *gin.Context){
...
if res.status == 1{
//登录成功
SaveUserToSession("userInfo",loginRule,c)
//跳转到首页
c.Redirect(301, "/index")
}
}