mirrors/navidrome
1
0
Fork 0
mirror of https://github.com/navidrome/navidrome.git synced 2025-04-04 21:17:37 +03:00
Code Issues Projects Releases Packages Wiki Activity

Make caches singletons

Browse source
This commit is contained in:
Deluan 2020-10-25 22:49:37 -04:00
parent 1e56f4da76
commit 81d7556cdf
4 changed files with 46 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

2
cmd/wire_gen.go
View file

@ -29,7 +29,7 @@ func CreateScanner(musicFolder string) scanner.Scanner {
dataStore := persistence.New() dataStore := persistence.New()
artworkCache := core.NewImageCache() artworkCache := core.NewImageCache()
artwork := core.NewArtwork(dataStore, artworkCache) artwork := core.NewArtwork(dataStore, artworkCache)
cacheWarmer := core.NewCacheWarmer(artworkCache, artwork) cacheWarmer := core.NewCacheWarmer(artwork)
scannerScanner := scanner.New(dataStore, cacheWarmer) scannerScanner := scanner.New(dataStore, cacheWarmer)
return scannerScanner return scannerScanner
} }

11
core/artwork.go
View file

@ -12,6 +12,7 @@ import (
"io" "io"
"os" "os"
"strings" "strings"
"sync"
"time" "time"
"github.com/deluan/navidrome/core/cache" "github.com/deluan/navidrome/core/cache"
@ -205,8 +206,14 @@ func readFromFile(path string) ([]byte, error) {
return buf.Bytes(), nil return buf.Bytes(), nil
} }
var (
onceImageCache sync.Once
imageCache ArtworkCache
)
func NewImageCache() ArtworkCache { func NewImageCache() ArtworkCache {
return cache.NewFileCache("Image", conf.Server.ImageCacheSize, consts.ImageCacheDir, consts.DefaultImageCacheMaxItems, onceImageCache.Do(func() {
imageCache = cache.NewFileCache("Image", conf.Server.ImageCacheSize, consts.ImageCacheDir, consts.DefaultImageCacheMaxItems,
func(ctx context.Context, arg cache.Item) (io.Reader, error) { func(ctx context.Context, arg cache.Item) (io.Reader, error) {
info := arg.(*imageInfo) info := arg.(*imageInfo)
reader, err := info.a.getArtwork(ctx, info.id, info.path, info.size) reader, err := info.a.getArtwork(ctx, info.id, info.path, info.size)
@ -216,4 +223,6 @@ func NewImageCache() ArtworkCache {
} }
return reader, nil return reader, nil
}) })
})
return imageCache
} }

4
core/cache_warmer.go
View file

@ -14,10 +14,9 @@ type CacheWarmer interface {
Flush(ctx context.Context) Flush(ctx context.Context)
} }
func NewCacheWarmer(cache ArtworkCache, artwork Artwork) CacheWarmer { func NewCacheWarmer(artwork Artwork) CacheWarmer {
w := &warmer{ w := &warmer{
artwork: artwork, artwork: artwork,
cache: cache,
albums: map[string]struct{}{}, albums: map[string]struct{}{},
} }
p, err := pool.NewPool("artwork", 3, &artworkItem{}, w.execute) p, err := pool.NewPool("artwork", 3, &artworkItem{}, w.execute)
@ -33,7 +32,6 @@ func NewCacheWarmer(cache ArtworkCache, artwork Artwork) CacheWarmer {
type warmer struct { type warmer struct {
pool *pool.Pool pool *pool.Pool
artwork Artwork artwork Artwork
cache ArtworkCache
albums map[string]struct{} albums map[string]struct{}
} }

11
core/media_streamer.go
View file

@ -6,6 +6,7 @@ import (
"io" "io"
"mime" "mime"
"os" "os"
"sync"
"time" "time"
"github.com/deluan/navidrome/conf" "github.com/deluan/navidrome/conf"
@ -167,8 +168,14 @@ func selectTranscodingOptions(ctx context.Context, ds model.DataStore, mf *model
return return
} }
var (
onceTranscodingCache sync.Once
transcodingCache TranscodingCache
)
func NewTranscodingCache() TranscodingCache { func NewTranscodingCache() TranscodingCache {
return cache.NewFileCache("Transcoding", conf.Server.TranscodingCacheSize, onceTranscodingCache.Do(func() {
transcodingCache = cache.NewFileCache("Transcoding", conf.Server.TranscodingCacheSize,
consts.TranscodingCacheDir, consts.DefaultTranscodingCacheMaxItems, consts.TranscodingCacheDir, consts.DefaultTranscodingCacheMaxItems,
func(ctx context.Context, arg cache.Item) (io.Reader, error) { func(ctx context.Context, arg cache.Item) (io.Reader, error) {
job := arg.(*streamJob) job := arg.(*streamJob)
@ -184,4 +191,6 @@ func NewTranscodingCache() TranscodingCache {
} }
return out, nil return out, nil
}) })
})
return transcodingCache
} }