fix: ssh tunnel检测导致死锁问题调整

server/pkg/pool/cache_pool.go

@@ -2,6 +2,7 @@ package pool
 
 import (
 	"context"
+	"math/rand"
 	"mayfly-go/pkg/logx"
 	"mayfly-go/pkg/utils/stringx"
 	"sync"
@@ -45,20 +46,36 @@ func NewCachePool[T Conn](factory func() (T, error), opts ...Option[T]) *CachePo
 }
 
 // Get 获取连接（自动创建或复用缓存连接）
-func (p *CachePool[T]) Get(ctx context.Context) (T, error) {
+func (p *CachePool[T]) Get(ctx context.Context, opts ...GetOption) (T, error) {
 	var zero T
 
+	options := getOptions{updateLastActive: true} // 默认更新 lastActive
+	for _, apply := range opts {
+		apply(&options)
+	}
+
 	// 先尝试加读锁，仅用于查找可用连接
 	p.mu.RLock()
-	for _, entry := range p.cache {
-		if time.Since(entry.lastActive) <= p.config.IdleTimeout {
-			p.mu.RUnlock() // 找到后释放读锁
-			return entry.conn, nil
+	if len(p.cache) >= p.config.MaxConns {
+		keys := make([]string, 0, len(p.cache))
+		for k := range p.cache {
+			keys = append(keys, k)
 		}
+
+		randomKey := keys[rand.Intn(len(keys))]
+		entry := p.cache[randomKey]
+		conn := entry.conn
+
+		if options.updateLastActive {
+			// 更新最后活跃时间
+			entry.lastActive = time.Now()
+		}
+		p.mu.RUnlock()
+		return conn, nil
 	}
 	p.mu.RUnlock()
 
-	// 没有找到可用连接，升级为写锁进行清理和创建
+	// 没有找到可用连接，升级为写锁进行创建
 	p.mu.Lock()
 	defer p.mu.Unlock()
 
@@ -66,14 +83,13 @@ func (p *CachePool[T]) Get(ctx context.Context) (T, error) {
 		return zero, ErrPoolClosed
 	}
 
-	// 再次检查是否已有可用连接（防止并发创建）
-	for key, entry := range p.cache {
-		if time.Since(entry.lastActive) <= p.config.IdleTimeout {
-			entry.lastActive = time.Now()
-			return entry.conn, nil
-		}
-		// 清理超时连接
-		if !p.closeConn(key, entry, false) {
+	// 再次检查是否已创建（防止并发）
+	if len(p.cache) >= p.config.MaxConns {
+		for _, entry := range p.cache {
+			if options.updateLastActive {
+				// 更新最后活跃时间
+				entry.lastActive = time.Now()
+			}
 			return entry.conn, nil
 		}
 	}
@@ -206,7 +222,7 @@ func (p *CachePool[T]) cleanupIdle() {
 
 	cutoff := time.Now().Add(-p.config.IdleTimeout)
 	for key, entry := range p.cache {
-		if entry.lastActive.Before(cutoff) || entry.conn.Ping() != nil {
+		if entry.lastActive.Before(cutoff) || !p.ping(entry.conn) {
 			logx.Infof("cache pool - cleaning up idle connection, key: %s", key)
 			// 如果连接超时或不可用，则关闭连接
 			p.closeConn(key, entry, false)
@@ -214,6 +230,21 @@ func (p *CachePool[T]) cleanupIdle() {
 	}
 }
 
+func (p *CachePool[T]) ping(conn T) bool {
+	done := make(chan struct{})
+	var result bool
+	go func() {
+		result = conn.Ping() == nil
+		close(done)
+	}()
+	select {
+	case <-done:
+		return result
+	case <-time.After(2 * time.Second): // 设置超时
+		return false // 超时认为不可用
+	}
+}
+
 func (p *CachePool[T]) closeConn(key string, entry *cacheEntry[T], force bool) bool {
 	if !force {
 		// 如果不是强制关闭且有连接关闭回调，则调用回调

server/pkg/pool/chan_pool.go

@@ -74,12 +74,17 @@ func NewChannelPool[T Conn](factory func() (T, error), opts ...Option[T]) *ChanP
 	return p
 }
 
-func (p *ChanPool[T]) Get(ctx context.Context) (T, error) {
+func (p *ChanPool[T]) Get(ctx context.Context, opts ...GetOption) (T, error) {
 	connChan := make(chan T, 1)
 	errChan := make(chan error, 1)
 
+	options := getOptions{updateLastActive: true} // 默认更新 lastActive
+	for _, apply := range opts {
+		apply(&options)
+	}
+
 	go func() {
-		conn, err := p.get()
+		conn, err := p.get(options)
 		if err != nil {
 			errChan <- err
 		} else {
@@ -108,7 +113,7 @@ func (p *ChanPool[T]) Get(ctx context.Context) (T, error) {
 	}
 }
 
-func (p *ChanPool[T]) get() (T, error) {
+func (p *ChanPool[T]) get(opts getOptions) (T, error) {
 	// 检查连接池是否已关闭
 	p.mu.RLock()
 	if p.closed {
@@ -123,7 +128,9 @@ func (p *ChanPool[T]) get() (T, error) {
 	case wrapper := <-p.idleConns:
 		atomic.AddInt32(&p.stats.IdleConns, -1)
 		atomic.AddInt32(&p.stats.ActiveConns, 1)
-		wrapper.lastActive = time.Now()
+		if opts.updateLastActive {
+			wrapper.lastActive = time.Now()
+		}
 		return wrapper.conn, nil
 	default:
 		return p.createConn()

server/pkg/pool/config.go

@@ -62,3 +62,20 @@ func WithOnConnClose[T Conn](fn func(conn T) error) Option[T] {
 		c.OnConnClose = fn
 	}
 }
+
+/**** GetOption Config ****/
+
+// GetOption 用于配置 Get 的行为
+type GetOption func(*getOptions)
+
+// 控制 Get 行为的选项
+type getOptions struct {
+	updateLastActive bool // 是否更新 lastActive，默认 true
+}
+
+// WithNoUpdateLastActive 返回一个 Option，禁用更新 lastActive
+func WithNoUpdateLastActive() GetOption {
+	return func(o *getOptions) {
+		o.updateLastActive = false
+	}
+}

server/pkg/pool/pool.go

@@ -17,7 +17,7 @@ type Conn interface {
 // Pool 连接池接口
 type Pool[T Conn] interface {
 	// 核心方法
-	Get(ctx context.Context) (T, error)
+	Get(ctx context.Context, opts ...GetOption) (T, error)
 	Put(T) error
 	Close()