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) }