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

…eration is canceled (#402)
taikoxyz · Sep 22, 2023 · 1eab54d · 1eab54d
1 parent ce8bd1d
commit 1eab54d
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()
+		}
 	}
 }