62 lines
2.0 KiB
Go
62 lines
2.0 KiB
Go
package repos
|
|
|
|
import (
|
|
"context"
|
|
"gralias/models"
|
|
"time"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
type SessionsRepo interface {
|
|
SessionByToken(ctx context.Context, token string) (*models.Session, error)
|
|
SessionCreate(ctx context.Context, session *models.Session) error
|
|
SessionUpdate(ctx context.Context, session *models.Session) error
|
|
SessionDelete(ctx context.Context, token string) error
|
|
}
|
|
|
|
func (p *RepoProvider) SessionByToken(ctx context.Context, token string) (*models.Session, error) {
|
|
session := &models.Session{}
|
|
// The lifetime in the DB is in seconds, but in the model it is in minutes.
|
|
err := p.DB.GetContext(ctx, session, `SELECT id, updated_at, lifetime / 60 as lifetime, cookie_token, username FROM sessions WHERE cookie_token = ?`, token)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return session, nil
|
|
}
|
|
|
|
func (p *RepoProvider) SessionCreate(ctx context.Context, session *models.Session) error {
|
|
p.Ext = p.DB
|
|
tx, ok := ctx.Value("sqltx").(*sqlx.Tx)
|
|
if ok && tx != nil {
|
|
// how to know if it is a final exec in chain?
|
|
// or is it better to commit outside?
|
|
tocommit, ok := ctx.Value("tocommit").(bool)
|
|
if ok && tocommit {
|
|
defer func() {
|
|
if err := tx.Commit(); err != nil {
|
|
// log
|
|
// return err
|
|
}
|
|
}()
|
|
}
|
|
p.Ext = tx
|
|
}
|
|
// The lifetime in the model is in minutes, but in the DB it is in seconds.
|
|
_, err := p.Ext.Exec(`INSERT INTO sessions (updated_at, lifetime, cookie_token, username) VALUES (?, ?, ?, ?)`,
|
|
time.Now(), session.Lifetime*60, session.CookieToken, session.Username)
|
|
return err
|
|
}
|
|
|
|
func (p *RepoProvider) SessionUpdate(ctx context.Context, session *models.Session) error {
|
|
// The lifetime in the model is in minutes, but in the DB it is in seconds.
|
|
_, err := p.DB.ExecContext(ctx, `UPDATE sessions SET updated_at = ?, lifetime = ? WHERE cookie_token = ?`,
|
|
time.Now(), session.Lifetime*60, session.CookieToken)
|
|
return err
|
|
}
|
|
|
|
func (p *RepoProvider) SessionDelete(ctx context.Context, token string) error {
|
|
_, err := p.DB.ExecContext(ctx, `DELETE FROM sessions WHERE cookie_token = ?`, token)
|
|
return err
|
|
}
|