diff options
| -rw-r--r-- | main.go | 16 | ||||
| -rw-r--r-- | store.go | 6 |
2 files changed, 16 insertions, 6 deletions
@@ -87,13 +87,17 @@ func (app *App) loginHandler(w http.ResponseWriter, r *http.Request) { } func (app *App) logoutHandler(w http.ResponseWriter, r *http.Request) { - c := http.Cookie{ - Name: "id", - Value: "", - Domain: "." + app.Domain, - MaxAge: 0, + if s, ok := app.validate(r); ok { + // should we save old sessions in another table? + app.DeleteSession(s.Id) + c := http.Cookie{ + Name: "id", + Value: "", + Domain: "." + app.Domain, + MaxAge: -1, + } + http.SetCookie(w, &c) } - http.SetCookie(w, &c) http.Redirect(w, r, "/login", http.StatusSeeOther) } @@ -24,6 +24,7 @@ type Store interface { GetSession(id string) (*Session, bool) NewSession(userId int64) *Session GetUser(name string) (*User, bool) + DeleteSession(id string) } type PgStore struct { @@ -66,6 +67,11 @@ func (store *PgStore) NewSession(userId int64) *Session { return s } +func (store *PgStore) DeleteSession(id string) { + store.Query("DELETE FROM sessions WHERE id = $1", id) + delete(store.sessionCache, id) +} + func (store *PgStore) GetUser(name string) (*User, bool) { u := &User{Name: name} row := store.QueryRow( |
