Add COPY_ONLY flag in backup options (#34)

Signed-off-by: Anisur Rahman <[email protected]>

cmd/sqlserver/backup_push.go

@@ -10,13 +10,14 @@ const backupPushShortDescription = "Creates new backup and pushes it to the stor
 
 var backupPushDatabases []string
 var backupUpdateLatest bool
+var copyOnly bool
 
 var backupPushCmd = &cobra.Command{
 	Use:   "backup-push",
 	Short: backupPushShortDescription,
 	Run: func(cmd *cobra.Command, args []string) {
 		internal.ConfigureLimiters()
-		sqlserver.HandleBackupPush(backupPushDatabases, backupUpdateLatest)
+		sqlserver.HandleBackupPush(backupPushDatabases, backupUpdateLatest, copyOnly)
 	},
 }
 
@@ -25,5 +26,7 @@ func init() {
 		"List of databases to backup. All not-system databases as default")
 	backupPushCmd.PersistentFlags().BoolVarP(&backupUpdateLatest, "update-latest", "u", false,
 		"Update latest backup instead of creating new one")
+	backupPushCmd.PersistentFlags().BoolVarP(&copyOnly, "copy-only", "c", false,
+		"Backup with COPY_ONLY option")
 	cmd.AddCommand(backupPushCmd)
 }

internal/databases/sqlserver/backup_push_handler.go

@@ -14,7 +14,7 @@ import (
 	"github.com/wal-g/wal-g/utility"
 )
 
-func HandleBackupPush(dbnames []string, updateLatest bool) {
+func HandleBackupPush(dbnames []string, updateLatest, copyOnly bool) {
 	ctx, cancel := context.WithCancel(context.Background())
 	signalHandler := utility.NewSignalHandler(ctx, cancel, []os.Signal{syscall.SIGINT, syscall.SIGTERM})
 	defer func() { _ = signalHandler.Close() }()
@@ -56,7 +56,7 @@ func HandleBackupPush(dbnames []string, updateLatest bool) {
 	}
 	builtinCompression := blob.UseBuiltinCompression()
 	err = runParallel(func(i int) error {
-		return backupSingleDatabase(ctx, db, backupName, dbnames[i], builtinCompression)
+		return backupSingleDatabase(ctx, db, backupName, dbnames[i], builtinCompression, copyOnly)
 	}, len(dbnames), getDBConcurrency())
 	tracelog.ErrorLogger.FatalfOnError("overall backup failed: %v", err)
 
@@ -71,7 +71,7 @@ func HandleBackupPush(dbnames []string, updateLatest bool) {
 	tracelog.InfoLogger.Printf("backup finished")
 }
 
-func backupSingleDatabase(ctx context.Context, db *sql.DB, backupName string, dbname string, builtinCompression bool) error {
+func backupSingleDatabase(ctx context.Context, db *sql.DB, backupName string, dbname string, builtinCompression, copyOnlyBackup bool) error {
 	baseURL := getDatabaseBackupURL(backupName, dbname)
 	size, blobCount, err := estimateDBSize(db, dbname)
 	if err != nil {
@@ -84,6 +84,9 @@ func backupSingleDatabase(ctx context.Context, db *sql.DB, backupName string, db
 	if builtinCompression {
 		sql += ", COMPRESSION"
 	}
+	if copyOnlyBackup {
+		sql += ", COPY_ONLY"
+	}
 	tracelog.InfoLogger.Printf("starting backup database [%s] to %s", dbname, urls)
 	tracelog.DebugLogger.Printf("SQL: %s", sql)
 	_, err = db.ExecContext(ctx, sql)