소스 검색

修复 UseUnderDocker 忘记传递值的问题···

Signed-off-by: 716 <[email protected]>
716 4 년 전
부모
커밋
674b72b2fa
1개의 변경된 파일15개의 추가작업 그리고 5개의 파일을 삭제
  1. 15 5
      downloader.go

+ 15 - 5
downloader.go

@@ -88,11 +88,21 @@ func (d Downloader) DownloadSub(dir string) error {
 		return err
 	}
 	// 构建每个字幕站点下载者的实例
-	subSupplierHub := sub_supplier.NewSubSupplierHub(shooter.NewSupplier(d.reqParam),
-									subhd.NewSupplier(d.reqParam),
-									xunlei.NewSupplier(d.reqParam),
-									zimuku.NewSupplier(d.reqParam),
-	)
+	var subSupplierHub *sub_supplier.SubSupplierHub
+	// 因为在 docker 下 rod 的模拟浏览器点击无效,所以···
+	if d.reqParam.UseUnderDocker == true {
+		subSupplierHub = sub_supplier.NewSubSupplierHub(shooter.NewSupplier(d.reqParam),
+			xunlei.NewSupplier(d.reqParam),
+			zimuku.NewSupplier(d.reqParam),
+		)
+	} else {
+		subSupplierHub = sub_supplier.NewSubSupplierHub(shooter.NewSupplier(d.reqParam),
+			subhd.NewSupplier(d.reqParam),
+			xunlei.NewSupplier(d.reqParam),
+			zimuku.NewSupplier(d.reqParam),
+		)
+	}
+
 	// TODO 后续再改为每个视频以上的流程都是一个 channel 来做,并且需要控制在一个并发量之下(很可能没必要,毕竟要在弱鸡机器上挂机用的)
 	// 一个视频文件同时多个站点查询,阻塞完毕后,在进行下一个
 	for i, oneVideoFullPath := range nowVideoList {