feat(prover): release capacity when the corresponding local proof gen…

…eration is canceled
taikoxyz · Sep 21, 2023 · aef127f · aef127f
1 parent ce8bd1d
commit aef127f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/prover/prover.go b/prover/prover.go
@@ -862,6 +862,9 @@ func (p *Prover) cancelProof(ctx context.Context, blockID uint64) {
 	if cancel, ok := p.currentBlocksBeingProven[blockID]; ok {
 		cancel()
 		delete(p.currentBlocksBeingProven, blockID)
+		if !p.cfg.OracleProver {
+			p.capacityManager.ReleaseOneCapacity()
+		}
 	}
 }