76 lines
1.7 KiB
Go
76 lines
1.7 KiB
Go
package scanner
|
|
|
|
import (
|
|
"path"
|
|
"sync"
|
|
)
|
|
|
|
type AlbumScannerCache struct {
|
|
path_contains_photos map[string]bool
|
|
photo_types map[string]ImageType
|
|
mutex sync.Mutex
|
|
}
|
|
|
|
func MakeAlbumCache() *AlbumScannerCache {
|
|
return &AlbumScannerCache{
|
|
path_contains_photos: make(map[string]bool),
|
|
photo_types: make(map[string]ImageType),
|
|
}
|
|
}
|
|
|
|
// Insert single album directory in cache
|
|
func (c *AlbumScannerCache) InsertAlbumPath(path string, contains_photo bool) {
|
|
c.mutex.Lock()
|
|
defer c.mutex.Unlock()
|
|
c.path_contains_photos[path] = contains_photo
|
|
}
|
|
|
|
// Insert album path and all parent directories up to the given root directory in cache
|
|
func (c *AlbumScannerCache) InsertAlbumPaths(end_path string, root string, contains_photo bool) {
|
|
curr_path := path.Clean(end_path)
|
|
root_path := path.Clean(root)
|
|
|
|
c.mutex.Lock()
|
|
defer c.mutex.Unlock()
|
|
|
|
for curr_path != root_path || curr_path == "." {
|
|
|
|
c.path_contains_photos[curr_path] = contains_photo
|
|
|
|
curr_path = path.Dir(curr_path)
|
|
}
|
|
}
|
|
|
|
func (c *AlbumScannerCache) AlbumContainsPhotos(path string) *bool {
|
|
c.mutex.Lock()
|
|
defer c.mutex.Unlock()
|
|
|
|
contains_photo, found := c.path_contains_photos[path]
|
|
if found {
|
|
// log.Printf("Album cache hit: %s\n", path)
|
|
return &contains_photo
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *AlbumScannerCache) InsertPhotoType(path string, content_type ImageType) {
|
|
c.mutex.Lock()
|
|
defer c.mutex.Unlock()
|
|
|
|
(c.photo_types)[path] = content_type
|
|
}
|
|
|
|
func (c *AlbumScannerCache) GetPhotoType(path string) *ImageType {
|
|
c.mutex.Lock()
|
|
defer c.mutex.Unlock()
|
|
|
|
result, found := c.photo_types[path]
|
|
if found {
|
|
// log.Printf("Image cache hit: %s\n", path)
|
|
return &result
|
|
}
|
|
|
|
return nil
|
|
}
|