35 lines
669 B
Go
35 lines
669 B
Go
|
package sessions
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"github.com/gorilla/context"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
sessionName = "sessionName"
|
||
|
|
sessionStore = "sessionStore"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Middleware helps you set up a session in context
|
||
|
|
func Middleware(name string, store Store) gin.HandlerFunc {
|
||
|
|
return func(c *gin.Context) {
|
||
|
|
c.Set(sessionName, name)
|
||
|
|
c.Set(sessionStore, store)
|
||
|
|
defer context.Clear(c.Request)
|
||
|
|
c.Next()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get returns a session in the context
|
||
|
|
func Get(c *gin.Context) Session {
|
||
|
|
name := c.MustGet(sessionName).(string)
|
||
|
|
store := c.MustGet(sessionStore).(Store)
|
||
|
|
|
||
|
|
return &session{
|
||
|
|
name: name,
|
||
|
|
request: c.Request,
|
||
|
|
writer: c.Writer,
|
||
|
|
store: store,
|
||
|
|
}
|
||
|
|
}
|