accelerate postgres restore with jobs

agence-webup · Feb 15, 2024 · a516766 · a516766
1 parent ed11223
commit a516766
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/actions/restore.go b/actions/restore.go
@@ -9,6 +9,7 @@ import (
  "os"
  "path"
  "path/filepath"
+ "runtime"
  "strings"
 
  "github.com/Songmu/prompter"
@@ -295,7 +296,7 @@ func restorePostgres(ctx domain.ExecutionContext, containerID string, dumpFilena
  ext := filepath.Ext(dumpFilename)
  database := strings.Replace(dumpFilename, ext, "", 1)
 
- cmd := domain.NewCommand([]string{"docker", "exec", "-i", "-e", fmt.Sprintf("PGPASSWORD=\"%s\"", password), containerID, "pg_restore", fmt.Sprintf("--username=%s", user), "-d", database, "-c"}, verbose)
+ cmd := domain.NewCommand([]string{"docker", "exec", "-i", "-e", fmt.Sprintf("PGPASSWORD=\"%s\"", password), containerID, "pg_restore", fmt.Sprintf("--username=%s", user), "-d", database, "-c", fmt.Sprintf("--jobs=%d", runtime.NumCPU())}, verbose)
  cmd.ExecuteWithStdin(postgresDumpReader)
 }