Method DialContext

common/proxy/manager.go:310–368

(ctx context.Context, network, address string)

}

func (s *socks5Dialer) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
	start := time.Now()
	atomic.AddInt64(&s.stats.TotalConnections, 1)

	// 创建一个带超时的上下文
	dialCtx, cancel := context.WithTimeout(ctx, s.config.Timeout)
	defer cancel()

	// 使用goroutine处理拨号，以支持取消
	connChan := make(chan struct {
		conn net.Conn
		err  error
	}, 1)

	go func() {
		conn, err := s.dialer.Dial(network, address)
		select {
		case <-dialCtx.Done():
			if conn != nil {
				_ = conn.Close() // context取消路径，Close错误可忽略
			}
		case connChan <- struct {
			conn net.Conn
			err  error
		}{conn, err}:
		}
	}()

	select {
	case <-dialCtx.Done():
		atomic.AddInt64(&s.stats.FailedConnections, 1)
		s.stats.mu.Lock()
		s.stats.LastError = dialCtx.Err().Error()
		s.stats.mu.Unlock()
		return nil, NewProxyError(ErrTypeTimeout, ErrMsgSOCKS5ConnTimeout, ErrCodeSOCKS5ConnTimeout, dialCtx.Err())
	case result := <-connChan:
		duration := time.Since(start)

		s.stats.mu.Lock()
		s.stats.LastConnectTime = start
		s.stats.mu.Unlock()

		if result.err != nil {
			atomic.AddInt64(&s.stats.FailedConnections, 1)
			s.stats.mu.Lock()
			s.stats.LastError = result.err.Error()
			s.stats.mu.Unlock()
			return nil, NewProxyError(ErrTypeConnection, ErrMsgSOCKS5ConnFailed, ErrCodeSOCKS5ConnFailed, result.err)
		}

		atomic.AddInt64(&s.stats.ActiveConnections, 1)
		s.updateAverageConnectTime(duration)

		return &trackedConn{
			Conn:  result.conn,
			stats: s.stats,
		}, nil
	}