From 2661ce9a2bda2b6c1efc0ba1fef873c9dc91bb7c Mon Sep 17 00:00:00 2001
From: Julien Dessaux
Date: Sat, 22 Feb 2020 12:35:05 +0100
Subject: Added tests to the spool package, and reworked the code around that.

---
 main.go | 30 ++++++++++++++++++++++--------
 1 file changed, 22 insertions(+), 8 deletions(-)

(limited to 'main.go')

diff --git a/main.go b/main.go
index bc5945c..62347ab 100644
--- a/main.go
+++ b/main.go
@@ -8,13 +8,17 @@ import (
 	"fmt"
 	"log"
 	"os"
+	"path/filepath"
 	"time"
 )
 
+const (
+	spoolFileName = "bareos-zabbix-check.spool"
+)
+
 func main() {
 	var (
 		config        config.Config
-		spool         spool.Spool
 		errorString   string
 		missingString string
 	)
@@ -58,14 +62,24 @@ func main() {
 	// We will check for errors in loading the spool file only at the end. If all jobs ran successfully without errors
 	// in the state file and we manage to write a new spool file without errors, then we will ignore any error here to
 	// avoid false positives during backup bootstrap
-	err = spool.Load(&config)
+	// Open the spool file
+	spoolFile, spoolErr := os.Open(filepath.Join(config.WorkDir(), spoolFileName))
+	var spoolJobs []job.Job
+	if err == nil {
+		defer spoolFile.Close()
+		spoolJobs, spoolErr = spool.Parse(spoolFile)
+	}
 
-	jobs = job.KeepOldestOnly(append(jobs, spool.Jobs()...))
-	spool.SetJobs(job.KeepSuccessOnly(jobs))
+	jobs = job.KeepOldestOnly(append(jobs, spoolJobs...))
 
 	// we write this new spool
-	if err2 := spool.Save(); err2 != nil {
-		fmt.Printf("AVERAGE: Error saving the spool file : %s\n", err2)
+	spoolFile, err = os.Create(filepath.Join(config.WorkDir(), spoolFileName))
+	if err == nil {
+		defer spoolFile.Close()
+		err = spool.Serialize(spoolFile, jobs)
+	}
+	if err != nil {
+		fmt.Printf("AVERAGE: Error saving the spool file : %s\n", err)
 		os.Exit(0)
 	}
 
@@ -91,8 +105,8 @@ func main() {
 	// Finally we output
 	if errorString != "" || missingString != "" {
 		fmt.Printf("AVERAGE: %s %s", errorString, missingString)
-		if err != nil {
-			fmt.Printf(" additionnal errors: %s", err)
+		if spoolErr != nil {
+			fmt.Printf(" additionnal errors: %s", spoolErr)
 		}
 	} else {
 		fmt.Printf("OK")
-- 
cgit v1.2.3