Moved MariaDBConnector and Compression do modules
This commit is contained in:
parent
33b44a6ea5
commit
465c0e7973
2 changed files with 0 additions and 0 deletions
93
SQL/MariaDBConnector.go
Normal file
93
SQL/MariaDBConnector.go
Normal file
|
|
@ -0,0 +1,93 @@
|
|||
package SQL
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
_ "github.com/go-sql-driver/mysql"
|
||||
"github.com/google/uuid"
|
||||
"scabiosa/Logging"
|
||||
"scabiosa/Tools"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
type MariaDBConnector struct {
|
||||
Address string
|
||||
Port uint16
|
||||
Database string
|
||||
DbUser string
|
||||
DbPassword string
|
||||
}
|
||||
|
||||
func GetMariaDBInstance() MariaDBConnector{
|
||||
config := Tools.GetConfig()
|
||||
var mariadb MariaDBConnector
|
||||
|
||||
mariadb.Address = config.SQLConfig.SqlAddress
|
||||
mariadb.Port = config.SQLConfig.SqlPort
|
||||
mariadb.Database = config.SQLConfig.Database
|
||||
mariadb.DbUser = config.SQLConfig.DbUser
|
||||
mariadb.DbPassword = config.SQLConfig.DbPassword
|
||||
|
||||
return mariadb
|
||||
}
|
||||
|
||||
|
||||
func checkIfEventLogTableExist(db *sql.DB, mariadb MariaDBConnector) 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
|
||||
}
|
||||
|
||||
func checkIfBackupTableExist(db *sql.DB, mariadb MariaDBConnector) 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
|
||||
}
|
||||
|
||||
func createMariaDBConnection(mariadb MariaDBConnector) *sql.DB{
|
||||
logger := Logging.DetailedLogger("MariaDB", "createConnection")
|
||||
db, err := sql.Open("mysql", mariadb.DbUser + ":" + mariadb.DbPassword + "@(" + mariadb.Address + ":" +strconv.Itoa(int(mariadb.Port))+ ")/" + mariadb.Database)
|
||||
if err != nil{
|
||||
logger.Fatal(err)
|
||||
}
|
||||
return db
|
||||
}
|
||||
|
||||
func (mariadb MariaDBConnector) createDefaultTables(){
|
||||
logger := Logging.DetailedLogger("MariaDB", "createDefaultTables")
|
||||
|
||||
eventLogSQL := "create table " + mariadb.Database +".EventLog\n(\n UUID text null,\n LogType enum ('INFO', 'WARNING', 'ERROR', 'FATAL') null,\n BackupName varchar(256) null,\n Stage enum ('COMPRESS', 'UPLOAD', 'DELETE TMP') null,\n RemoteStorage enum ('AZURE-FILE', 'AZURE-BLOB', 'NONE') null,\n Description text null,\n Timestamp datetime null\n);"
|
||||
backupSQL := "create table " + mariadb.Database +".Backups\n(\n UUID text null,\n BackupName varchar(256) null,\n LastBackup datetime null,\n LocalBackup tinyint(1) null,\n FilePath varchar(256) null,\n RemoteStorage enum ('AZURE-FILE', 'AZURE-BLOB', 'NONE') null,\n RemotePath varchar(256) null,\n `DurationToBackup (s)` double null,\n HadErrors tinyint(1) null\n);\n\n"
|
||||
|
||||
|
||||
db := createMariaDBConnection(mariadb)
|
||||
|
||||
if !checkIfBackupTableExist(db, mariadb){
|
||||
_, err := db.Exec(backupSQL)
|
||||
if err != nil{
|
||||
logger.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
if !checkIfEventLogTableExist(db, mariadb){
|
||||
_, err := db.Exec(eventLogSQL)
|
||||
if err != nil{
|
||||
logger.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
_ = db.Close()
|
||||
}
|
||||
|
||||
func (mariadb MariaDBConnector) newLogEntry(uuid uuid.UUID, logType LogType, backupName string, stage SQLStage, storageType RemoteStorageType, description string, timestamp time.Time){
|
||||
logger := Logging.DetailedLogger("MariaDB", "newLogEntry")
|
||||
|
||||
db := createMariaDBConnection(mariadb)
|
||||
|
||||
_, err := db.Query("INSERT INTO `" + mariadb.Database + "`.EventLog VALUES (?, ?, ?, ?, ?, ?, ?);", uuid.String(), strconv.FormatInt(int64(logType), 10), backupName, stage, strconv.FormatInt(int64(storageType), 10), description ,timestamp)
|
||||
if err != nil{
|
||||
logger.Fatal(err)
|
||||
}
|
||||
|
||||
}
|
||||
func (mariadb MariaDBConnector) newBackupEntry(uuid uuid.UUID, backupName string, lastBackup time.Time, localBackup bool, filePath string, storageType RemoteStorageType, remotePath string, durationToBackup time.Duration, hadErrors bool){}
|
||||
Reference in a new issue