Function NewCheckPool

core/pool/checkpool.go:19–51

类似httpx的无状态, 无scope, 无并发池的检测模式

(ctx context.Context, config *Config)

// 类似httpx的无状态, 无scope, 无并发池的检测模式
func NewCheckPool(ctx context.Context, config *Config) (*CheckPool, error) {
	pctx, cancel := context.WithCancel(ctx)
	config.ClientType = ihttp.STANDARD
	pool := &CheckPool{
		BasePool: &BasePool{
			Config:    config,
			Statistor: pkg.NewStatistor(""),
			ctx:       pctx,
			Cancel:    cancel,
			client: ihttp.NewClient(&ihttp.ClientConfig{
				Thread:      config.Thread,
				Type:        config.ClientType,
				Timeout:     config.Timeout,
				ProxyClient: config.ProxyClient,
			}),
			wg:          &sync.WaitGroup{},
			additionCh:  make(chan *Unit, config.Thread*10),
			closeCh:     make(chan struct{}),
			processCh:   make(chan *baseline.Baseline, config.Thread*2),
			handlerDone: make(chan struct{}),
		},
	}
	pool.Request.Headers.Set("Connection", "close")
	p, err := ants.NewPoolWithFunc(config.Thread, pool.Invoke)
	if err != nil {
		cancel()
		return nil, fmt.Errorf("create check pool: %w", err)
	}

	pool.Pool = p
	go pool.Handler()
	return pool, nil
}

type CheckPool struct {
	*BasePool