From 223b14e5564261e3f65b223536bf4bd836d859e2 Mon Sep 17 00:00:00 2001
From: James Houlahan <james.houlahan@proton.ch>
Date: Thu, 10 Nov 2022 20:06:36 +0100
Subject: [PATCH] Other: Configure attachment pool size in vault

---
 internal/bridge/bridge.go        | 1 +
 internal/vault/settings.go       | 5 +++++
 internal/vault/settings_test.go  | 1 +
 internal/vault/types_settings.go | 2 ++
 4 files changed, 9 insertions(+)

diff --git a/internal/bridge/bridge.go b/internal/bridge/bridge.go
index 07e29c81..412c9506 100644
--- a/internal/bridge/bridge.go
+++ b/internal/bridge/bridge.go
@@ -136,6 +136,7 @@ func New( //nolint:funlen
 		liteapi.WithAppVersion(constants.AppVersion(curVersion.Original())),
 		liteapi.WithCookieJar(cookieJar),
 		liteapi.WithTransport(roundTripper),
+		liteapi.WithAttPoolSize(vault.SyncAttPool()),
 		liteapi.WithLogger(logrus.StandardLogger()),
 	)
 
diff --git a/internal/vault/settings.go b/internal/vault/settings.go
index 5d98cef1..fd61d9c3 100644
--- a/internal/vault/settings.go
+++ b/internal/vault/settings.go
@@ -211,3 +211,8 @@ func (vault *Vault) SyncWorkers() int {
 func (vault *Vault) SyncBuffer() int {
 	return vault.get().Settings.SyncBuffer
 }
+
+// SyncAttPool returns the size of the attachment pool.
+func (vault *Vault) SyncAttPool() int {
+	return vault.get().Settings.SyncAttPool
+}
diff --git a/internal/vault/settings_test.go b/internal/vault/settings_test.go
index b053a8a6..56967595 100644
--- a/internal/vault/settings_test.go
+++ b/internal/vault/settings_test.go
@@ -224,4 +224,5 @@ func TestVault_Settings_SyncWorkers(t *testing.T) {
 
 	require.Equal(t, runtime.NumCPU(), s.SyncWorkers())
 	require.Equal(t, runtime.NumCPU(), s.SyncBuffer())
+	require.Equal(t, runtime.NumCPU(), s.SyncAttPool())
 }
diff --git a/internal/vault/types_settings.go b/internal/vault/types_settings.go
index 48bb3319..a6d1974f 100644
--- a/internal/vault/types_settings.go
+++ b/internal/vault/types_settings.go
@@ -47,6 +47,7 @@ type Settings struct {
 
 	SyncWorkers int
 	SyncBuffer  int
+	SyncAttPool int
 }
 
 func newDefaultSettings(gluonDir string) Settings {
@@ -73,5 +74,6 @@ func newDefaultSettings(gluonDir string) Settings {
 
 		SyncWorkers: runtime.NumCPU(),
 		SyncBuffer:  runtime.NumCPU(),
+		SyncAttPool: runtime.NumCPU(),
 	}
 }