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

Updated SQL

Browse source
This commit is contained in:
netbenix 2021-11-28 12:33:22 +01:00
parent fcbaaa092f
commit 675a8dafb3
5 changed files with 32 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

16
StorageTypes/AzureFileStorage.go
View file

@ -23,7 +23,7 @@ type AzureFileStorage struct{
}
func (azure AzureFileStorage) upload(fileName string){
func (azure AzureFileStorage) upload(fileName string, backupName string){
logger := Logging.DetailedLogger("AzureFileStorage", "upload")
file, err := os.Open(fileName)
@ -48,8 +48,8 @@ func (azure AzureFileStorage) upload(fileName string){
ctx := context.Background()
fmt.Printf("[%s] Starting upload to Azure File Share...\n", strings.Trim(filepath.Base(fileName), ".bak"))
SQL.NewLogEntry(SQL.GetSQLInstance(), uuid.New(), SQL.LogInfo, filepath.Base(fileName), SQL.SQLStage_Upload, SQL.REMOTE_AZURE_FILE, "Starting upload.", time.Now())
fmt.Printf("[%s] Starting upload to Azure File Share...\n", backupName, ".bak")
SQL.NewLogEntry(SQL.GetSQLInstance(), uuid.New(), SQL.LogInfo, backupName, SQL.SQLStage_Upload, SQL.REMOTE_AZURE_FILE, "Starting upload.", time.Now())
err = azfile.UploadFileToAzureFile(ctx, file, fileURL,
azfile.UploadToAzureFileOptions{
@ -58,11 +58,15 @@ func (azure AzureFileStorage) upload(fileName string){
CacheControl: "no-transform",
},
Progress: func(bytesTransferred int64){
fmt.Printf("[%s] Uploaded %d of %d bytes.\n", strings.Trim(filepath.Base(fileName), ".bak") ,bytesTransferred, fileSize.Size())
fmt.Printf("[%s] Uploaded %d of %d bytes.\n", strings.Trim(backupName, ".bak") ,bytesTransferred, fileSize.Size())
}})
fmt.Printf("[%s] Upload finished.\n", strings.Trim(filepath.Base(fileName), ".bak"))
SQL.NewLogEntry(SQL.GetSQLInstance(), uuid.New(), SQL.LogInfo, filepath.Base(fileName), SQL.SQLStage_Upload, SQL.REMOTE_AZURE_FILE, "Finished upload.", time.Now())
if err != nil{
logger.Fatal(err)
}
fmt.Printf("[%s] Upload finished.\n", strings.Trim(backupName, ".bak"))
SQL.NewLogEntry(SQL.GetSQLInstance(), uuid.New(), SQL.LogInfo, backupName, SQL.SQLStage_Upload, SQL.REMOTE_AZURE_FILE, "Finished upload.", time.Now())
}
func readConfig() []byte {

16
StorageTypes/StorageInterface.go
View file

@ -1,11 +1,13 @@
package StorageTypes
import "scabiosa/SQL"
type Storage interface {
upload(fileName string)
upload(fileName string, backupName string)
}
func UploadFile(storage Storage, fileName string){
storage.upload(fileName)
func UploadFile(storage Storage, fileName string, backupName string){
storage.upload(fileName, backupName)
}
func CheckStorageType(storageType string) Storage{
@ -13,6 +15,12 @@ func CheckStorageType(storageType string) Storage{
if storageType == "azure-fileshare"{
return GetAzureStorage()
}
return nil
}
func CheckRemoteStorageType(storageType string) SQL.RemoteStorageType {
if storageType == "azure-fileshare"{
return SQL.REMOTE_AZURE_FILE
}
return 3
}