add pprof

commands.go

@@ -303,8 +303,8 @@ func NewGetCommand() cli.Command {
 					PrintErrorAndExit("get %s: parse mtime: %v", upPath, err)
 				}
 			}
-			if c.Int("w") > 10 || c.Int("w") < 1 {
-				PrintErrorAndExit("max concurrent threads must between (1 - 10)")
+			if c.Int("w") > 50 || c.Int("w") < 1 {
+				PrintErrorAndExit("max concurrent threads must between (1 - 50)")
 			}
 			if mc.Start != "" || mc.End != "" {
 				session.GetStartBetweenEndFiles(upPath, localPath, mc, c.Int("w"))

session.go

@@ -165,7 +165,7 @@ func (sess *Session) FormatUpInfo(upInfo *upyun.FileInfo) string {
 
 func (sess *Session) Init() error {
 	sess.scores = make(map[int]int)
-	PrintOnlyVerbose("hosts %v\n", sess.Hosts)
+	// PrintOnlyVerbose("hosts %v\n", sess.Hosts)
 	sess.updriver = upyun.NewUpYun(&upyun.UpYunConfig{
 		Bucket:    sess.Bucket,
 		Operator:  sess.Operator,
@@ -336,7 +336,7 @@ func (sess *Session) getDir(upPath, localPath string, match *MatchConfig, worker
 						}
 					}
 					if e != nil {
-						return
+						continue
 					}
 				}
 			}

upx.go

@@ -5,6 +5,9 @@ import (
 	"runtime"
 	"time"
 
+	"net/http"
+	_ "net/http/pprof"
+
 	"github.com/urfave/cli"
 )
 
@@ -23,6 +26,7 @@ func CreateUpxApp() *cli.App {
 	app.Flags = []cli.Flag{
 		cli.BoolFlag{Name: "quiet, q", Usage: "not verbose"},
 		cli.StringFlag{Name: "auth", Usage: "auth string"},
+		cli.StringFlag{Name: "pprof-addr", Usage: "enable pprof"},
 	}
 	app.Before = func(c *cli.Context) error {
 		if c.Bool("q") {
@@ -34,6 +38,11 @@ func CreateUpxApp() *cli.App {
 				PrintErrorAndExit("%s: invalid auth string", c.Command.FullName())
 			}
 		}
+		if c.String("pprof-addr") != "" {
+			go func() {
+				http.ListenAndServe(c.String("pprof-addr"), nil)
+			}()
+		}
 		return nil
 	}
 	app.Commands = []cli.Command{