maxiaoshan 3 лет назад
Родитель
Сommit
7cb721d8fa
3 измененных файлов с 13 добавлено и 13 удалено
  1. 6 6
      src/spider/handler.go
  2. 5 5
      src/spider/script.go
  3. 2 2
      src/spider/spider.go

+ 6 - 6
src/spider/handler.go

@@ -539,7 +539,7 @@ func UpdateSpiderByCodeState(code, state string) (bool, error) {
 					}
 					sp2.MUserName = v["modifyuser"]
 					sp2.MUserEmail = v["modifyemail"]
-					sp2.LoadScript(k, sp2.ScriptFile, true) //更新上架,重载脚本
+					sp2.LoadScript(sp2.Name, sp2.Channel, sp2.MUserName, k, sp2.ScriptFile, true) //更新上架,重载脚本
 					Allspiders2.Store(k, sp2)
 					// up = true
 					// err = nil
@@ -690,9 +690,9 @@ func UpdateSpiderByCodeState(code, state string) (bool, error) {
 						}
 						sp2.MUserName = v["modifyuser"]
 						sp2.MUserEmail = v["modifyemail"]
-						sp2.LoadScript(k, sp2.ScriptFile, true) //更新上架,重载脚本
-						Allspiders2.Store(k, sp2)               //重载后放入集合
-						UpdateHighListDataByCode(k)             //爬虫更新上架后,重置数据state=0
+						sp2.LoadScript(sp2.Name, sp2.Channel, sp2.MUserName, k, sp2.ScriptFile, true) //更新上架,重载脚本
+						Allspiders2.Store(k, sp2)                                                     //重载后放入集合
+						UpdateHighListDataByCode(k)                                                   //爬虫更新上架后,重置数据state=0
 						// up = true
 						// err = nil
 						logger.Info("Allspiders2上架重载脚本", sp2.Code)
@@ -892,7 +892,7 @@ func ReloadSpiderFile() {
 func NewSpider_New(code, luafile string, newstate bool) (*Spider, string) {
 	defer mu.Catch()
 	spider := &Spider{}
-	err := spider.LoadScript(code, luafile, newstate)
+	err := spider.LoadScript(spider.Name, spider.Channel, spider.MUserName, code, luafile, newstate)
 	if err != "" {
 		return nil, err
 	}
@@ -953,7 +953,7 @@ func NewSpider_New(code, luafile string, newstate bool) (*Spider, string) {
 func NewSpider(code, luafile string) (*Spider, string) {
 	defer mu.Catch()
 	spider := &Spider{}
-	err := spider.LoadScript(code, luafile, true)
+	err := spider.LoadScript(spider.Name, spider.Channel, spider.MUserName, code, luafile, true)
 	if err != "" {
 		return nil, err
 	}

+ 5 - 5
src/spider/script.go

@@ -70,7 +70,7 @@ func init() {
 var TimeSleepChan = make(chan bool, 1)
 
 //加载文件
-func (s *Script) LoadScript(code, script_file string, newstate bool) string {
+func (s *Script) LoadScript(site, channel, user, code, script_file string, newstate bool) string {
 	defer mu.Catch()
 	s.SCode = code
 	s.ScriptFile = script_file
@@ -236,9 +236,9 @@ func (s *Script) LoadScript(code, script_file string, newstate bool) string {
 		content := S.ToString(-1)
 		ret := s.L.NewTable()
 		util.FindListHtml(gpath, content, ret)
-		if ret.Len() > 0 {
-			UpdateHeart("", "", code, "", "findlist") //记录列表页实际采集数据量心跳
-		}
+		//if ret.Len() > 0 {
+		//	UpdateHeart(site, channel, code, user, "findlist") //记录列表页实际采集数据量心跳
+		//}
 		S.Push(ret)
 		return 1
 	}))
@@ -247,7 +247,7 @@ func (s *Script) LoadScript(code, script_file string, newstate bool) string {
 		table := S.ToTable(-1)
 		list := util.TableToMap(table)
 		if len(list) > 0 {
-			UpdateHeart("", "", code, "", "findlist") //记录列表页实际采集数据量心跳
+			UpdateHeart(site, channel, code, user, "findlist") //记录列表页实际采集数据量心跳
 		}
 		return 1
 	}))

+ 2 - 2
src/spider/spider.go

@@ -162,7 +162,7 @@ func (s *Spider) ExecJob(reload bool) {
 		}
 	}()
 	if reload && util.Config.Working == 0 { //高效模式,轮询调度时重载脚本
-		s.LoadScript(s.Code, s.ScriptFile, true)
+		s.LoadScript(s.Name, s.Channel, s.MUserName, s.Code, s.ScriptFile, true)
 	}
 	logger.Debug(s.Code, s.Name, "频率:", s.SpiderRunRate, ",", s.Timeout)
 	s.LastDowncount = 0
@@ -906,7 +906,7 @@ func (s *Spider) DownloadHighDetail() {
 					Mgo.Update("spider_highlistdata", query, set, false, false)
 				}
 				//重载spider
-				s.LoadScript(s.Code, s.ScriptFile, true)
+				s.LoadScript(s.Name, s.Channel, s.MUserName, s.Code, s.ScriptFile, true)
 			} else { //没有数据
 				time.Sleep(2 * time.Minute)
 			}