From 706ec2997d31edfea0c075715b9df44db842922e Mon Sep 17 00:00:00 2001
From: Cassandra Heart <cassandra@quilibrium.com>
Date: Sat, 16 Nov 2024 18:29:41 -0600
Subject: [PATCH] adjust runonce

---
 node/consensus/data/main_data_loop.go | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/node/consensus/data/main_data_loop.go b/node/consensus/data/main_data_loop.go
index 14d7291..96be2f5 100644
--- a/node/consensus/data/main_data_loop.go
+++ b/node/consensus/data/main_data_loop.go
@@ -71,9 +71,20 @@ func (e *DataClockConsensusEngine) runLoop() {
 				panic(err)
 			}
 
+			if runOnce {
+				if e.GetFrameProverTries()[0].Contains(e.provingKeyAddress) {
+					dataFrame, err := e.dataTimeReel.Head()
+					if err != nil {
+						panic(err)
+					}
+
+					latestFrame = e.processFrame(latestFrame, dataFrame)
+				}
+				runOnce = false
+			}
+
 			select {
 			case dataFrame := <-dataFrameCh:
-				runOnce = false
 				if e.GetFrameProverTries()[0].Contains(e.provingKeyAddress) {
 					if err = e.publishProof(dataFrame); err != nil {
 						e.logger.Error("could not publish", zap.Error(err))
@@ -86,14 +97,10 @@ func (e *DataClockConsensusEngine) runLoop() {
 				}
 				latestFrame = e.processFrame(latestFrame, dataFrame)
 			case <-time.After(20 * time.Second):
-				if e.GetFrameProverTries()[0].Contains(e.provingKeyAddress) && !runOnce {
+				if e.GetFrameProverTries()[0].Contains(e.provingKeyAddress) {
 					continue
 				}
 
-				if runOnce {
-					runOnce = false
-				}
-
 				dataFrame, err := e.dataTimeReel.Head()
 				if err != nil {
 					panic(err)