feat(backup): use Scylla backup API

Fixes #4143

pkg/service/backup/worker_scylla_upload.go

@@ -3,9 +3,14 @@
 package backup
 
 import (
+	"context"
 	"slices"
+	"time"
 
+	"github.com/pkg/errors"
+	"github.com/scylladb/scylla-manager/v3/pkg/scyllaclient"
 	. "github.com/scylladb/scylla-manager/v3/pkg/service/backup/backupspec"
+	"github.com/scylladb/scylla-manager/v3/swagger/gen/scylla/v1/models"
 )
 
 // Decides whether we should use Scylla backup API for uploading the files.
@@ -24,3 +29,92 @@ func (w *worker) useScyllaBackupAPI(hi hostInfo) (bool, error) {
 	}
 	return nc.SupportsScyllaBackupRestoreAPI()
 }
+
+func (w *worker) scyllaBackup(ctx context.Context, hi hostInfo, d snapshotDir) error {
+	if d.Progress.ScyllaTaskID == "" || !w.scyllaCanAttachToTask(ctx, hi.IP, d.Progress.ScyllaTaskID) {
+		prefix := w.remoteSSTableDir(hi, d)
+		// Agent's proxy can resolve provider into the endpoint.
+		id, err := w.Client.ScyllaBackup(ctx, hi.IP, string(hi.Location.Provider), hi.Location.Path, prefix, d.Keyspace, d.Table, w.SnapshotTag)
+		if err != nil {
+			return errors.Wrap(err, "backup")
+		}
+
+		w.Logger.Info(ctx, "Backing up dir", "host", d.Host, "keyspace", d.Keyspace, "table", d.Table, "prefix", prefix, "task id", id)
+		d.Progress.ScyllaTaskID = id
+		w.onRunProgress(ctx, d.Progress)
+	}
+
+	if err := w.scyllaWaitTask(ctx, d.Progress.ScyllaTaskID, d); err != nil {
+		w.Logger.Error(ctx, "Backing up dir failed", "host", d.Host, "task id", d.Progress.TaskID, "error", err)
+		return err
+	}
+	return nil
+}
+
+func (w *worker) scyllaCanAttachToTask(ctx context.Context, host, taskID string) bool {
+	task, err := w.Client.ScyllaTaskProgress(ctx, host, taskID)
+	if err != nil {
+		w.Logger.Error(ctx, "Failed to fetch task info",
+			"host", host,
+			"task id", taskID,
+			"error", err,
+		)
+		return false
+	}
+
+	state := scyllaclient.ScyllaTaskState(task.State)
+	return state == scyllaclient.ScyllaTaskStateDone ||
+		state == scyllaclient.ScyllaTaskStateRunning ||
+		state == scyllaclient.ScyllaTaskStateCreated
+}
+
+func (w *worker) scyllaWaitTask(ctx context.Context, id string, d snapshotDir) (err error) {
+	defer func() {
+		// On error abort task
+		if err != nil {
+			w.Logger.Info(ctx, "Stop task", "host", d.Host, "id", id)
+			// Watch out for already cancelled context
+			if e := w.Client.ScyllaAbortTask(context.Background(), d.Host, id); e != nil {
+				w.Logger.Error(ctx, "Failed to abort task",
+					"host", d.Host,
+					"id", id,
+					"error", e,
+				)
+			}
+		}
+	}()
+
+	for {
+		if ctx.Err() != nil {
+			return ctx.Err()
+		}
+
+		task, err := w.Client.ScyllaWaitTask(ctx, d.Host, id, int64(w.Config.LongPollingTimeoutSeconds))
+		if err != nil {
+			return errors.Wrap(err, "wait for scylla task")
+		}
+		w.scyllaUpdateProgress(ctx, d, task)
+		switch scyllaclient.ScyllaTaskState(task.State) {
+		case scyllaclient.ScyllaTaskStateFailed:
+			return errors.Errorf("task error (%s): %s", id, task.Error)
+		case scyllaclient.ScyllaTaskStateDone:
+			return nil
+		}
+	}
+}
+
+func (w *worker) scyllaUpdateProgress(ctx context.Context, d snapshotDir, task *models.TaskStatus) {
+	p := d.Progress
+	p.StartedAt = nil
+	if t := time.Time(task.StartTime); !t.IsZero() {
+		p.StartedAt = &t
+	}
+	p.CompletedAt = nil
+	if t := time.Time(task.EndTime); !t.IsZero() {
+		p.CompletedAt = &t
+	}
+	p.Error = task.Error
+	p.Uploaded = int64(task.ProgressCompleted)
+	p.Skipped = d.SkippedBytesOffset
+	w.onRunProgress(ctx, p)
+}

pkg/service/backup/worker_upload.go

@@ -71,6 +71,12 @@ func (w *worker) uploadHost(ctx context.Context, h hostInfo) error {
 			err = errors.Wrap(w.deleteTableSnapshot(ctx, h, d), "delete table snapshot")
 		}()
 
+		if ok, err := w.useScyllaBackupAPI(h); err != nil {
+			return errors.Wrap(err, "check Scylla backup API support")
+		} else if ok {
+			return w.scyllaBackup(ctx, h, d)
+		}
+
 		// Check if we should attach to a previous job and wait for it to complete.
 		if attached, err := w.attachToJob(ctx, h, d); err != nil {
 			return errors.Wrap(err, "attach to the agent job")