160 lines
3.9 KiB
Go
160 lines
3.9 KiB
Go
// Code generated by sqlc. DO NOT EDIT.
|
|
// versions:
|
|
// sqlc v1.30.0
|
|
// source: refresh_tokens.sql
|
|
|
|
package db
|
|
|
|
import (
|
|
"context"
|
|
|
|
"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)
|
|
RETURNING id, user_id, token_hash, created_at, expires_at
|
|
`
|
|
|
|
type CreateRefreshTokenParams struct {
|
|
UserID int64 `json:"user_id"`
|
|
TokenHash string `json:"token_hash"`
|
|
ExpiresAt pgtype.Timestamptz `json:"expires_at"`
|
|
}
|
|
|
|
func (q *Queries) CreateRefreshToken(ctx context.Context, arg CreateRefreshTokenParams) (RefreshToken, error) {
|
|
row := q.db.QueryRow(ctx, createRefreshToken, arg.UserID, arg.TokenHash, arg.ExpiresAt)
|
|
var i RefreshToken
|
|
err := row.Scan(
|
|
&i.ID,
|
|
&i.UserID,
|
|
&i.TokenHash,
|
|
&i.CreatedAt,
|
|
&i.ExpiresAt,
|
|
)
|
|
return i, err
|
|
}
|
|
|
|
const deleteAllUserRefreshTokens = `-- name: DeleteAllUserRefreshTokens :exec
|
|
DELETE FROM refresh_tokens
|
|
WHERE user_id = $1
|
|
`
|
|
|
|
func (q *Queries) DeleteAllUserRefreshTokens(ctx context.Context, userID int64) error {
|
|
_, err := q.db.Exec(ctx, deleteAllUserRefreshTokens, userID)
|
|
return err
|
|
}
|
|
|
|
const deleteOldestRefreshToken = `-- name: DeleteOldestRefreshToken :exec
|
|
DELETE FROM refresh_tokens
|
|
WHERE id = (
|
|
SELECT id
|
|
FROM refresh_tokens rt
|
|
WHERE rt.user_id = $1
|
|
ORDER BY rt.created_at ASC
|
|
LIMIT 1
|
|
)
|
|
`
|
|
|
|
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
|
|
}
|
|
|
|
const getRefreshToken = `-- name: GetRefreshToken :one
|
|
SELECT id, user_id, token_hash, created_at, expires_at FROM refresh_tokens
|
|
WHERE token_hash = $1 LIMIT 1
|
|
`
|
|
|
|
func (q *Queries) GetRefreshToken(ctx context.Context, tokenHash string) (RefreshToken, error) {
|
|
row := q.db.QueryRow(ctx, getRefreshToken, tokenHash)
|
|
var i RefreshToken
|
|
err := row.Scan(
|
|
&i.ID,
|
|
&i.UserID,
|
|
&i.TokenHash,
|
|
&i.CreatedAt,
|
|
&i.ExpiresAt,
|
|
)
|
|
return i, err
|
|
}
|
|
|
|
const getUserRefreshTokens = `-- name: GetUserRefreshTokens :many
|
|
SELECT id, user_id, token_hash, created_at, expires_at FROM refresh_tokens
|
|
WHERE user_id = $1
|
|
`
|
|
|
|
func (q *Queries) GetUserRefreshTokens(ctx context.Context, userID int64) ([]RefreshToken, error) {
|
|
rows, err := q.db.Query(ctx, getUserRefreshTokens, userID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
var items []RefreshToken
|
|
for rows.Next() {
|
|
var i RefreshToken
|
|
if err := rows.Scan(
|
|
&i.ID,
|
|
&i.UserID,
|
|
&i.TokenHash,
|
|
&i.CreatedAt,
|
|
&i.ExpiresAt,
|
|
); err != nil {
|
|
return nil, err
|
|
}
|
|
items = append(items, i)
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
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
|
|
}
|