Add -gops option to init gops agent

README.md

@@ -23,6 +23,8 @@ Usage: ./s5cmd [OPTION]... [COMMAND [PARAMS...]]
         Multipart chunk size in MB for uploads (default 5, auto calculated)
   -f string
         Commands-file or - for stdin
+  -gops
+        Initialize gops agent
   -numworkers int
         Number of worker goroutines. Negative numbers mean multiples of runtime.NumCPU (default 256)
   -r int

s5cmd.go

@@ -14,6 +14,7 @@ import (
 	"time"
 
 	"github.com/aws/aws-sdk-go/service/s3/s3manager"
+	"github.com/google/gops/agent"
 	"github.com/peakgames/s5cmd/core"
 	"github.com/peakgames/s5cmd/stats"
 	"github.com/peakgames/s5cmd/version"
@@ -63,6 +64,7 @@ func main() {
 	flag.IntVar(&retries, "r", 10, "Retry S3 operations N times before failing")
 	printStats := flag.Bool("stats", false, "Always print stats")
 	version := flag.Bool("version", false, "Prints current version")
+	gops := flag.Bool("gops", false, "Initialize gops agent")
 
 	flag.Usage = func() {
 		printUsageLine()
@@ -71,6 +73,12 @@ func main() {
 	}
 	flag.Parse()
 
+	if *gops {
+		if err := agent.Listen(&agent.Options{NoShutdownCleanup: true}); err != nil {
+			log.Fatal("-ERR", err)
+		}
+	}
+
 	if *version {
 		fmt.Printf("s5cmd version %s", GitSummary)
 		if GitBranch != "" {
@@ -91,11 +99,9 @@ func main() {
 	cmd := strings.Join(flag.Args(), " ")
 	if cmd != "" && cmdFile != "" {
 		log.Fatal("-ERR Only specify -f or command, not both")
-		os.Exit(1)
 	}
 	if (cmd == "" && cmdFile == "") || numWorkers == 0 || multipartChunkSize < 1 || retries < 0 {
 		log.Fatal("-ERR Please specify all arguments.")
-		os.Exit(1)
 	}
 
 	multipartChunkSizeBytes := int64(multipartChunkSize * int(bytesInMb))