admin/spray
1
0
Fork 0
mirror of https://github.com/chainreactors/spray.git synced 2025-05-08 11:36:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

优化index的init

Browse Source
This commit is contained in:
M09Ic 2023-01-10 23:55:03 +08:00
parent 880b11e860
commit 75236c7708
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
internal/pool.go
View File

@ -81,6 +81,12 @@ func NewPool(ctx context.Context, config *pkg.Config) (*Pool, error) {
pool.Statistor.Counts[bl.Status] = 1 pool.Statistor.Counts[bl.Status] = 1
} }
if _, ok := pool.Statistor.Sources[bl.Source]; ok {
pool.Statistor.Sources[bl.Source]++
} else {
pool.Statistor.Sources[bl.Source] = 1
}
var params map[string]interface{} var params map[string]interface{}
if pool.MatchExpr != nil || pool.FilterExpr != nil || pool.RecuExpr != nil { if pool.MatchExpr != nil || pool.FilterExpr != nil || pool.RecuExpr != nil {
params = map[string]interface{}{ params = map[string]interface{}{
@ -172,7 +178,7 @@ type Pool struct {
func (pool *Pool) Init() error { func (pool *Pool) Init() error {
// 分成两步是为了避免闭包的线程安全问题 // 分成两步是为了避免闭包的线程安全问题
pool.initwg.Add(2) pool.initwg.Add(2)
pool.reqPool.Invoke(newUnit("", InitIndexSource)) pool.reqPool.Invoke(newUnit(pool.url.Path, InitIndexSource))
pool.reqPool.Invoke(newUnit(pool.safePath(pkg.RandPath()), InitRandomSource)) pool.reqPool.Invoke(newUnit(pool.safePath(pkg.RandPath()), InitRandomSource))
pool.initwg.Wait() pool.initwg.Wait()
if pool.index.ErrString != "" { if pool.index.ErrString != "" {