pass MaxProcs explicitly to mongodump/mongorestore

Author: defbin

pbm/snapshot/backup.go

@@ -3,6 +3,7 @@ package snapshot
 import (
 	"io"
 	"log"
+	"runtime"
 	"time"
 
 	"github.com/mongodb/mongo-tools/common/archive"
@@ -45,6 +46,9 @@ func NewBackup(curi string, maxParallelColls int, d, c string) (*backuper, error
 		}
 	}
 
+	// mongodump calls runtime.GOMAXPROCS(MaxProcs).
+	opts.MaxProcs = runtime.GOMAXPROCS(0)
+
 	if maxParallelColls < 1 {
 		maxParallelColls = 1
 	}

pbm/snapshot/restore.go

@@ -2,6 +2,7 @@ package snapshot
 
 import (
 	"io"
+	"runtime"
 
 	"github.com/mongodb/mongo-tools/common/options"
 	"github.com/mongodb/mongo-tools/mongorestore"
@@ -99,6 +100,8 @@ func NewRestore(uri string, cfg *config.Config, numParallelColls int) (io.Reader
 	mopts.NSOptions = &mongorestore.NSOptions{
 		NSExclude: ExcludeFromRestore,
 	}
+	// mongorestore calls runtime.GOMAXPROCS(MaxProcs).
+	mopts.MaxProcs = runtime.GOMAXPROCS(0)
 
 	mr, err := mongorestore.New(mopts)
 	if err != nil {