diff options
Diffstat (limited to 'spool')
-rw-r--r-- | spool/load.go | 40 | ||||
-rw-r--r-- | spool/save.go | 28 | ||||
-rw-r--r-- | spool/spool.go | 26 |
3 files changed, 94 insertions, 0 deletions
diff --git a/spool/load.go b/spool/load.go new file mode 100644 index 0000000..282fdc6 --- /dev/null +++ b/spool/load.go @@ -0,0 +1,40 @@ +package spool + +import ( + "bareos-zabbix-check/config" + "bareos-zabbix-check/job" + "encoding/csv" + "fmt" + "log" + "os" + "path" + "strconv" +) + +// Load loads a spool file in path +func (s *Spool) Load(c *config.Config) (err error) { + s.config = c + // We read the spool + file, err := os.Open(path.Join(c.WorkDir(), spoolFile)) + if err != nil { + return fmt.Errorf("Couldn't open spool file, starting from scratch: %s", err) + } + defer file.Close() + lines, err := csv.NewReader(file).ReadAll() + if err != nil { + return fmt.Errorf("Corrupted spool file, starting from scratch : %s", err) + } + if c.Verbose() { + log.Printf("Spool file content : %v\n", lines) + } + + for _, line := range lines { + var i int + i, err = strconv.Atoi(line[1]) + if err != nil { + return fmt.Errorf("Corrupted spool file : couldn't parse timestamp entry") + } + s.jobs = append(s.jobs, job.Job{Name: line[0], Timestamp: uint64(i), Success: true}) + } + return +} diff --git a/spool/save.go b/spool/save.go new file mode 100644 index 0000000..b01dc7b --- /dev/null +++ b/spool/save.go @@ -0,0 +1,28 @@ +package spool + +import ( + "encoding/csv" + "fmt" + "os" + "path" +) + +// Save writes a spool on the disk +func (s *Spool) Save() (err error) { + file, err := os.Create(path.Join(s.config.WorkDir(), spoolFile)) + if err != nil { + return + } + defer file.Close() + + lines := make([][]string, len(s.jobs)) + var i int = 0 + for _, job := range s.jobs { + lines[i] = make([]string, 2) + lines[i][0] = job.Name + lines[i][1] = fmt.Sprintf("%d", job.Timestamp) + i++ + } + err = csv.NewWriter(file).WriteAll(lines) + return +} diff --git a/spool/spool.go b/spool/spool.go new file mode 100644 index 0000000..e095979 --- /dev/null +++ b/spool/spool.go @@ -0,0 +1,26 @@ +package spool + +import ( + "bareos-zabbix-check/config" + "bareos-zabbix-check/job" +) + +const ( + spoolFile = "bareos-zabbix-check.spool" +) + +// Spool is an object for manipulating a bareos spool file +type Spool struct { + config *config.Config + jobs []job.Job +} + +// Jobs exports a spool to a jobs list +func (s *Spool) Jobs() []job.Job { + return s.jobs +} + +// SetJobs sets a jobs list +func (s *Spool) SetJobs(jobs []job.Job) { + s.jobs = jobs +} |