71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
package model
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"lyra/internal/cache"
|
|
"lyra/internal/pkg/constant"
|
|
"lyra/internal/util"
|
|
|
|
sq "github.com/Masterminds/squirrel"
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
type Category struct {
|
|
ID int64 `db:"id"`
|
|
Slug string `db:"slug"`
|
|
Title string `db:"title"`
|
|
Sort int `db:"sort"`
|
|
}
|
|
|
|
type CategoryModel interface {
|
|
FindAll(ctx context.Context) ([]*Category, error)
|
|
}
|
|
type defaultCategoryModel struct {
|
|
db *sqlx.DB
|
|
cache *cache.Cache
|
|
}
|
|
|
|
func NewCategoryModel(db *sqlx.DB, cache *cache.Cache) CategoryModel {
|
|
return &defaultCategoryModel{db: db, cache: cache}
|
|
}
|
|
|
|
const categoryListCacheTTL = 5 * time.Minute
|
|
|
|
func (m *defaultCategoryModel) FindAll(ctx context.Context) ([]*Category, error) {
|
|
cached, hit, err := m.cache.Get[[]*Category](ctx, constant.CategoryListCacheKey)
|
|
if err != nil {
|
|
util.Logger(ctx).Error("读取分类列表缓存失败", "key", constant.CategoryListCacheKey, "error", err)
|
|
}
|
|
if hit {
|
|
return cached, nil
|
|
}
|
|
|
|
query, args, err := sq.
|
|
Select("id", "slug", "title", "sort").
|
|
From("public.categories").
|
|
Where(sq.GtOrEq{"sort": 1}).
|
|
OrderBy("sort ASC", "id ASC").
|
|
Limit(10).
|
|
PlaceholderFormat(sq.Dollar).
|
|
ToSql()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("squirrel 出错: %w", err)
|
|
}
|
|
rows := make([]*Category, 0)
|
|
err = m.db.SelectContext(ctx, &rows, query, args...)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("数据库查询出错: %w", err)
|
|
}
|
|
if err := m.cache.Set(ctx, cache.Entry[[]*Category]{
|
|
Key: constant.CategoryListCacheKey,
|
|
Value: rows,
|
|
TTL: categoryListCacheTTL,
|
|
}); err != nil {
|
|
util.Logger(ctx).Error("写入分类列表缓存失败", "key", constant.CategoryListCacheKey, "error", err)
|
|
}
|
|
return rows, nil
|
|
}
|