125 lines
3.9 KiB
Go
125 lines
3.9 KiB
Go
package cache
|
|
|
|
import (
|
|
"context"
|
|
"github/fthvgb1/wp-go/cache"
|
|
dao "github/fthvgb1/wp-go/internal/dao"
|
|
"github/fthvgb1/wp-go/internal/pkg/config"
|
|
"github/fthvgb1/wp-go/internal/pkg/logs"
|
|
models2 "github/fthvgb1/wp-go/internal/pkg/models"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
var postContextCache *cache.MapCache[uint64, dao.PostContext]
|
|
var archivesCaches *Arch
|
|
var categoryCaches *cache.SliceCache[models2.TermsMy]
|
|
var recentPostsCaches *cache.SliceCache[models2.Posts]
|
|
var recentCommentsCaches *cache.SliceCache[models2.Comments]
|
|
var postCommentCaches *cache.MapCache[uint64, []uint64]
|
|
var postsCache *cache.MapCache[uint64, models2.Posts]
|
|
|
|
var postMetaCache *cache.MapCache[uint64, map[string]any]
|
|
|
|
var monthPostsCache *cache.MapCache[string, []uint64]
|
|
var postListIdsCache *cache.MapCache[string, dao.PostIds]
|
|
var searchPostIdsCache *cache.MapCache[string, dao.PostIds]
|
|
var maxPostIdCache *cache.SliceCache[uint64]
|
|
|
|
var usersCache *cache.MapCache[uint64, models2.Users]
|
|
var usersNameCache *cache.MapCache[string, models2.Users]
|
|
var commentsCache *cache.MapCache[uint64, models2.Comments]
|
|
|
|
func InitActionsCommonCache() {
|
|
c := config.Conf.Load()
|
|
archivesCaches = &Arch{
|
|
mutex: &sync.Mutex{},
|
|
setCacheFunc: dao.Archives,
|
|
}
|
|
|
|
searchPostIdsCache = cache.NewMapCacheByFn[string, dao.PostIds](dao.SearchPostIds, c.SearchPostCacheTime)
|
|
|
|
postListIdsCache = cache.NewMapCacheByFn[string, dao.PostIds](dao.SearchPostIds, c.PostListCacheTime)
|
|
|
|
monthPostsCache = cache.NewMapCacheByFn[string, []uint64](dao.MonthPost, c.MonthPostCacheTime)
|
|
|
|
postContextCache = cache.NewMapCacheByFn[uint64, dao.PostContext](dao.GetPostContext, c.ContextPostCacheTime)
|
|
|
|
postsCache = cache.NewMapCacheByBatchFn[uint64, models2.Posts](dao.GetPostsByIds, c.PostDataCacheTime)
|
|
|
|
postMetaCache = cache.NewMapCacheByBatchFn[uint64, map[string]any](dao.GetPostMetaByPostIds, c.PostDataCacheTime)
|
|
|
|
categoryCaches = cache.NewSliceCache[models2.TermsMy](dao.Categories, c.CategoryCacheTime)
|
|
|
|
recentPostsCaches = cache.NewSliceCache[models2.Posts](dao.RecentPosts, c.RecentPostCacheTime)
|
|
|
|
recentCommentsCaches = cache.NewSliceCache[models2.Comments](dao.RecentComments, c.RecentCommentsCacheTime)
|
|
|
|
postCommentCaches = cache.NewMapCacheByFn[uint64, []uint64](dao.PostComments, c.PostCommentsCacheTime)
|
|
|
|
maxPostIdCache = cache.NewSliceCache[uint64](dao.GetMaxPostId, c.MaxPostIdCacheTime)
|
|
|
|
usersCache = cache.NewMapCacheByFn[uint64, models2.Users](dao.GetUserById, c.UserInfoCacheTime)
|
|
|
|
usersNameCache = cache.NewMapCacheByFn[string, models2.Users](dao.GetUserByName, c.UserInfoCacheTime)
|
|
|
|
commentsCache = cache.NewMapCacheByBatchFn[uint64, models2.Comments](dao.GetCommentByIds, c.CommentsCacheTime)
|
|
}
|
|
|
|
func ClearCache() {
|
|
searchPostIdsCache.ClearExpired()
|
|
postsCache.ClearExpired()
|
|
postMetaCache.ClearExpired()
|
|
postListIdsCache.ClearExpired()
|
|
monthPostsCache.ClearExpired()
|
|
postContextCache.ClearExpired()
|
|
usersCache.ClearExpired()
|
|
commentsCache.ClearExpired()
|
|
usersNameCache.ClearExpired()
|
|
}
|
|
func FlushCache() {
|
|
searchPostIdsCache.Flush()
|
|
postsCache.Flush()
|
|
postMetaCache.Flush()
|
|
postListIdsCache.Flush()
|
|
monthPostsCache.Flush()
|
|
postContextCache.Flush()
|
|
usersCache.Flush()
|
|
commentsCache.Flush()
|
|
usersCache.Flush()
|
|
}
|
|
|
|
func Archives(ctx context.Context) (r []models2.PostArchive) {
|
|
return archivesCaches.getArchiveCache(ctx)
|
|
}
|
|
|
|
type Arch struct {
|
|
data []models2.PostArchive
|
|
mutex *sync.Mutex
|
|
setCacheFunc func(context.Context) ([]models2.PostArchive, error)
|
|
month time.Month
|
|
}
|
|
|
|
func (c *Arch) getArchiveCache(ctx context.Context) []models2.PostArchive {
|
|
l := len(c.data)
|
|
m := time.Now().Month()
|
|
if l > 0 && c.month != m || l < 1 {
|
|
r, err := c.setCacheFunc(ctx)
|
|
if err != nil {
|
|
logs.ErrPrintln(err, "set cache err[%s]")
|
|
return nil
|
|
}
|
|
c.mutex.Lock()
|
|
defer c.mutex.Unlock()
|
|
c.month = m
|
|
c.data = r
|
|
}
|
|
return c.data
|
|
}
|
|
|
|
func Categories(ctx context.Context) []models2.TermsMy {
|
|
r, err := categoryCaches.GetCache(ctx, time.Second, ctx)
|
|
logs.ErrPrintln(err, "get category ")
|
|
return r
|
|
}
|