feat: export LongPool Config

Author: ppzqh

client/option_test.go

@@ -44,6 +44,9 @@ import (
 	"github.com/cloudwego/kitex/pkg/http"
 	"github.com/cloudwego/kitex/pkg/loadbalance"
 	"github.com/cloudwego/kitex/pkg/proxy"
+	connpool2 "github.com/cloudwego/kitex/pkg/remote/connpool"
+	"github.com/cloudwego/kitex/pkg/remote/trans/gonet"
+	"github.com/cloudwego/kitex/pkg/remote/trans/netpoll"
 	"github.com/cloudwego/kitex/pkg/remote/trans/nphttp2/grpc"
 	"github.com/cloudwego/kitex/pkg/retry"
 	"github.com/cloudwego/kitex/pkg/rpcinfo"
@@ -764,6 +767,20 @@ func TestTailOption(t *testing.T) {
 	test.Assert(t, opts.RemoteOpt.Dialer != nil)
 }
 
+func TestGonetOption(t *testing.T) {
+	// gonet
+	opt := client.NewOptions([]Option{WithDialer(gonet.NewDialer()), WithLongConnection(connpool.IdleConfig{MaxIdlePerAddress: 10})})
+	d := opt.RemoteOpt.ConnPool.(*connpool2.LongPool)
+	pcfg := d.Config()
+	test.Assert(t, pcfg.Enable)
+
+	// netpoll
+	opt = client.NewOptions([]Option{WithDialer(netpoll.NewDialer()), WithLongConnection(connpool.IdleConfig{MaxIdlePerAddress: 10})})
+	d = opt.RemoteOpt.ConnPool.(*connpool2.LongPool)
+	pcfg = d.Config()
+	test.Assert(t, !pcfg.Enable)
+}
+
 func checkOneOptionDebugInfo(t *testing.T, opt Option, expectStr string) error {
 	o := &Options{}
 	o.Apply([]Option{opt})

pkg/remote/connpool/long_pool.go

@@ -42,7 +42,7 @@ var (
 
 const (
 	DefaultProactiveConnCheckInterval = 3 * time.Second
-	ConfigDumpKey                     = "longpool_config"
+	idleConfigDumpKey                 = "idle_config"
 )
 
 func getSharedTicker(p *LongPool, refreshInterval time.Duration) *utils.SharedTicker {
@@ -461,7 +461,7 @@ func (lp *LongPool) Clean(network, address string) {
 // Dump is used to dump current long pool info when needed, like debug query.
 func (lp *LongPool) Dump() interface{} {
 	m := make(map[string]interface{})
-	m[ConfigDumpKey] = lp.config
+	m[idleConfigDumpKey] = lp.config.IdleConfig
 	lp.peerMap.Range(func(key, value interface{}) bool {
 		t := value.(*peer).pool.Dump()
 		m[key.(netAddr).String()] = t
@@ -470,6 +470,11 @@ func (lp *LongPool) Dump() interface{} {
 	return m
 }
 
+// Config returns the config of the long pool
+func (lp *LongPool) Config() LongPoolConfig {
+	return lp.config
+}
+
 // Close releases all peers in the pool, it is executed when client is closed.
 func (lp *LongPool) Close() error {
 	if !atomic.CompareAndSwapInt32(&lp.closed, 0, 1) {