104 lines
2.8 KiB
Go
104 lines
2.8 KiB
Go
package zabbix
|
|
|
|
import (
|
|
"bareos-zabbix-check/job"
|
|
"bareos-zabbix-check/spool"
|
|
"bareos-zabbix-check/state"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
spoolFileName = "bareos-zabbix-check.spool"
|
|
)
|
|
|
|
var now = uint64(time.Now().Unix())
|
|
|
|
// Main the true main function of this program
|
|
func Main() string {
|
|
err := processFlags()
|
|
if err != nil {
|
|
return fmt.Sprintf("INFO Failed to init programm : %s", err)
|
|
}
|
|
// Open the state file
|
|
stateFile, err := os.Open(stateFileName)
|
|
if err != nil {
|
|
return fmt.Sprintf("INFO Could not open state file : %s", err)
|
|
}
|
|
defer stateFile.Close()
|
|
// parse the state file
|
|
header, err := state.ParseHeader(stateFile)
|
|
if err != nil {
|
|
return fmt.Sprintf("INFO Could not parse state file header : %s", err)
|
|
}
|
|
// seek to the job entries in the state file
|
|
offset, err := stateFile.Seek(int64(header.LastJobsAddr), 0)
|
|
if err != nil {
|
|
return fmt.Sprintf("INFO Couldn't seek to jobs position in state file : %s", err)
|
|
}
|
|
if uint64(offset) != header.LastJobsAddr {
|
|
return fmt.Sprint("INFO Truncated state file")
|
|
}
|
|
// Then parse the jobs in the state file
|
|
jobs, err := state.ParseJobs(stateFile)
|
|
if err != nil {
|
|
return fmt.Sprintf("INFO Could not parse jobs in state file : %s", err)
|
|
}
|
|
|
|
// 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
|
|
// Open the spool file
|
|
spoolFile, spoolErr := os.Open(filepath.Join(workDir, spoolFileName))
|
|
var spoolJobs []job.Job
|
|
if err == nil {
|
|
defer spoolFile.Close()
|
|
spoolJobs, spoolErr = spool.Parse(spoolFile)
|
|
}
|
|
|
|
jobs = job.KeepOldestOnly(append(jobs, spoolJobs...))
|
|
|
|
// we write this new spool
|
|
spoolFile, err = os.Create(filepath.Join(workDir, spoolFileName))
|
|
if err == nil {
|
|
defer spoolFile.Close()
|
|
err = spool.Serialize(spoolFile, jobs)
|
|
}
|
|
if err != nil {
|
|
return fmt.Sprintf("AVERAGE: Error saving the spool file : %s\n", err)
|
|
}
|
|
|
|
var (
|
|
errorString string
|
|
missingString string
|
|
)
|
|
// We build the error strings
|
|
for i := 0; i < len(jobs); i++ {
|
|
job := jobs[i]
|
|
if job.Success {
|
|
if job.Timestamp < now-24*3600 {
|
|
if missingString == "" {
|
|
missingString = fmt.Sprintf("missing: %s", job.Name)
|
|
} else {
|
|
missingString = fmt.Sprintf("%s, %s", missingString, job.Name)
|
|
}
|
|
}
|
|
} else {
|
|
if errorString == "" {
|
|
errorString = fmt.Sprintf("errors: %s", job.Name)
|
|
} else {
|
|
errorString = fmt.Sprintf("%s, %s", errorString, job.Name)
|
|
}
|
|
}
|
|
}
|
|
// Finally we output
|
|
if errorString != "" || missingString != "" {
|
|
if spoolErr != nil {
|
|
return fmt.Sprintf("AVERAGE: %s %s %s", errorString, missingString, spoolErr)
|
|
}
|
|
return fmt.Sprintf("AVERAGE: %s %s", errorString, missingString)
|
|
}
|
|
return "OK"
|
|
}
|