diff --git a/SQL/MSSQLConnector.go b/SQL/MSSQLConnector.go index 24ab2fe..448182a 100644 --- a/SQL/MSSQLConnector.go +++ b/SQL/MSSQLConnector.go @@ -34,27 +34,18 @@ func GetMSSQLInstance(sqlConfig Tools.SQLConfig) MSSQLConnector { func (mssql MSSQLConnector) checkIfEventLogTableExist(db *sql.DB) bool { rows, _ := db.Query("SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = 'dbo' AND TABLE_NAME = 'EventLog';") - if !rows.Next() { - return false - } - return true + return rows.Next() } func (mssql MSSQLConnector) checkIfBackupTableExist(db *sql.DB) bool { rows, _ := db.Query("SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = 'dbo' AND TABLE_NAME = 'Backups';") - if !rows.Next() { - return false - } - return true + return rows.Next() } func (mssql MSSQLConnector) checkIfBackupEntryExist(db *sql.DB, backupName string, hostname string) bool { query := fmt.Sprintf("SELECT * FROM dbo.Backups WHERE Hostname = '%s' AND BackupName = '%s'", hostname, backupName) rows, _ := db.Query(query) - if !rows.Next() { - return false - } - return true + return rows.Next() } func createMSSQLConnection(mssql MSSQLConnector) *sql.DB { diff --git a/SQL/MariaDBConnector.go b/SQL/MariaDBConnector.go index adc50a2..09b0575 100644 --- a/SQL/MariaDBConnector.go +++ b/SQL/MariaDBConnector.go @@ -33,26 +33,17 @@ func GetMariaDBInstance(sqlConfig Tools.SQLConfig) MariaDBConnector { 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) - if !rows.Next() { - return false - } - return true + return rows.Next() } 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) - if !rows.Next() { - return false - } - return true + return rows.Next() } 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) - if !rows.Next() { - return false - } - return true + return rows.Next() } func createMariaDBConnection(mariadb MariaDBConnector) *sql.DB {