overflow_backup_2018_0455/crawler_go
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
jackdaw@loafle.com 2017-04-11 12:20:05 +09:00
parent 031ba9e162
commit 74b626cd9d
1 changed files with 45 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
socket_health_crawler.go
View File

@ -9,6 +9,11 @@ import (
type SocketHeahthCrawler struct { type SocketHeahthCrawler struct {
CrawlerImpl CrawlerImpl
m matcher.Matcher
}
func (s *SocketHeahthCrawler) SetMatcher(m matcher.Matcher) {
s.m = m
} }
func (s *SocketHeahthCrawler) getConnection(params map[string]interface{}) (net.Conn, error) { func (s *SocketHeahthCrawler) getConnection(params map[string]interface{}) (net.Conn, error) {
@ -44,38 +49,57 @@ func (s *SocketHeahthCrawler) getConnection(params map[string]interface{}) (net.
} }
} }
func (s *SocketHeahthCrawler) CheckHeahth(params map[string]interface{}, mat matcher.Matcher) (bool, error) { func (s *SocketHeahthCrawler) CheckHeahth(params map[string]interface{}) (bool, error) {
conn, err := s.getConnection(params) conn, err := s.getConnection(params)
if err != nil { if err != nil {
return false, err return false, err
} }
defer conn.Close() defer conn.Close()
if mat.IsPrePacket() == true { if s.m.IsPrePacket() == true {
bytes := make([]byte, 1024) bytes := make([]byte, 1024)
n, _ := conn.Read(bytes) n, _ := conn.Read(bytes)
p := packet.NewPacket(bytes, n) p := packet.NewPacket(bytes, n)
if mat.Match(0, p, nil) == false { if s.m.Match(0, p, nil) == false {
return false, nil return false, nil
} else {
for i := 1 ; i < s.m.PacketCount(); i++ {
pack := s.m.Packet(i)
conn.Write(pack.Buffer)
bytes := make([]byte, 1024)
n, _ := conn.Read(bytes)
if s.m.IsNoResponse(i) == true { // empty last response
break
}
p := packet.NewPacket(bytes, n)
if s.m.Match(i, p, nil) == false {
return false, nil
}
}
}
} else {
for i := 0 ; i < s.m.PacketCount(); i++ {
pack := s.m.Packet(i)
conn.Write(pack.Buffer)
bytes := make([]byte, 1024)
n, _ := conn.Read(bytes)
if s.m.IsNoResponse(i) == true { // empty last response
break
}
p := packet.NewPacket(bytes, n)
if s.m.Match(i, p, nil) == false {
return false, nil
}
} }
} }
for i := 0; i < mat.PacketCount(); i++ {
pack := mat.Packet(i)
conn.Write(pack.Buffer)
bytes := make([]byte, 1024)
n, _ := conn.Read(bytes)
if mat.IsNoResponse(i) == true { // empty last response
break
}
p := packet.NewPacket(bytes, n)
if mat.Match(i, p, nil) == false {
return false, nil
}
}
return true, nil return true, nil
} }