运行参数解析遗漏获取抢购连接也调整为多线程获取

walternie · Jan 7, 2021 · 3e6a9b5 · 3e6a9b5
1 parent f991b09
commit 3e6a9b5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -16,6 +16,10 @@ var num = flag.Int("num", 2, "茅台商品ID")
 var works = flag.Int("works", 7, "并发数")
 var start = flag.String("time", "09:59:59", "开始时间---不带日期")
 
+func init() {
+	flag.Parse()
+}
+
 func main() {
 	var err error
 	execPath := ""

diff --git a/secKill/jdSecKill.go b/secKill/jdSecKill.go
@@ -216,9 +216,9 @@ func (jsk *jdSecKill) Run() error {
 		jsk.WaitStart(),
 		chromedp.ActionFunc(func(ctx context.Context) error {
 			//提取抢购连接
-			jsk.FetchSecKillUrl()
 			for _, v := range jsk.bWorksCtx {
 				go func(ctx2 context.Context) {
+					jsk.FetchSecKillUrl()
 					SecKillRE:
 					//请求抢购连接，提交订单
 					err := jsk.ReqSubmitSecKillOrder(ctx2)
@@ -328,10 +328,10 @@ func (jsk *jdSecKill) GetEidAndFp() chromedp.ActionFunc {
 func (jsk *jdSecKill) FetchSecKillUrl() {
 	logs.PrintlnInfo("开始获取抢购连接.....")
 	for {
-		jsk.SecKillUrl = jsk.GetSecKillUrl()
 		if jsk.SecKillUrl != "" {
 			break
 		}
+		jsk.SecKillUrl = jsk.GetSecKillUrl()
 		logs.PrintlnWarning("抢购链接获取失败.....正在重试")
 	}
 	jsk.SecKillUrl = "https:" + strings.TrimPrefix(jsk.SecKillUrl, "https:")