feat: add threads parameter to openGauss restore subcommand (#465)

* feat: add threads parameter to restore command * feat: add threadsnum to agent server * feat: add threadsNum to cli * chore: update opengauss mock
apache · Nov 16, 2023 · 3d87f11 · 3d87f11
1 parent d186b41
commit 3d87f11
Show file tree

Hide file tree

Showing 6 changed files with 123 additions and 121 deletions.
diff --git a/pitr/agent/internal/handler/restore.go b/pitr/agent/internal/handler/restore.go
@@ -78,7 +78,7 @@ func Restore(ctx *fiber.Ctx) (err error) {
 	}()
 
 	// restore data from backup
-	if err = pkg.OG.Restore(in.DnBackupPath, in.Instance, in.DnBackupID); err != nil {
+	if err = pkg.OG.Restore(in.DnBackupPath, in.Instance, in.DnBackupID, in.DnThreadsNum); err != nil {
 		efmt := "pkg.OG.Restore failure[path=%s,instance=%s,backupID=%s], err wrap: %w"
 		err = fmt.Errorf(efmt, in.DnBackupPath, in.Instance, in.DnBackupID, err)
 		status = "restore failure"

diff --git a/pitr/agent/internal/handler/view/restore.go b/pitr/agent/internal/handler/view/restore.go
@@ -27,6 +27,7 @@ type RestoreIn struct {
 	Instance     string `json:"instance"`
 	DnBackupPath string `json:"dn_backup_path"`
 	DnBackupID   string `json:"dn_backup_id"`
+	DnThreadsNum uint8  `json:"dn_threads_num"`
 }
 
 //nolint:dupl

diff --git a/pitr/agent/internal/pkg/mocks/opengauss.go b/pitr/agent/internal/pkg/mocks/opengauss.go