aboutsummaryrefslogtreecommitdiff
path: root/spool.go
diff options
context:
space:
mode:
authorJulien Dessaux2020-01-08 10:51:59 +0100
committerJulien Dessaux2020-01-08 10:51:59 +0100
commit18e91937cd94ea199acff95c2e25015482d82005 (patch)
treee09378ea4ee0969d638554bf1f9c58f65b3468ff /spool.go
downloadbareos-zabbix-check-18e91937cd94ea199acff95c2e25015482d82005.tar.gz
bareos-zabbix-check-18e91937cd94ea199acff95c2e25015482d82005.tar.bz2
bareos-zabbix-check-18e91937cd94ea199acff95c2e25015482d82005.zip
Initial import
Diffstat (limited to 'spool.go')
-rw-r--r--spool.go72
1 files changed, 72 insertions, 0 deletions
diff --git a/spool.go b/spool.go
new file mode 100644
index 0000000..e3d00d9
--- /dev/null
+++ b/spool.go
@@ -0,0 +1,72 @@
+package main
+
+import (
+ "encoding/csv"
+ "fmt"
+ "log"
+ "os"
+ "path"
+ "strconv"
+)
+
+// jobs is a map that matches a job name string to its last successfull run timestamp
+type jobs map[string]uint64
+
+func loadSpool() (entries jobs, err error) {
+ var (
+ file *os.File
+ lines [][]string
+ )
+ // We read the spool
+ file, err = os.Open(path.Join(workDir, spoolFile))
+ if err != nil {
+ return nil, fmt.Errorf("INFO Couldn't open spool file: %s", err)
+ }
+ defer file.Close()
+ lines, err = csv.NewReader(file).ReadAll()
+ if err != nil {
+ return nil, fmt.Errorf("INFO Corrupted spool file : %s", err)
+ }
+ if verbose {
+ log.Printf("Spool file content : %v\n", lines)
+ }
+
+ entries = make(map[string]uint64)
+ for _, line := range lines {
+ var (
+ i int
+ )
+ i, err = strconv.Atoi(line[1])
+ if err != nil {
+ return nil, fmt.Errorf("INFO Corrupted spool file : couldn't parse timestamp entry")
+ }
+ entries[line[0]] = uint64(i)
+ }
+ return
+}
+
+func saveSpool(entries jobs) (err error) {
+ var (
+ file *os.File
+ lines [][]string
+ jobName string
+ ts uint64
+ i int
+ )
+ file, err = os.Create(path.Join(workDir, spoolFile))
+ if err != nil {
+ return fmt.Errorf("INFO Couldn't open spool file for writing : %s", err)
+ }
+ defer file.Close()
+
+ lines = make([][]string, len(entries))
+ i = 0
+ for jobName, ts = range entries {
+ lines[i] = make([]string, 2)
+ lines[i][0] = jobName
+ lines[i][1] = fmt.Sprintf("%d", ts)
+ i++
+ }
+ err = csv.NewWriter(file).WriteAll(lines)
+ return
+}