admin/spray
1
0
Fork 0
mirror of https://github.com/chainreactors/spray.git synced 2025-05-07 19:16:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

去掉不必要的锁

Browse Source
This commit is contained in:
M09Ic 2022-11-23 10:59:15 +08:00
parent cc512829ec
commit 922ed78611
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
internal/pool.go
View File

@ -69,7 +69,6 @@ func NewPool(ctx context.Context, config *pkg.Config) (*Pool, error) {
} }
} }
var mutex sync.Mutex
p, _ := ants.NewPoolWithFunc(config.Thread, func(i interface{}) { p, _ := ants.NewPoolWithFunc(config.Thread, func(i interface{}) {
pool.Statistor.ReqNumber++ pool.Statistor.ReqNumber++
unit := i.(*Unit) unit := i.(*Unit)
@ -91,13 +90,6 @@ func NewPool(ctx context.Context, config *pkg.Config) (*Pool, error) {
bl = &pkg.Baseline{Url: pool.BaseURL + unit.path, IsValid: false, ErrString: reqerr.Error(), Reason: ErrRequestFailed.Error()} bl = &pkg.Baseline{Url: pool.BaseURL + unit.path, IsValid: false, ErrString: reqerr.Error(), Reason: ErrRequestFailed.Error()}
pool.failedBaselines = append(pool.failedBaselines, bl) pool.failedBaselines = append(pool.failedBaselines, bl)
} else { } else {
mutex.Lock()
if _, ok := pool.Statistor.Counts[resp.StatusCode()]; ok {
pool.Statistor.Counts[resp.StatusCode()]++
} else {
pool.Statistor.Counts[resp.StatusCode()] = 1
}
mutex.Unlock()
if unit.source != WordSource { if unit.source != WordSource {
bl = pkg.NewBaseline(req.URI(), req.Host(), resp) bl = pkg.NewBaseline(req.URI(), req.Host(), resp)
} else { } else {
@ -162,6 +154,11 @@ func NewPool(ctx context.Context, config *pkg.Config) (*Pool, error) {
pool.pool = p pool.pool = p
go func() { go func() {
for bl := range pool.tempCh { for bl := range pool.tempCh {
if _, ok := pool.Statistor.Counts[bl.Status]; ok {
pool.Statistor.Counts[bl.Status]++
} else {
pool.Statistor.Counts[bl.Status] = 1
}
var status bool var status bool
if pool.MatchExpr != nil { if pool.MatchExpr != nil {
if pool.CompareWithExpr(pool.MatchExpr, bl) { if pool.CompareWithExpr(pool.MatchExpr, bl) {