cache: Pre-allocated buffer for SDK client

Signed-off-by: Evgenii Baidakov <evgenii@nspcc.io>
nspcc-dev · Nov 2, 2023 · 1809e15 · 1809e15
1 parent 4fefe67
commit 1809e15
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/cmd/neofs-node/config.go b/cmd/neofs-node/config.go
@@ -536,11 +536,18 @@ func initCfg(appCfg *config.Config) *cfg {
 	)
 	fatalOnErr(err)
 
+	var buffers sync.Pool
+	buffers.New = func() any {
+		b := make([]byte, cache.DefaultBufferSize)
+		return &b
+	}
+
 	cacheOpts := cache.ClientCacheOpts{
 		DialTimeout:      apiclientconfig.DialTimeout(appCfg),
 		StreamTimeout:    apiclientconfig.StreamTimeout(appCfg),
 		AllowExternal:    apiclientconfig.AllowExternal(appCfg),
 		ReconnectTimeout: apiclientconfig.ReconnectTimeout(appCfg),
+		Buffers:          &buffers,
 	}
 	c.shared = shared{
 		key:            key,

diff --git a/pkg/network/cache/client.go b/pkg/network/cache/client.go
@@ -26,6 +26,7 @@ type (
 		ReconnectTimeout time.Duration
 		ResponseCallback func(client.ResponseMetaInfo) error
 		AllowExternal    bool
+		Buffers          *sync.Pool
 	}
 )
 

diff --git a/pkg/network/cache/multi.go b/pkg/network/cache/multi.go
@@ -58,6 +58,8 @@ func (x *multiClient) createForAddress(addr network.Address) (clientcore.Client,
 		prmDial client.PrmDial
 	)
 
+	prmInit.SetSignMessageBuffers(x.opts.Buffers)
+
 	prmDial.SetServerURI(addr.URIAddr())
 
 	if x.opts.DialTimeout > 0 {