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

clean code

Browse Source
This commit is contained in:
M09Ic 2024-07-14 04:28:08 +08:00
parent f942c9c1b3
commit 3e419a52a5
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
internal/runner.go
View File

@ -154,9 +154,7 @@ func (r *Runner) Prepare(ctx context.Context) error {
r.Pools, err = ants.NewPoolWithFunc(r.PoolSize, func(i interface{}) {
t := i.(*Task)
if t.origin != nil && t.origin.End == t.origin.Total {
if r.StatFile != nil {
r.StatFile.SafeWrite(t.origin.Json())
}
r.saveStat(t.origin.Json())
r.Done()
return
}
@ -233,9 +231,7 @@ Loop:
if len(r.taskCh) > 0 {
for t := range r.taskCh {
stat := pkg.NewStatistor(t.baseUrl)
if r.StatFile != nil {
r.StatFile.SafeWrite(stat.Json())
}
r.saveStat(stat.Json())
}
}
if r.StatFile != nil {
@ -346,8 +342,12 @@ func (r *Runner) PrintStat(pool *pool.BrutePool) {
}
}
r.saveStat(pool.Statistor.Json())
}
func (r *Runner) saveStat(content string) {
if r.StatFile != nil {
r.StatFile.SafeWrite(pool.Statistor.Json())
r.StatFile.SafeWrite(content)
r.StatFile.SafeSync()
}
}