fix reservations

2021-03-31 14:30:24 +08:00
parent f4a2c8fa49
commit 3895b2ccca
1 changed files with 6 additions and 0 deletions
@@ -392,8 +392,10 @@ func (st *Local) Reserve(ctx context.Context, sid storage.SectorRef, ft storifac
 		}

 		p.reserved += overhead
+		p.reservations[sid.ID] |= fileType

 		prevDone := done
+		saveFileType := fileType
 		done = func() {
 			prevDone()

@@ -401,6 +403,10 @@ func (st *Local) Reserve(ctx context.Context, sid storage.SectorRef, ft storifac
 			defer st.localLk.Unlock()

 			p.reserved -= overhead
+			p.reservations[sid.ID] ^= saveFileType
+			if p.reservations[sid.ID] == storiface.FTNone {
+				delete(p.reservations, sid.ID)
+			}
 		}
 	}