mirrors/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty synced 2026-06-08 14:18:26 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix pruning on DiskCache object creation

Browse Source
This commit is contained in:
Kovid Goyal
2025-09-30 13:18:00 +05:30
parent 8fdbf8d7da
commit 712faa7aa5
2 changed files with 26 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
tools/disk_cache/implementation.go
View File

@@ -22,17 +22,26 @@ func new_disk_cache(path string, max_size int64) (dc *DiskCache, err error) {
if err = os.MkdirAll(path, 0o700); err != nil { if err = os.MkdirAll(path, 0o700); err != nil {
return return
} }
dc = &DiskCache{Path: path, MaxSize: max_size} ans := &DiskCache{Path: path, MaxSize: max_size}
dc.lock() ans.lock()
defer dc.unlock() defer ans.unlock()
if err = dc.prune(); err != nil { if err = ans.ensure_entries(); err != nil {
return return
} }
if dc.get_dir, err = os.MkdirTemp(dc.Path, "getdir-*"); err != nil { if pruned, err := ans.prune(); err != nil {
return nil, err
} else if pruned {
if err = ans.write_entries(); err != nil {
return nil, err
}
}
if ans.get_dir, err = os.MkdirTemp(ans.Path, "getdir-*"); err != nil {
return return
} }
err = utils.AtExitRmtree(dc.get_dir) if err = utils.AtExitRmtree(ans.get_dir); err != nil {
return return
}
return ans, nil
} }
func key_for_path(path string) (key string, err error) { func key_for_path(path string) (key string, err error) {
@@ -211,9 +220,9 @@ func (dc *DiskCache) remove(key string) (err error) {
return return
} }
func (dc *DiskCache) prune() error { func (dc *DiskCache) prune() (bool, error) {
if dc.MaxSize < 1 || dc.entries.TotalSize <= dc.MaxSize { if dc.MaxSize < 1 || dc.entries.TotalSize <= dc.MaxSize {
return nil return false, nil
} }
for dc.entries.TotalSize > dc.MaxSize && len(dc.entries.SortedEntries) > 0 { for dc.entries.TotalSize > dc.MaxSize && len(dc.entries.SortedEntries) > 0 {
base := dc.folder_for_key(dc.entries.SortedEntries[0].Key) base := dc.folder_for_key(dc.entries.SortedEntries[0].Key)
@@ -223,10 +232,10 @@ func (dc *DiskCache) prune() error {
dc.entries.TotalSize = max(0, dc.entries.TotalSize-t.Size) dc.entries.TotalSize = max(0, dc.entries.TotalSize-t.Size)
dc.entries.SortedEntries = dc.entries.SortedEntries[1:] dc.entries.SortedEntries = dc.entries.SortedEntries[1:]
} else { } else {
return err return false, err
} }
} }
return nil return true, nil
} }
func (dc *DiskCache) update_timestamp(key string) { func (dc *DiskCache) update_timestamp(key string) {

6
tools/disk_cache/implementation_test.go
View File

@@ -69,4 +69,10 @@ func TestDiskCache(t *testing.T) {
ak("k1", "k2") ak("k1", "k2")
dc.Add("k3", map[string][]byte{"1": []byte(strings.Repeat("a", int(dc.MaxSize)-10))}) dc.Add("k3", map[string][]byte{"1": []byte(strings.Repeat("a", int(dc.MaxSize)-10))})
ak("k3", "k2") ak("k3", "k2")
// check that creating a new disk cache prunes
_, err = NewDiskCache(tdir, dc.MaxSize-8)
if err != nil {
t.Fatal(err)
}
ak("k3")
} }