mirrors/owncast
1
0
Fork 0
mirror of https://github.com/owncast/owncast.git synced 2024-12-18 07:12:33 +03:00
Code Issues Projects Releases Packages Wiki Activity

feat(api): cache thumb and preview in memory for fast retrieval. Help address #3462

Browse source
This commit is contained in:
Gabe Kangas 2024-05-01 18:55:59 -07:00
parent aa0b74e236
commit 1d49801a27
No known key found for this signature in database
GPG key ID: 4345B2060657F330
1 changed files with 24 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
controllers/images.go
View file

@ -3,7 +3,9 @@ package controllers
import (
"net/http"
"path/filepath"
"time"
"github.com/jellydator/ttlcache/v3"
"github.com/owncast/owncast/config"
"github.com/owncast/owncast/utils"
)
@ -13,16 +15,28 @@ const (
contentTypeGIF = "image/gif"
)
var previewThumbCache = ttlcache.New(
ttlcache.WithTTL[string, []byte](15),
ttlcache.WithCapacity[string, []byte](1),
ttlcache.WithDisableTouchOnHit[string, []byte](),
)
// GetThumbnail will return the thumbnail image as a response.
func GetThumbnail(w http.ResponseWriter, r *http.Request) {
imageFilename := "thumbnail.jpg"
imagePath := filepath.Join(config.TempDir, imageFilename)
httpCacheTime := utils.GetCacheDurationSecondsForPath(imagePath)
inMemoryCacheTime := time.Duration(15) * time.Second
var imageBytes []byte
var err error
if utils.DoesFileExists(imagePath) {
if previewThumbCache.Get(imagePath) != nil {
ci := previewThumbCache.Get(imagePath)
imageBytes = ci.Value()
} else if utils.DoesFileExists(imagePath) {
imageBytes, err = getImage(imagePath)
previewThumbCache.Set(imagePath, imageBytes, inMemoryCacheTime)
} else {
GetLogo(w, r)
return
@ -33,20 +47,25 @@ func GetThumbnail(w http.ResponseWriter, r *http.Request) {
return
}
cacheTime := utils.GetCacheDurationSecondsForPath(imagePath)
writeBytesAsImage(imageBytes, contentTypeJPEG, w, cacheTime)
writeBytesAsImage(imageBytes, contentTypeJPEG, w, httpCacheTime)
}
// GetPreview will return the preview gif as a response.
func GetPreview(w http.ResponseWriter, r *http.Request) {
imageFilename := "preview.gif"
imagePath := filepath.Join(config.TempDir, imageFilename)
httpCacheTime := utils.GetCacheDurationSecondsForPath(imagePath)
inMemoryCacheTime := time.Duration(15) * time.Second
var imageBytes []byte
var err error
if utils.DoesFileExists(imagePath) {
if previewThumbCache.Get(imagePath) != nil {
ci := previewThumbCache.Get(imagePath)
imageBytes = ci.Value()
} else if utils.DoesFileExists(imagePath) {
imageBytes, err = getImage(imagePath)
previewThumbCache.Set(imagePath, imageBytes, inMemoryCacheTime)
} else {
GetLogo(w, r)
return
@ -57,6 +76,5 @@ func GetPreview(w http.ResponseWriter, r *http.Request) {
return
}
cacheTime := utils.GetCacheDurationSecondsForPath(imagePath)
writeBytesAsImage(imageBytes, contentTypeGIF, w, cacheTime)
writeBytesAsImage(imageBytes, contentTypeGIF, w, httpCacheTime)
}