74 lines
1.5 KiB
Go
74 lines
1.5 KiB
Go
package services
|
|
|
|
import (
|
|
"time"
|
|
)
|
|
|
|
type SessionsService struct {
|
|
ctx *Context
|
|
}
|
|
|
|
type SessionItem struct {
|
|
Id string `json:"id"`
|
|
ExpiresAt int64 `json:"expiresAt"`
|
|
}
|
|
|
|
func (s *SessionsService) GetById(id string) (*SessionItem, error) {
|
|
item := SessionItem{}
|
|
|
|
row := s.ctx.DB.QueryRow("SELECT id, expires_at FROM sessions WHERE id = ? AND expires_at > ?", id, time.Now().Unix())
|
|
|
|
err := row.Scan(&item.Id, &item.ExpiresAt)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &item, nil
|
|
}
|
|
|
|
func (s *SessionsService) Create() (*SessionItem, error) {
|
|
id, err := generateRandomString(128)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
item := SessionItem{
|
|
// TODO: The key is not guaranteed to be unique, how do we guarantee that?
|
|
Id: id,
|
|
ExpiresAt: generateExpiryDate().Unix(),
|
|
}
|
|
|
|
_, err = s.ctx.DB.Exec("INSERT INTO sessions (id, expires_at) VALUES (?, ?)", item.Id, item.ExpiresAt)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &item, nil
|
|
}
|
|
|
|
func (s *SessionsService) Destroy(sessionId string) error {
|
|
_, err := s.ctx.DB.Exec("DELETE FROM sessions WHERE id = ?", sessionId)
|
|
|
|
return err
|
|
}
|
|
|
|
func (s *SessionsService) Extend(session *SessionItem) error {
|
|
session.ExpiresAt = generateExpiryDate().Unix()
|
|
|
|
_, err := s.ctx.DB.Exec("UPDATE sessions SET expires_at = ? WHERE id = ?", session.ExpiresAt, session.Id)
|
|
|
|
return err
|
|
}
|
|
|
|
func (s *SessionsService) Cleanup() error {
|
|
_, err := s.ctx.DB.Exec("DELETE FROM sessions WHERE expires_at < ?", time.Now().Unix())
|
|
|
|
return err
|
|
}
|
|
|
|
func generateExpiryDate() time.Time {
|
|
return time.Now().Add(time.Hour * 24)
|
|
}
|