admin/spray
1
0
Fork 0
mirror of https://github.com/chainreactors/spray.git synced 2025-09-15 11:40:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

修复check过于频繁的bug

Browse Source
This commit is contained in:
M09Ic 2022-11-11 10:37:30 +08:00
parent 3534a7b668
commit 9f6199c243
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
internal/pool.go
View File

@ -127,7 +127,7 @@ func NewPool(ctx context.Context, config *pkg.Config) (*Pool, error) {
case WordSource: case WordSource:
// 异步进行性能消耗较大的深度对比 // 异步进行性能消耗较大的深度对比
pool.tempCh <- bl pool.tempCh <- bl
pool.reqCount++
if pool.reqCount%pool.checkPeriod == 0 { if pool.reqCount%pool.checkPeriod == 0 {
pool.reqCount++ pool.reqCount++
go pool.check() go pool.check()
@ -236,7 +236,6 @@ Loop:
if u == "" { if u == "" {
continue continue
} }
p.reqCount++
p.wg.Add(1) p.wg.Add(1)
_ = p.pool.Invoke(newUnit(u, WordSource)) _ = p.pool.Invoke(newUnit(u, WordSource))
case <-ctx.Done(): case <-ctx.Done():
@ -323,7 +322,7 @@ func (p *Pool) PutToFuzzy(bl *pkg.Baseline) {
} }
func (p *Pool) resetFailed() { func (p *Pool) resetFailed() {
p.failedCount = 0 p.failedCount = 1
p.failedBaselines = nil p.failedBaselines = nil
} }