liliesh/scabiosa
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

refactor(MariaDB): Moved private functions into struct

Browse source
This commit is contained in:
netbenix 2022-01-31 20:11:08 +01:00
parent 36175632e0
commit 8d1e992631
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
SQL/MariaDBConnector.go
View file

@ -31,7 +31,7 @@ func GetMariaDBInstance(sqlConfig Tools.SQLConfig) MariaDBConnector {
return mariadb return mariadb
} }
func checkIfEventLogTableExist(db *sql.DB, mariadb MariaDBConnector) bool { func (mariadb MariaDBConnector) checkIfEventLogTableExist(db *sql.DB) bool {
rows, _ := db.Query("SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = ? AND TABLE_NAME = 'EventLog';", mariadb.Database) rows, _ := db.Query("SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = ? AND TABLE_NAME = 'EventLog';", mariadb.Database)
if !rows.Next() { if !rows.Next() {
return false return false
@ -39,7 +39,7 @@ func checkIfEventLogTableExist(db *sql.DB, mariadb MariaDBConnector) bool {
return true return true
} }
func checkIfBackupTableExist(db *sql.DB, mariadb MariaDBConnector) bool { func (mariadb MariaDBConnector) checkIfBackupTableExist(db *sql.DB) bool {
rows, _ := db.Query("SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = ? AND TABLE_NAME = 'Backups';", mariadb.Database) rows, _ := db.Query("SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = ? AND TABLE_NAME = 'Backups';", mariadb.Database)
if !rows.Next() { if !rows.Next() {
return false return false
@ -47,7 +47,7 @@ func checkIfBackupTableExist(db *sql.DB, mariadb MariaDBConnector) bool {
return true return true
} }
func checkIfBackupEntryExist(db *sql.DB, mariadb MariaDBConnector, backupName string, hostname string) bool { func (mariadb MariaDBConnector) checkIfBackupEntryExist(db *sql.DB, backupName string, hostname string) bool {
rows, _ := db.Query("SELECT * FROM `"+mariadb.Database+"`.Backups WHERE Hostname = ? AND BackupName = ?;", hostname, backupName) rows, _ := db.Query("SELECT * FROM `"+mariadb.Database+"`.Backups WHERE Hostname = ? AND BackupName = ?;", hostname, backupName)
if !rows.Next() { if !rows.Next() {
return false return false
@ -72,14 +72,14 @@ func (mariadb MariaDBConnector) createDefaultTables() {
db := createMariaDBConnection(mariadb) db := createMariaDBConnection(mariadb)
if !checkIfBackupTableExist(db, mariadb) { if !mariadb.checkIfBackupTableExist(db) {
_, err := db.Exec(backupSQL) _, err := db.Exec(backupSQL)
if err != nil { if err != nil {
logger.Fatal(err) logger.Fatal(err)
} }
} }
if !checkIfEventLogTableExist(db, mariadb) { if !mariadb.checkIfEventLogTableExist(db) {
_, err := db.Exec(eventLogSQL) _, err := db.Exec(eventLogSQL)
if err != nil { if err != nil {
logger.Fatal(err) logger.Fatal(err)
@ -108,7 +108,7 @@ func (mariadb MariaDBConnector) newBackupEntry(backupName string, lastBackup tim
hostname, _ := os.Hostname() hostname, _ := os.Hostname()
if checkIfBackupEntryExist(db, mariadb, backupName, hostname) { if mariadb.checkIfBackupEntryExist(db, backupName, hostname) {
_, err := db.Query("UPDATE `"+mariadb.Database+"`.Backups SET LastBackup = ?, LocalBackup = ?, RemoteStorage = ?, RemotePath = ?, LocalPath = ? WHERE Hostname = ? AND BackupName = ?;", lastBackup, localBackup, strconv.FormatInt(int64(storageType), 10), remotePath, localPath, hostname, backupName) _, err := db.Query("UPDATE `"+mariadb.Database+"`.Backups SET LastBackup = ?, LocalBackup = ?, RemoteStorage = ?, RemotePath = ?, LocalPath = ? WHERE Hostname = ? AND BackupName = ?;", lastBackup, localBackup, strconv.FormatInt(int64(storageType), 10), remotePath, localPath, hostname, backupName)
if err != nil { if err != nil {
logger.Fatal(err) logger.Fatal(err)