Refactor collector to use ticker for periodic data recording and cleanup

coollabsio · Oct 15, 2024 · 696bbc9 · 696bbc9
1 parent 7e682b2
commit 696bbc9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/collector.go b/collector.go
@@ -78,6 +78,8 @@ func collector() {
 
 					cleanupTable("cpu_usage")
 					cleanupTable("memory_usage")
+					checkpoint()
+					vacuum()
 				}()
 			}
 		}

diff --git a/helpers.go b/helpers.go
@@ -14,6 +14,12 @@ func getUnixTimeInMilliUTC() string {
 
 func vacuum() {
 	go func() {
+		defer func() {
+			if r := recover(); r != nil {
+				log.Printf("Recovered from panic in vacuum: %v", r)
+			}
+		}()
+
 		_, err := db.Exec("VACUUM")
 		if err != nil {
 			log.Printf("Error vacuuming: %v", err)
@@ -22,6 +28,12 @@ func vacuum() {
 }
 func checkpoint() {
 	go func() {
+		defer func() {
+			if r := recover(); r != nil {
+				log.Printf("Recovered from panic in checkpoint: %v", r)
+			}
+		}()
+
 		_, err := db.Exec("CHECKPOINT")
 		if err != nil {
 			log.Printf("Error checkpointing: %v", err)