More queries forrefresh token sqlc files.
This commit is contained in:
@@ -11,6 +11,18 @@ import (
|
||||
"github.com/jackc/pgx/v5/pgtype"
|
||||
)
|
||||
|
||||
const countUserRefreshTokens = `-- name: CountUserRefreshTokens :one
|
||||
SELECT count(*) FROM refresh_tokens
|
||||
WHERE user_id = $1
|
||||
`
|
||||
|
||||
func (q *Queries) CountUserRefreshTokens(ctx context.Context, userID int64) (int64, error) {
|
||||
row := q.db.QueryRow(ctx, countUserRefreshTokens, userID)
|
||||
var count int64
|
||||
err := row.Scan(&count)
|
||||
return count, err
|
||||
}
|
||||
|
||||
const createRefreshToken = `-- name: CreateRefreshToken :one
|
||||
INSERT INTO refresh_tokens (user_id, token_hash, expires_at)
|
||||
VALUES ($1, $2, $3)
|
||||
@@ -46,13 +58,34 @@ func (q *Queries) DeleteAllUserRefreshTokens(ctx context.Context, userID int64)
|
||||
return err
|
||||
}
|
||||
|
||||
const deleteRefreshToken = `-- name: DeleteRefreshToken :exec
|
||||
const deleteOldestRefreshToken = `-- name: DeleteOldestRefreshToken :exec
|
||||
DELETE FROM refresh_tokens
|
||||
WHERE token_hash = $1
|
||||
WHERE id = (
|
||||
SELECT id
|
||||
FROM refresh_tokens rt
|
||||
WHERE rt.user_id = $1
|
||||
ORDER BY rt.created_at ASC
|
||||
LIMIT 1
|
||||
)
|
||||
`
|
||||
|
||||
func (q *Queries) DeleteRefreshToken(ctx context.Context, tokenHash string) error {
|
||||
_, err := q.db.Exec(ctx, deleteRefreshToken, tokenHash)
|
||||
func (q *Queries) DeleteOldestRefreshToken(ctx context.Context, userID int64) error {
|
||||
_, err := q.db.Exec(ctx, deleteOldestRefreshToken, userID)
|
||||
return err
|
||||
}
|
||||
|
||||
const deleteRefreshToken = `-- name: DeleteRefreshToken :exec
|
||||
DELETE FROM refresh_tokens
|
||||
WHERE token_hash = $1 and user_id = $2
|
||||
`
|
||||
|
||||
type DeleteRefreshTokenParams struct {
|
||||
TokenHash string `json:"token_hash"`
|
||||
UserID int64 `json:"user_id"`
|
||||
}
|
||||
|
||||
func (q *Queries) DeleteRefreshToken(ctx context.Context, arg DeleteRefreshTokenParams) error {
|
||||
_, err := q.db.Exec(ctx, deleteRefreshToken, arg.TokenHash, arg.UserID)
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -104,3 +137,23 @@ func (q *Queries) GetUserRefreshTokens(ctx context.Context, userID int64) ([]Ref
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
|
||||
const getValidRefreshToken = `-- name: GetValidRefreshToken :one
|
||||
SELECT user_id, token_hash
|
||||
FROM refresh_tokens
|
||||
WHERE token_hash = $1
|
||||
AND expires_at > NOW()
|
||||
LIMIT 1
|
||||
`
|
||||
|
||||
type GetValidRefreshTokenRow struct {
|
||||
UserID int64 `json:"user_id"`
|
||||
TokenHash string `json:"token_hash"`
|
||||
}
|
||||
|
||||
func (q *Queries) GetValidRefreshToken(ctx context.Context, tokenHash string) (GetValidRefreshTokenRow, error) {
|
||||
row := q.db.QueryRow(ctx, getValidRefreshToken, tokenHash)
|
||||
var i GetValidRefreshTokenRow
|
||||
err := row.Scan(&i.UserID, &i.TokenHash)
|
||||
return i, err
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user