adyxax/bareos-zabbix-check
1
0
Fork 0
Code Issues Activity

Big rafactoring : code split in several modules and some other best practices

Browse source
This commit is contained in:
Julien Dessaux 2020-01-30 17:44:42 +01:00
parent e07ce016c4
commit bea8e5aba8
16 changed files with 506 additions and 386 deletions
Download patch file Download diff file Expand all files Collapse all files

151
main.go
View file

@ -1,146 +1,63 @@
package main
import (
"flag"
"bareos-zabbix-check/config"
"bareos-zabbix-check/job"
"bareos-zabbix-check/spool"
"bareos-zabbix-check/state"
"fmt"
"log"
"os"
"path"
"time"
)
const (
bareosWorkDir = "/var/lib/bareos"
bareosStateFile = "bareos-fd.9102.state"
baculaWorkDir = "/var/lib/bacula"
baculaStateFile = "bacula-fd.9102.state"
spoolFile = "bareos-zabbix-check.spool"
)
// We declare globally the variables that will hold the command line arguments
var (
verbose bool
quiet bool
stateFile string
workDir string
)
func main() {
var (
info os.FileInfo
err error
successfulJobs jobs
errorJobs jobs
spoolJobs jobs
jobName string
ts uint64
now uint64
errorString string
missingString string
config config.Config
spool spool.Spool
state state.State
errorString string
missingString string
)
// command line arguments parsing
flag.BoolVar(&verbose, "v", false, "Activates verbose debugging output, defaults to false.")
flag.BoolVar(&quiet, "q", false, "Suppress all output, suitable to force a silent update of the spool file.")
flag.StringVar(&stateFile, "f", "", "Force the state file to use, defaults to "+bareosStateFile+" if it exists else "+baculaStateFile+".")
flag.StringVar(&workDir, "w", "", "Force the work directory to use, defaults to "+bareosWorkDir+" if it exists else "+baculaWorkDir+".")
flag.Parse()
// Determine the work directory to use.
if workDir != "" {
info, err = os.Stat(workDir)
if os.IsNotExist(err) || !info.IsDir() {
fmt.Printf("INFO Invalid work directory %s : it does not exist or is not a directory.\n", workDir)
os.Exit(0)
}
} else {
workDir = "/var/lib/bareos"
info, err = os.Stat(workDir)
if os.IsNotExist(err) || !info.IsDir() {
workDir = "/var/lib/bacula"
info, err := os.Stat(workDir)
if os.IsNotExist(err) || !info.IsDir() {
fmt.Println("INFO Could not find a suitable work directory. Is bareos or bacula installed?")
os.Exit(0)
}
}
}
workDir = path.Clean(workDir)
if verbose {
log.Println("Setting work directory to ", workDir)
}
// Finds the state file to parse
if stateFile != "" {
stateFile = path.Join(workDir, stateFile)
info, err = os.Stat(stateFile)
if os.IsNotExist(err) || info.IsDir() {
fmt.Printf("INFO The state file %s does not exist.\n", stateFile)
os.Exit(0)
}
} else {
stateFile = path.Join(workDir, bareosStateFile)
info, err = os.Stat(stateFile)
if os.IsNotExist(err) || info.IsDir() {
stateFile = path.Join(workDir, baculaStateFile)
info, err = os.Stat(stateFile)
if os.IsNotExist(err) || info.IsDir() {
fmt.Println("INFO Could not find a suitable state file. Has a job ever run?")
os.Exit(0)
}
}
}
if verbose {
log.Println("Using state file ", stateFile)
}
successfulJobs, errorJobs, err = parseStateFile()
config.Init()
err := state.Parse(&config)
if err != nil {
fmt.Print(err)
os.Exit(0)
}
// We will check for errors in loading the spool file only if necessary. 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
spoolJobs, err = loadSpool()
// 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)
jobs := job.KeepOldestOnly(append(state.Jobs(), spool.Jobs()...))
spool.SetJobs(job.KeepSuccessOnly(jobs))
// if we have jobs in the spool we merge this list with successfull jobs from the state file
if err == nil {
for jobName, ts = range spoolJobs {
var (
current uint64
ok bool
)
current, ok = successfulJobs[jobName]
if !ok || current < ts {
successfulJobs[jobName] = ts
}
}
}
// we write this new spool
if err2 := saveSpool(successfulJobs); err2 != nil {
if err2 := spool.Save(); err2 != nil {
fmt.Printf("AVERAGE: Error saving the spool file : %s\n", err2)
os.Exit(0)
}
// We build the error string listing the jobs in error
for jobName, ts = range errorJobs {
if errorString == "" {
errorString = fmt.Sprintf("errors: %s", jobName)
now := uint64(time.Now().Unix())
// We build the error strings
for _, job := range jobs {
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 {
errorString = fmt.Sprintf("%s, %s", errorString, jobName)
}
}
now = uint64(time.Now().Unix())
// Next we check if all jobs ran recently and build the missing string
for jobName, ts = range successfulJobs {
if ts < now-24*3600 {
if missingString == "" {
missingString = fmt.Sprintf("missing: %s", jobName)
if errorString == "" {
errorString = fmt.Sprintf("errors: %s", job.Name)
} else {
missingString = fmt.Sprintf("%s, %s", missingString, jobName)
errorString = fmt.Sprintf("%s, %s", errorString, job.Name)
}
}
}
// Finally we output
if errorString != "" || missingString != "" {
fmt.Printf("AVERAGE: %s %s", errorString, missingString)
if err != nil {