fix: 移除隧道连接时检测是否正在使用

server/pkg/pool/config.go

@@ -8,48 +8,57 @@ import (
 var ErrPoolClosed = errors.New("pool is closed")
 
 // PoolConfig 连接池配置
-type PoolConfig struct {
+type PoolConfig[T Conn] struct {
 	MaxConns            int           // 最大连接数
 	IdleTimeout         time.Duration // 空闲连接超时时间
 	WaitTimeout         time.Duration // 获取连接超时时间
 	HealthCheckInterval time.Duration // 健康检查间隔
 	OnPoolClose         func() error  // 连接池关闭时的回调
+
+	OnConnClose func(conn T) error // 连接关闭时的回调,若err != nil则不关闭连接
 }
 
 // Option 函数类型，用于配置 Pool
-type Option func(*PoolConfig)
+type Option[T Conn] func(*PoolConfig[T])
 
 // WithMaxConns 设置最大连接数
-func WithMaxConns(maxConns int) Option {
-	return func(c *PoolConfig) {
+func WithMaxConns[T Conn](maxConns int) Option[T] {
+	return func(c *PoolConfig[T]) {
 		c.MaxConns = maxConns
 	}
 }
 
 // WithIdleTimeout 设置空闲超时
-func WithIdleTimeout(timeout time.Duration) Option {
-	return func(c *PoolConfig) {
+func WithIdleTimeout[T Conn](timeout time.Duration) Option[T] {
+	return func(c *PoolConfig[T]) {
 		c.IdleTimeout = timeout
 	}
 }
 
 // WithWaitTimeout 设置等待超时
-func WithWaitTimeout(timeout time.Duration) Option {
-	return func(c *PoolConfig) {
+func WithWaitTimeout[T Conn](timeout time.Duration) Option[T] {
+	return func(c *PoolConfig[T]) {
 		c.WaitTimeout = timeout
 	}
 }
 
 // WithHealthCheckInterval 设置健康检查间隔
-func WithHealthCheckInterval(interval time.Duration) Option {
-	return func(c *PoolConfig) {
+func WithHealthCheckInterval[T Conn](interval time.Duration) Option[T] {
+	return func(c *PoolConfig[T]) {
 		c.HealthCheckInterval = interval
 	}
 }
 
 // WithOnPoolClose 设置连接池关闭回调
-func WithOnPoolClose(fn func() error) Option {
-	return func(c *PoolConfig) {
+func WithOnPoolClose[T Conn](fn func() error) Option[T] {
+	return func(c *PoolConfig[T]) {
 		c.OnPoolClose = fn
 	}
 }
+
+// WithOnConnClose 设置连接关闭回调, 若返回的错误不为nil，则不关闭连接
+func WithOnConnClose[T Conn](fn func(conn T) error) Option[T] {
+	return func(c *PoolConfig[T]) {
+		c.OnConnClose = fn
+	}
+}