scabiosa/SQL/MSSQLConnector.go

package SQL

import (
	"database/sql"
	"fmt"
	mssqlpkg "github.com/denisenkom/go-mssqldb"
	"github.com/google/uuid"
	"net/url"
	"os"
	"scabiosa/Logging"
	"scabiosa/Tools"
	"time"
)

type MSSQLConnector struct {
	Address    string
	Port       uint16
	Database   string
	DbUser     string
	DbPassword string
}

func GetMSSQLInstance(sqlConfig Tools.SQLConfig) MSSQLConnector {
	var mssql MSSQLConnector

	mssql.Address = sqlConfig.SqlAddress
	mssql.Port = sqlConfig.SqlPort
	mssql.Database = sqlConfig.Database
	mssql.DbUser = sqlConfig.DbUser
	mssql.DbPassword = sqlConfig.DbPassword

	return mssql
}

func (mssql MSSQLConnector) checkIfEventLogTableExist(db *sql.DB) bool {
	rows, _ := db.Query("SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = 'dbo' AND TABLE_NAME = 'EventLog';")
	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';")
	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)
	return rows.Next()
}

func createMSSQLConnection(mssql MSSQLConnector) *sql.DB {
	logger := Logging.DetailedLogger("MS-SQL", "createConnection")

	query := url.Values{}
	query.Add("app name", "scabiosa")
	query.Add("database", "scabiosa-test")

	sqlSettings := &url.URL{
		Scheme:   "sqlserver",
		User:     url.UserPassword(mssql.DbUser, mssql.DbPassword),
		Host:     fmt.Sprintf("%s:%d", mssql.Address, mssql.Port),
		RawQuery: query.Encode(),
	}

	connector, err := mssqlpkg.NewConnector(sqlSettings.String())
	if err != nil {
		logger.Fatal(err)
	}

	connector.SessionInitSQL = "SET ANSI_NULLS ON"

	db := sql.OpenDB(connector)

	return db
}

func (mssql MSSQLConnector) createDefaultTables() {
	logger := Logging.DetailedLogger("MS-SQL", "createDefaultTables")

	eventLogSQL := "create table dbo.EventLog(UUID text null, LogType VARCHAR(20) NOT NULL CHECK (LogType IN('INFO', 'WARNING', 'ERROR', 'FATAL')), Hostname varchar(256) null, BackupName varchar(256) null, Stage VARCHAR(20) NOT NULL CHECK (Stage IN('COMPRESS', 'UPLOAD', 'DELETE TMP')), RemoteStorage VARCHAR(20) NOT NULL CHECK (RemoteStorage IN('AZURE-FILE', 'AZURE-BLOB', 'NONE')), Description text null, Timestamp datetime null);"
	backupSQL := "create table dbo.Backups(UUID text null, Hostname varchar(256) null, BackupName varchar(256) null, LastBackup datetime null, LocalBackup tinyint null, FilePath varchar(256) null, RemoteStorage VARCHAR(20) NOT NULL CHECK (RemoteStorage IN('AZURE-FILE', 'AZURE-BLOB', 'NONE')), RemotePath varchar(256) null, LocalPath varchar(256) null);"

	db := createMSSQLConnection(mssql)

	if !mssql.checkIfBackupTableExist(db) {
		_, err := db.Exec(backupSQL)
		if err != nil {
			logger.Fatal(err)
		}
	}

	if !mssql.checkIfEventLogTableExist(db) {
		_, err := db.Exec(eventLogSQL)
		if err != nil {
			logger.Fatal(err)
		}
	}

	_ = db.Close()
}
func (mssql MSSQLConnector) newLogEntry(uuid uuid.UUID, logType LogType, backupName string, stage SQLStage, storageType RemoteStorageType, description string, timestamp time.Time) {
	logger := Logging.DetailedLogger("MS-SQL", "newLogEntry")
	db := createMSSQLConnection(mssql)

	hostname, _ := os.Hostname()
	query := fmt.Sprintf("INSERT INTO dbo.EventLog VALUES ('%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s')", uuid.String(), logType.String(), hostname, backupName, stage.String(), storageType.String(), description, timestamp.Format("2006-01-02 15:04:05.999"))
	_, err := db.Query(query)
	if err != nil {
		logger.Fatal(err)
	}
}
func (mssql MSSQLConnector) newBackupEntry(backupName string, lastBackup time.Time, localBackup bool, filePath string, storageType RemoteStorageType, remotePath string, localPath string) {
	logger := Logging.DetailedLogger("MS-SQL", "newBackupEntry")
	db := createMSSQLConnection(mssql)

	hostname, _ := os.Hostname()
	var localBackupInt uint8
	if localBackup {
		localBackupInt = 1
	}

	if mssql.checkIfBackupEntryExist(db, backupName, hostname) {
		queryUpdate := fmt.Sprintf("UPDATE dbo.Backups SET Lastbackup = '%s', LocalBackup = %d, RemoteStorage = '%s', RemotePath = '%s', LocalPath = '%s' WHERE Hostname = '%s' AND BackupName = '%s'", lastBackup.Format("2006-01-02 15:04:05.999"), localBackupInt, storageType.String(), remotePath, localPath, hostname, backupName)
		_, err := db.Query(queryUpdate)
		if err != nil {
			logger.Fatal(err)
		}
	} else {
		queryInsert := fmt.Sprintf("INSERT INTO dbo.Backups VALUES ('%s', '%s', '%s', '%s', %d, '%s', '%s', '%s', '%s')", uuid.New(), hostname, backupName, lastBackup.Format("2006-01-02 15:04:05.999"), localBackupInt, filePath, storageType.String(), remotePath, localPath)
		_, err := db.Query(queryInsert)
		if err != nil {
			logger.Fatal(err)
		}
	}
}