Feat: session db methods and try at tx

repos/main.go

@@ -15,7 +15,8 @@ type AllRepos interface {
 }
 
 type RepoProvider struct {
-	DB *sqlx.DB
+	DB  *sqlx.DB
+	Ext sqlx.Ext
 }
 
 func NewRepoProvider(pathToDB string) *RepoProvider {

repos/session.go

@@ -1,9 +1,61 @@
 package repos
 
-import "gralias/models"
+import (
+	"context"
+	"gralias/models"
+	"time"
+
+	"github.com/jmoiron/sqlx"
+)
 
 type SessionsRepo interface {
-	SessionByToken(token string) (*models.Session, error)
-	SessionCreate(req *models.Session) error
-	SessionUpdateLifetime(minutes uint32) error
+	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
 }