From 508978d586a9e9f98e6c1080160b520fc2bad441 Mon Sep 17 00:00:00 2001
From: Jan Eitzinger <jan@moebiusband.org>
Date: Tue, 3 Sep 2024 15:59:01 +0200
Subject: [PATCH] Initial attempt to update footprints in transaction

---
 internal/archiver/archiver.go                 |  3 ++-
 .../taskManager/updateFootprintService.go     | 20 +++++++++++++++++--
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/internal/archiver/archiver.go b/internal/archiver/archiver.go
index abaecd6..de84cf0 100644
--- a/internal/archiver/archiver.go
+++ b/internal/archiver/archiver.go
@@ -24,7 +24,8 @@ func ArchiveJob(job *schema.Job, ctx context.Context) (*schema.JobMeta, error) {
 	}
 
 	scopes := []schema.MetricScope{schema.MetricScopeNode}
-	if job.NumNodes <= 8 { // FIXME: Add a config option for this
+	// FIXME: Add a config option for this
+	if job.NumNodes <= 8 {
 		// This will add the native scope if core scope is not available
 		scopes = append(scopes, schema.MetricScopeCore)
 	}
diff --git a/internal/taskManager/updateFootprintService.go b/internal/taskManager/updateFootprintService.go
index 510c73e..2fdd6b9 100644
--- a/internal/taskManager/updateFootprintService.go
+++ b/internal/taskManager/updateFootprintService.go
@@ -25,6 +25,12 @@ func RegisterFootprintWorker() {
 			func() {
 				s := time.Now()
 				log.Printf("Update Footprints started at %s", s.Format(time.RFC3339))
+
+				t, err := jobRepo.TransactionInit()
+				if err != nil {
+					log.Errorf("Failed TransactionInit %v", err)
+				}
+
 				for _, cluster := range archive.Clusters {
 					jobs, err := jobRepo.FindRunningJobs(cluster.Name)
 					if err != nil {
@@ -87,12 +93,22 @@ func RegisterFootprintWorker() {
 							log.Errorf("Update job (dbid: %d) failed at update Energy step: %s", job.ID, err.Error())
 							continue
 						}
-						if err := jobRepo.Execute(stmt); err != nil {
-							log.Errorf("Update job (dbid: %d) failed at db execute: %s", job.ID, err.Error())
+
+						query, args, err := stmt.ToSql()
+						if err != nil {
+							log.Errorf("Failed in ToSQL conversion %v", err)
 							continue
 						}
+						jobRepo.TransactionAdd(t, query, args)
+						// if err := jobRepo.Execute(stmt); err != nil {
+						// 	log.Errorf("Update job (dbid: %d) failed at db execute: %s", job.ID, err.Error())
+						// 	continue
+						// }
 					}
+
+					jobRepo.TransactionCommit(t)
 				}
+				jobRepo.TransactionEnd(t)
 				log.Printf("Update Footprints is done and took %s", time.Since(s))
 			}))
 }