Former-commit-id: 0dee4c6be389f47f938ba71c64d2224d610f7720
This commit is contained in:
Gerasimos (Makis) Maropoulos 2017-12-21 14:43:15 +02:00
commit e4ed775ce0
3 changed files with 10 additions and 6 deletions

View File

@ -58,7 +58,8 @@ type (
// AllowReclaim will allow to
// Destroy and Start a session in the same request handler.
// All it does is that it removes the cookie for both `Request` and `ResponseWriter`.
// All it does is that it removes the cookie for both `Request` and `ResponseWriter` while `Destroy`
// or add a new cookie to `Request` while `Start`.
//
// Defaults to false.
AllowReclaim bool

View File

@ -31,9 +31,11 @@ func GetCookie(ctx context.Context, name string) string {
}
// AddCookie adds a cookie
func AddCookie(ctx context.Context, cookie *http.Cookie) {
func AddCookie(ctx context.Context, cookie *http.Cookie, reclaim bool) {
// http.SetCookie(ctx.ResponseWriter(), cookie)
// ctx.Request().AddCookie(cookie)
if reclaim {
ctx.Request().AddCookie(cookie)
}
ctx.SetCookie(cookie)
}
@ -50,7 +52,7 @@ func RemoveCookie(ctx context.Context, name string, purge bool) {
c.MaxAge = -1
c.Value = ""
c.Path = "/"
AddCookie(ctx, c)
AddCookie(ctx, c, purge)
if purge {
// delete request's cookie also, which is temporary available.

View File

@ -96,7 +96,7 @@ func (s *Sessions) updateCookie(ctx context.Context, sid string, expires time.Du
// encode the session id cookie client value right before send it.
cookie.Value = s.encodeCookieValue(cookie.Value)
AddCookie(ctx, cookie)
AddCookie(ctx, cookie, s.config.AllowReclaim)
}
// Start should start the session for the particular request.
@ -131,7 +131,8 @@ func (s *Sessions) UpdateExpiration(ctx context.Context, expires time.Duration)
cookieValue := s.decodeCookieValue(GetCookie(ctx, s.config.Cookie))
if cookieValue != "" {
if s.provider.UpdateExpiration(cookieValue, expires) {
// we should also allow it to expire when the browser closed
if s.provider.UpdateExpiration(cookieValue, expires) || expires == -1 {
s.updateCookie(ctx, cookieValue, expires)
}
}