Mathis/lemma
1
0
Fork 0
mirror of https://github.com/lordmathis/lemma.git synced 2025-11-05 15:44:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use ScanStruct in sessions

Browse Source
This commit is contained in:
Matúš Námešný
2025-03-05 21:07:05 +01:00
parent 0f97927219
commit 976425d660
1 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
server/internal/db/sessions.go
View File

@@ -26,15 +26,18 @@ func (db *database) CreateSession(session *models.Session) error {
// GetSessionByRefreshToken retrieves a session by its refresh token // GetSessionByRefreshToken retrieves a session by its refresh token
func (db *database) GetSessionByRefreshToken(refreshToken string) (*models.Session, error) { func (db *database) GetSessionByRefreshToken(refreshToken string) (*models.Session, error) {
session := &models.Session{} session := &models.Session{}
query := db.NewQuery(). query := db.NewQuery()
Select("id", "user_id", "refresh_token", "expires_at", "created_at"). query, err := query.SelectStruct(session, "sessions")
From("sessions"). if err != nil {
Where("refresh_token = "). return nil, fmt.Errorf("failed to create query: %w", err)
}
query = query.Where("refresh_token = ").
Placeholder(refreshToken). Placeholder(refreshToken).
And("expires_at >"). And("expires_at >").
Placeholder(time.Now()) Placeholder(time.Now())
err := db.QueryRow(query.String(), query.Args()...).Scan(&session.ID, &session.UserID, &session.RefreshToken, &session.ExpiresAt, &session.CreatedAt)
row := db.QueryRow(query.String(), query.Args()...)
err = db.ScanStruct(row, session)
if err == sql.ErrNoRows { if err == sql.ErrNoRows {
return nil, fmt.Errorf("session not found or expired") return nil, fmt.Errorf("session not found or expired")
} }
@@ -48,15 +51,18 @@ func (db *database) GetSessionByRefreshToken(refreshToken string) (*models.Sessi
// GetSessionByID retrieves a session by its ID // GetSessionByID retrieves a session by its ID
func (db *database) GetSessionByID(sessionID string) (*models.Session, error) { func (db *database) GetSessionByID(sessionID string) (*models.Session, error) {
session := &models.Session{} session := &models.Session{}
query := db.NewQuery(). query := db.NewQuery()
Select("id", "user_id", "refresh_token", "expires_at", "created_at"). query, err := query.SelectStruct(session, "sessions")
From("sessions"). if err != nil {
Where("id = "). return nil, fmt.Errorf("failed to create query: %w", err)
}
query = query.Where("id = ").
Placeholder(sessionID). Placeholder(sessionID).
And("expires_at >"). And("expires_at >").
Placeholder(time.Now()) Placeholder(time.Now())
err := db.QueryRow(query.String(), query.Args()...).Scan(&session.ID, &session.UserID, &session.RefreshToken, &session.ExpiresAt, &session.CreatedAt)
row := db.QueryRow(query.String(), query.Args()...)
err = db.ScanStruct(row, session)
if err == sql.ErrNoRows { if err == sql.ErrNoRows {
return nil, fmt.Errorf("session not found") return nil, fmt.Errorf("session not found")
} }