octoleo/restic
2
2
Fork 0
mirror of https://github.com/octoleo/restic.git synced 2024-11-05 04:47:51 +00:00
Code Issues Releases Wiki Activity

Lock MasterIndex and InFlight store together

Browse Source 
fixes: #358
This commit is contained in:
Philipp Serr 2015-12-28 18:18:25 +01:00
parent e7bf936d2b
commit 0fde09a866
1 changed files with 33 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
repository/master_index.go
View File

@ -157,27 +157,47 @@ func (mi *MasterIndex) Current() *Index {
// AddInFlight add the given ID to the list of in-flight IDs. An error is // AddInFlight add the given ID to the list of in-flight IDs. An error is
// returned when the ID is already in the list. // returned when the ID is already in the list.
func (mi *MasterIndex) AddInFlight(id backend.ID) error { func (mi *MasterIndex) AddInFlight(id backend.ID) error {
mi.inFlight.Lock() // The index + inFlight store must be searched for a matching id in one
defer mi.inFlight.Unlock() // atomic operation. This requires locking the inFlight store and the
// index together!
mi.inFlight.Lock()
defer mi.inFlight.Unlock()
debug.Log("MasterIndex.AddInFlight", "adding %v", id) // Note: mi.Has read locks the index again.
if mi.inFlight.Has(id) { mi.idxMutex.RLock()
return fmt.Errorf("%v is already in flight", id) defer mi.idxMutex.RUnlock()
}
mi.inFlight.Insert(id) debug.Log("MasterIndex.AddInFlight", "adding %v", id)
return nil if mi.inFlight.Has(id) {
return fmt.Errorf("%v is already in flight", id)
}
if mi.Has(id) {
return fmt.Errorf("%v is already indexed (fully processed)", id)
}
mi.inFlight.Insert(id)
return nil
} }
// IsInFlight returns true iff the id is contained in the list of in-flight IDs. // IsInFlight returns true iff the id is contained in the list of in-flight IDs.
func (mi *MasterIndex) IsInFlight(id backend.ID) bool { func (mi *MasterIndex) IsInFlight(id backend.ID) bool {
mi.inFlight.RLock() // The index + inFlight store must be searched for a matching id in one
defer mi.inFlight.RUnlock() // atomic operation. This requires locking the inFlight store and the
// index together!
mi.inFlight.RLock()
defer mi.inFlight.RUnlock()
inFlight := mi.inFlight.Has(id) // Note: mi.Has read locks the index again.
debug.Log("MasterIndex.IsInFlight", "testing whether %v is in flight: %v", id.Str(), inFlight) mi.idxMutex.RLock()
defer mi.idxMutex.RUnlock()
return inFlight inFlight := mi.inFlight.Has(id)
debug.Log("MasterIndex.IsInFlight", "testing whether %v is in flight: %v", id.Str(), inFlight)
indexed := mi.Has(id)
debug.Log("MasterIndex.IsInFlight", "testing whether %v is indexed (fully processed): %v", id.Str(), indexed)
return inFlight
} }
// RemoveFromInFlight deletes the given ID from the liste of in-flight IDs. // RemoveFromInFlight deletes the given ID from the liste of in-flight IDs.