fix: stop ec2 if no proof is generating

kroma-network · Aug 28, 2023 · cbef5d0 · cbef5d0
1 parent 0da6feb
commit cbef5d0
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/internal/proof/service.go b/internal/proof/service.go
@@ -43,7 +43,14 @@ func (s *Service) Prove(traceString string, proofType Type) (*ProveResponse, err
 			s.inProgressProof[id] = wg
 			go func(id, blockNumber string) {
 				defer wg.Done()
-				defer delete(s.inProgressProof, id)
+				defer func() {
+					s.mu.Lock()
+					delete(s.inProgressProof, id)
+					s.mu.Unlock()
+					if len(s.inProgressProof) == 0 {
+						s.ec2.StopIfRunning()
+					}
+				}()
 				log.Println("prove start.", "blockNumber:", blockNumber, "id:", id)
 				res, err := c.Prove(traceString, proofType)
 				log.Println("prove complete.", "blockNumber:", blockNumber, "id:", id, "err:", err)