mirror of
https://github.com/photoprism/photoprism.git
synced 2026-01-23 02:24:24 +00:00
47 lines
1 KiB
Go
47 lines
1 KiB
Go
package entity
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
gc "github.com/patrickmn/go-cache"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/photoprism/photoprism/pkg/clean"
|
|
"github.com/photoprism/photoprism/pkg/rnd"
|
|
)
|
|
|
|
var albumCache = gc.New(15*time.Minute, 15*time.Minute)
|
|
|
|
// FlushAlbumCache clears any cached album entries.
|
|
func FlushAlbumCache() {
|
|
albumCache.Flush()
|
|
}
|
|
|
|
// CachedAlbumByUID returns the album with the specified UID from cache or database.
|
|
func CachedAlbumByUID(uid string) (m Album, err error) {
|
|
// Valid album UID?
|
|
if uid == "" || rnd.InvalidUID(uid, AlbumUID) {
|
|
return m, fmt.Errorf("invalid album uid %s", clean.LogQuote(uid))
|
|
}
|
|
|
|
// Cached?
|
|
if cacheData, hit := albumCache.Get(uid); hit {
|
|
log.Tracef("album: cache hit for %s", uid)
|
|
return cacheData.(Album), nil
|
|
}
|
|
|
|
// Find in database.
|
|
m = Album{}
|
|
|
|
r := Db().First(&m, "album_uid = ?", uid)
|
|
if errors.Is(r.Error, gorm.ErrRecordNotFound) {
|
|
return m, fmt.Errorf("album not found")
|
|
} else if r.Error != nil {
|
|
return m, r.Error
|
|
} else {
|
|
albumCache.SetDefault(m.AlbumUID, m)
|
|
return m, nil
|
|
}
|
|
}
|