lsm/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2025-09-23 07:25:01 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add lock for vidCache

Browse Source
This commit is contained in:
Chris Lu
2020-10-21 19:28:59 -07:00
parent 81cf8d04df
commit bd103c143a
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
weed/filer/reader_at.go
View File

@@ -38,9 +38,12 @@ type LookupFileIdFunctionType func(fileId string) (targetUrls []string, err erro
func LookupFn(filerClient filer_pb.FilerClient) LookupFileIdFunctionType { func LookupFn(filerClient filer_pb.FilerClient) LookupFileIdFunctionType {
vidCache := make(map[string]*filer_pb.Locations) vidCache := make(map[string]*filer_pb.Locations)
var vicCacheLock sync.RWMutex
return func(fileId string) (targetUrls []string, err error) { return func(fileId string) (targetUrls []string, err error) {
vid := VolumeId(fileId) vid := VolumeId(fileId)
vicCacheLock.RLock()
locations, found := vidCache[vid] locations, found := vidCache[vid]
vicCacheLock.RUnlock()
waitTime := time.Second waitTime := time.Second
for !found && waitTime < ReadWaitTime { for !found && waitTime < ReadWaitTime {
@@ -58,7 +61,9 @@ func LookupFn(filerClient filer_pb.FilerClient) LookupFileIdFunctionType {
glog.V(0).Infof("failed to locate %s", fileId) glog.V(0).Infof("failed to locate %s", fileId)
return fmt.Errorf("failed to locate %s", fileId) return fmt.Errorf("failed to locate %s", fileId)
} }
vicCacheLock.Lock()
vidCache[vid] = locations vidCache[vid] = locations
vicCacheLock.Unlock()
return nil return nil
}) })