allow concurrency to be specified

Author: ybirader

cli.go

@@ -10,6 +10,7 @@ import (
 type CLI struct {
 	ArchivePath string
 	Files       []string
+	Concurrency int
 }
 
 func (c *CLI) Archive() error {
@@ -19,7 +20,7 @@ func (c *CLI) Archive() error {
 	}
 	defer archive.Close()
 
-	archiver, err := NewArchiver(archive)
+	archiver, err := NewArchiver(archive, Concurrency(c.Concurrency))
 	if err != nil {
 		return errors.Wrap(err, "ERROR: could not create archiver")
 	}

cli_test.go

@@ -3,6 +3,7 @@ package pzip_test
 import (
 	"os"
 	"path/filepath"
+	"runtime"
 	"testing"
 
 	"github.com/alecthomas/assert/v2"
@@ -18,7 +19,7 @@ func TestCLI(t *testing.T) {
 		archivePath := "testdata/archive.zip"
 		defer os.RemoveAll(archivePath)
 
-		cli := pzip.CLI{archivePath, files}
+		cli := pzip.CLI{archivePath, files, runtime.GOMAXPROCS(0)}
 		cli.Archive()
 
 		archiveReader := testutils.GetArchiveReader(t, archivePath)
@@ -32,7 +33,7 @@ func BenchmarkCLI(b *testing.B) {
 	dirPath := filepath.Join(benchmarkRoot, "minibench")
 	archivePath := filepath.Join(benchmarkRoot, "minibench.zip")
 
-	cli := pzip.CLI{archivePath, []string{dirPath}}
+	cli := pzip.CLI{archivePath, []string{dirPath}, runtime.GOMAXPROCS(0)}
 
 	b.ReportAllocs()
 	b.ResetTimer()

cmd/cli/main.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"log"
 	"os"
+	"runtime"
 
 	"github.com/pzip"
 )
@@ -18,6 +19,9 @@ func main() {
 		flag.PrintDefaults()
 	}
 
+	var concurrency int
+	flag.IntVar(&concurrency, "concurrency", runtime.GOMAXPROCS(0), "allow up to n compression routines")
+
 	flag.Parse()
 
 	args := flag.Args()
@@ -30,9 +34,7 @@ func main() {
 		return
 	}
 
-	archivePath := os.Args[1]
-
-	cli := pzip.CLI{ArchivePath: archivePath, Files: os.Args[2:]}
+	cli := pzip.CLI{ArchivePath: args[0], Files: args[1:], Concurrency: concurrency}
 	err := cli.Archive()
 
 	if err != nil {