volker.raschek/prometheus-fail2ban-exporter
1
0
Fork 0
Code Issues Pull Requests 2 Packages Projects Releases Wiki Activity

feat: fail on startup if database file does not exist (#8)

Browse Source 
Add a check when connecting to the sqlite3 database to ensure that the file
exists before connecting. If the file does not exist, the connection fails.
This commit is contained in:
Hector 2021-02-07 10:36:08 +00:00
parent 188626198f
commit 6355c9e8e1
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
db/db.go
View File

@ -3,6 +3,7 @@ package db
import ( import (
"database/sql" "database/sql"
"log" "log"
"os"
) )
const queryBadIpsPerJail = "SELECT j.name, (SELECT COUNT(1) FROM bips b WHERE j.name = b.jail) FROM jails j" const queryBadIpsPerJail = "SELECT j.name, (SELECT COUNT(1) FROM bips b WHERE j.name = b.jail) FROM jails j"
@ -14,6 +15,9 @@ type Fail2BanDB struct {
} }
func MustConnectToDb(databasePath string) *Fail2BanDB { func MustConnectToDb(databasePath string) *Fail2BanDB {
if _, err := os.Stat(databasePath); os.IsNotExist(err) {
log.Fatalf("database path does not exist: %v", err)
}
db, err := sql.Open("sqlite3", databasePath) db, err := sql.Open("sqlite3", databasePath)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)