sovonegr/TorrServerJellyfin
1
0
Fork 0
mirror of https://github.com/Ernous/TorrServerJellyfin.git synced 2025-12-19 13:36:09 +05:00
Code Issues Packages Projects Releases Wiki Activity

fix race in DBReadCache

Browse Source
This commit is contained in:
nikk gitanes
2024-04-10 00:04:48 +03:00
parent 78db941fd7
commit 9f93de6305
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
server/settings/dbreadcache.go
View File

@@ -1,11 +1,15 @@
package settings
import "server/log"
import (
"server/log"
"sync"
)
type DBReadCache struct {
db TorrServerDB
listCache map[string][]string
dataCache map[[2]string][]byte
db TorrServerDB
listCache map[string][]string
dataCache map[[2]string][]byte
dataCacheMutex sync.RWMutex
}
func NewDBReadCache(db TorrServerDB) TorrServerDB {
@@ -26,11 +30,16 @@ func (v *DBReadCache) CloseDB() {
func (v *DBReadCache) Get(xPath, name string) []byte {
cacheKey := v.makeDataCacheKey(xPath, name)
v.dataCacheMutex.RLock()
defer v.dataCacheMutex.RUnlock()
if data, ok := v.dataCache[cacheKey]; ok {
return data
}
v.dataCacheMutex.RUnlock()
data := v.db.Get(xPath, name)
v.dataCacheMutex.Lock()
v.dataCache[cacheKey] = data
v.dataCacheMutex.Unlock()
return data
}
@@ -40,7 +49,9 @@ func (v *DBReadCache) Set(xPath, name string, value []byte) {
return
}
cacheKey := v.makeDataCacheKey(xPath, name)
v.dataCacheMutex.Lock()
v.dataCache[cacheKey] = value
v.dataCacheMutex.Unlock()
delete(v.listCache, xPath)
v.db.Set(xPath, name, value)
}