Browse Source

修复结束 Chome 的代码

Signed-off-by: allan716 <[email protected]>
allan716 4 years ago
parent
commit
a7eef1de0e
4 changed files with 18 additions and 34 deletions
  1. 1 0
      .gitignore
  2. 4 29
      internal/logic/sub_supplier/subhd/subhd.go
  3. 5 5
      internal/pkg/util.go
  4. 8 0
      internal/pkg/util_test.go

+ 1 - 0
.gitignore

@@ -24,3 +24,4 @@
 /TestData/sub_helper/test
 /settings.db
 /TestData/zips/test
+/internal/pkg/Logs

+ 4 - 29
internal/logic/sub_supplier/subhd/subhd.go

@@ -632,34 +632,6 @@ search:
 
 func (s Supplier) httpGet(inputUrl string) (string, error) {
 	s.reqParam.Referer = inputUrl
-	// TODO subhd 搜索限制太大,更换方式也差不多
-	// 是用本地的 Browser 还是远程的,推荐是远程的
-	//browser, err := rod_helper.NewBrowser(s.reqParam.HttpProxy)
-	//if err != nil {
-	//	return "", err
-	//}
-	//defer browser.Close()
-	//// 默认超时是 60s,如果是调试模式则是 5 min
-	//tt := common.HTMLTimeOut
-	//if s.reqParam.DebugMode == true {
-	//	tt = common.OneVideoProcessTimeOut
-	//}
-	//page, err := rod_helper.NewPageNavigate(browser, inputUrl, tt, 5)
-	//if err != nil {
-	//	return "", err
-	//}
-	//defer page.Close()
-	//page.MustSetUserAgent(&proto.NetworkSetUserAgentOverride{
-	//	UserAgent: pkg.RandomUserAgent(true),
-	//})
-	//err = page.WaitLoad()
-	//if err != nil {
-	//	return "", err
-	//}
-	//recvText, err := page.HTML()
-	//if err != nil {
-	//	return "", err
-	//}
 	httpClient := pkg.NewHttpClient(s.reqParam)
 	resp, err := httpClient.R().Get(inputUrl)
 	if err != nil {
@@ -669,9 +641,12 @@ func (s Supplier) httpGet(inputUrl string) (string, error) {
 	//搜索验证 点击继续搜索
 	if strings.Contains(recvText, "搜索验证") || strings.Contains(recvText, "搜索频率") {
 		s.log.Debugln("搜索验证 or 搜索频率 reload", inputUrl)
-		time.Sleep(pkg.RandomSecondDuration(5, 10))
+		// 每次搜索间隔在 30-40s
+		time.Sleep(pkg.RandomSecondDuration(30, 40))
 		return s.httpGet(inputUrl)
 	}
+	// 每次搜索间隔在 30-40s
+	time.Sleep(pkg.RandomSecondDuration(30, 40))
 	return recvText, nil
 }
 

+ 5 - 5
internal/pkg/util.go

@@ -396,22 +396,22 @@ func CopyTestData(srcDir string) (string, error) {
 func CloseChrome() {
 
 	cmdString := ""
+	var command *exec.Cmd
 	sysType := runtime.GOOS
 	if sysType == "linux" {
 		// LINUX系统
 		cmdString = "pkill chrome"
+		command = exec.Command("/bin/sh", "-c", cmdString)
 	}
 	if sysType == "windows" {
 		// windows系统
-		cmdString = "taskkill /F /im chromedriver.exe"
+		cmdString = "taskkill /F /im notepad.exe"
+		command = exec.Command("cmd.exe", "/c", cmdString)
 	}
-
-	if cmdString == "" {
+	if cmdString == "" || command == nil {
 		log_helper.GetLogger().Errorln("CloseChrome OS:", sysType)
 		return
 	}
-
-	command := exec.Command(cmdString)
 	err := command.Run()
 	if err != nil {
 		log_helper.GetLogger().Errorln("CloseChrome", err)

+ 8 - 0
internal/pkg/util_test.go

@@ -0,0 +1,8 @@
+package pkg
+
+import "testing"
+
+func TestCloseChrome(t *testing.T) {
+
+	CloseChrome()
+}