sealing: Fix sector state accounting with FinalizeEarly

filecoin-project · Sep 2, 2021 · 7ef1b62 · 7ef1b62
1 parent 591f8f5
commit 7ef1b62
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/extern/storage-sealing/sector_state.go b/extern/storage-sealing/sector_state.go
@@ -106,11 +106,19 @@ const (
 	Removed      SectorState = "Removed"
 )
 
-func toStatState(st SectorState) statSectorState {
+func toStatState(st SectorState, finEarly bool) statSectorState {
 	switch st {
 	case UndefinedSectorState, Empty, WaitDeals, AddPiece:
 		return sstStaging
-	case Packing, GetTicket, PreCommit1, PreCommit2, PreCommitting, PreCommitWait, SubmitPreCommitBatch, PreCommitBatchWait, WaitSeed, Committing, CommitFinalize, SubmitCommit, CommitWait, SubmitCommitAggregate, CommitAggregateWait, FinalizeSector:
+	case Packing, GetTicket, PreCommit1, PreCommit2, PreCommitting, PreCommitWait, SubmitPreCommitBatch, PreCommitBatchWait, WaitSeed, Committing, CommitFinalize, FinalizeSector:
+		return sstSealing
+	case SubmitCommit, CommitWait, SubmitCommitAggregate, CommitAggregateWait:
+		if finEarly {
+			// we use statSectorState for throttling storage use. With FinalizeEarly
+			// we can consider sectors in states after CommitFinalize as finalized, so
+			// that more sectors can enter the sealing pipeline (and later be aggregated together)
+			return sstProving
+		}
 		return sstSealing
 	case Proving, Removed, Removing, Terminating, TerminateWait, TerminateFinality, TerminateFailed:
 		return sstProving

diff --git a/extern/storage-sealing/stats.go b/extern/storage-sealing/stats.go
@@ -37,7 +37,7 @@ func (ss *SectorStats) updateSector(cfg sealiface.Config, id abi.SectorID, st Se
 		ss.totals[oldst]--
 	}
 
-	sst := toStatState(st)
+	sst := toStatState(st, cfg.FinalizeEarly)
 	ss.bySector[id] = sst
 	ss.totals[sst]++