From ff8c6b2995b00113b56f1703b9bf80f6ad4e0991 Mon Sep 17 00:00:00 2001
From: jeff <113397187+cyberhorsey@users.noreply.github.com>
Date: Fri, 29 Sep 2023 17:21:54 -0700
Subject: [PATCH] chore(prover): Update capacity logging (#417)

---
 prover/capacity_manager/capacity_manager.go | 43 +++++++++++++++++----
 prover/prover.go                            |  2 +
 2 files changed, 38 insertions(+), 7 deletions(-)

diff --git a/prover/capacity_manager/capacity_manager.go b/prover/capacity_manager/capacity_manager.go
index 1f0d0074d..532a94a4b 100644
--- a/prover/capacity_manager/capacity_manager.go
+++ b/prover/capacity_manager/capacity_manager.go
@@ -31,7 +31,12 @@ func (m *CapacityManager) ReadCapacity() uint64 {
 	m.mutex.RLock()
 	defer m.mutex.RUnlock()
 
-	log.Info("Reading capacity", "capacity", len(m.capacity))
+	log.Info("Reading capacity",
+		"maxCapacity", m.maxCapacity,
+		"currentCapacity", m.maxCapacity-uint64(len(m.capacity)),
+		"currentUsage", len(m.capacity),
+		"currentTempCapacityUsage", len(m.tempCapacity),
+	)
 
 	return m.maxCapacity - uint64((len(m.capacity)))
 }
@@ -44,14 +49,21 @@ func (m *CapacityManager) ReleaseOneCapacity(blockID uint64) (uint64, bool) {
 	if _, ok := m.capacity[blockID]; !ok {
 		log.Info("Can not release capacity",
 			"blockID", blockID,
+			"maxCapacity", m.maxCapacity,
 			"currentCapacity", m.maxCapacity-uint64(len(m.capacity)),
-			"maxCapacity", m.maxCapacity)
+			"currentUsage", len(m.capacity),
+		)
 		return uint64(len(m.capacity)), false
 	}
 
 	delete(m.capacity, blockID)
 
-	log.Info("Released capacity", "blockID", blockID, "capacityAfterRelease", len(m.capacity))
+	log.Info("Released capacity",
+		"blockID", blockID,
+		"maxCapacity", m.maxCapacity,
+		"currentCapacityAfterRelease", m.maxCapacity-uint64(len(m.capacity)),
+		"currentUsageAfterRelease", len(m.capacity),
+	)
 
 	return m.maxCapacity - uint64(len(m.capacity)), true
 }
@@ -64,7 +76,10 @@ func (m *CapacityManager) TakeOneCapacity(blockID uint64) (uint64, bool) {
 	if len(m.capacity) == int(m.maxCapacity) {
 		log.Info("Could not take one capacity",
 			"blockID", blockID,
-			"currentCapacity", m.maxCapacity-uint64(len(m.capacity)))
+			"maxCapacity", m.maxCapacity,
+			"currentCapacity", m.maxCapacity-uint64(len(m.capacity)),
+			"currentUsage", len(m.capacity),
+		)
 		return 0, false
 	}
 
@@ -72,7 +87,10 @@ func (m *CapacityManager) TakeOneCapacity(blockID uint64) (uint64, bool) {
 
 	log.Info("Took one capacity",
 		"blockID", blockID,
-		"capacityAfterTaking", m.maxCapacity-uint64(len(m.capacity)))
+		"maxCapacity", m.maxCapacity,
+		"currentCapacityAfterTaking", m.maxCapacity-uint64(len(m.capacity)),
+		"currentUsageAfterTaking", len(m.capacity),
+	)
 
 	return m.maxCapacity - uint64((len(m.capacity))), true
 }
@@ -87,8 +105,12 @@ func (m *CapacityManager) TakeOneTempCapacity() (uint64, bool) {
 
 	if len(m.capacity)+len(m.tempCapacity) >= int(m.maxCapacity) {
 		log.Info("Could not take one temp capacity",
-			"capacity", m.maxCapacity-uint64(len(m.capacity)),
-			"tempCapacity", len(m.tempCapacity))
+			"maxCapacity", m.maxCapacity,
+			"currentCapacityAfterTaking", m.maxCapacity-uint64(len(m.capacity)),
+			"currentUsageAfterTaking", len(m.capacity),
+			"tempCapacity", m.maxCapacity-uint64(len((m.tempCapacity))),
+			"tempCapacityUsage", len(m.tempCapacity),
+		)
 		return 0, false
 	}
 
@@ -101,6 +123,13 @@ func (m *CapacityManager) clearExpiredTempCapacities() {
 	for i, c := range m.tempCapacity {
 		if time.Now().UTC().Sub(c) > m.tempCapacityExpiresAt {
 			m.tempCapacity = append(m.tempCapacity[:i], m.tempCapacity[i+1:]...)
+			log.Info("Cleared one temp capacity",
+				"maxCapacity", m.maxCapacity,
+				"currentCapacityAfterClearing", m.maxCapacity-uint64(len(m.capacity)),
+				"currentUsageAfterClearing", len(m.capacity),
+				"tempCapacity", m.maxCapacity-uint64(len((m.tempCapacity))),
+				"tempCapacityUsage", len(m.tempCapacity),
+			)
 		}
 	}
 }
diff --git a/prover/prover.go b/prover/prover.go
index 1ed725a6a..86d1a2add 100644
--- a/prover/prover.go
+++ b/prover/prover.go
@@ -361,6 +361,8 @@ func (p *Prover) onBlockProposed(
 ) error {
 	// If there is newly generated proofs, we need to submit them as soon as possible.
 	if len(p.proofGenerationCh) > 0 {
+		log.Info("onBlockProposed early return", "proofGenerationChannelLength", len(p.proofGenerationCh))
+
 		end()
 		return nil
 	}