mirror of
https://github.com/idanoo/autobrr
synced 2025-07-23 08:49:13 +00:00
0e88117702
* feat(logger): add module context * feat(logger): change errors package * feat(logger): update tests
101 lines
2.2 KiB
Go
101 lines
2.2 KiB
Go
package database
|
|
|
|
import (
|
|
"database/sql"
|
|
"time"
|
|
|
|
"github.com/autobrr/autobrr/internal/domain"
|
|
"github.com/autobrr/autobrr/internal/logger"
|
|
"github.com/autobrr/autobrr/pkg/errors"
|
|
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
type FeedCacheRepo struct {
|
|
log zerolog.Logger
|
|
db *DB
|
|
}
|
|
|
|
func NewFeedCacheRepo(log logger.Logger, db *DB) domain.FeedCacheRepo {
|
|
return &FeedCacheRepo{
|
|
log: log.With().Str("repo", "feed_cache").Logger(),
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (r *FeedCacheRepo) Get(bucket string, key string) ([]byte, error) {
|
|
queryBuilder := r.db.squirrel.
|
|
Select(
|
|
"value",
|
|
"ttl",
|
|
).
|
|
From("feed_cache").
|
|
Where("bucket = ?", bucket).
|
|
Where("key = ?", key).
|
|
Where("ttl > ?", time.Now())
|
|
|
|
query, args, err := queryBuilder.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "error building query")
|
|
}
|
|
|
|
row := r.db.handler.QueryRow(query, args...)
|
|
if err := row.Err(); err != nil {
|
|
return nil, errors.Wrap(err, "error executing query")
|
|
}
|
|
|
|
var value []byte
|
|
var ttl time.Duration
|
|
|
|
if err := row.Scan(&value, &ttl); err != nil && err != sql.ErrNoRows {
|
|
return nil, errors.Wrap(err, "error scanning row")
|
|
}
|
|
|
|
return value, nil
|
|
}
|
|
|
|
func (r *FeedCacheRepo) Exists(bucket string, key string) (bool, error) {
|
|
queryBuilder := r.db.squirrel.
|
|
Select("1").
|
|
Prefix("SELECT EXISTS (").
|
|
From("feed_cache").
|
|
Where("bucket = ?", bucket).
|
|
Where("key = ?", key).
|
|
Suffix(")")
|
|
|
|
query, args, err := queryBuilder.ToSql()
|
|
if err != nil {
|
|
return false, errors.Wrap(err, "error building query")
|
|
}
|
|
|
|
var exists bool
|
|
err = r.db.handler.QueryRow(query, args...).Scan(&exists)
|
|
if err != nil && err != sql.ErrNoRows {
|
|
return false, errors.Wrap(err, "error query")
|
|
}
|
|
|
|
return exists, nil
|
|
}
|
|
|
|
func (r *FeedCacheRepo) Put(bucket string, key string, val []byte, ttl time.Time) error {
|
|
queryBuilder := r.db.squirrel.
|
|
Insert("feed_cache").
|
|
Columns("bucket", "key", "value", "ttl").
|
|
Values(bucket, key, val, ttl)
|
|
|
|
query, args, err := queryBuilder.ToSql()
|
|
if err != nil {
|
|
return errors.Wrap(err, "error building query")
|
|
}
|
|
|
|
if _, err = r.db.handler.Exec(query, args...); err != nil {
|
|
return errors.Wrap(err, "error executing query")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *FeedCacheRepo) Delete(bucket string, key string) error {
|
|
//TODO implement me
|
|
panic("implement me")
|
|
}
|