chore: configurable shared memory size for postgres container (#113)

rudderlabs · Sep 11, 2023 · ff2dc63 · ff2dc63
1 parent 1f9dde1
commit ff2dc63
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/testhelper/docker/resource/postgres.go b/testhelper/docker/resource/postgres.go
@@ -7,7 +7,9 @@ import (
 
 	_ "github.com/lib/pq"
 	"github.com/ory/dockertest/v3"
+	dc "github.com/ory/dockertest/v3/docker"
 
+	"github.com/rudderlabs/rudder-go-kit/bytesize"
 	"github.com/rudderlabs/rudder-go-kit/testhelper/docker/resource/postgres"
 )
 
@@ -29,7 +31,8 @@ type PostgresResource struct {
 
 func SetupPostgres(pool *dockertest.Pool, d cleaner, opts ...func(*postgres.Config)) (*PostgresResource, error) {
 	c := &postgres.Config{
-		Tag: "15-alpine",
+		Tag:     "15-alpine",
+		ShmSize: 512 * bytesize.MB,
 	}
 	for _, opt := range opts {
 		opt(c)
@@ -49,6 +52,8 @@ func SetupPostgres(pool *dockertest.Pool, d cleaner, opts ...func(*postgres.Conf
 			"POSTGRES_USER=" + postgresDefaultUser,
 		},
 		Cmd: cmd,
+	}, func(hc *dc.HostConfig) {
+		hc.ShmSize = c.ShmSize
 	})
 	if err != nil {
 		return nil, err

diff --git a/testhelper/docker/resource/postgres/config.go b/testhelper/docker/resource/postgres/config.go
@@ -14,7 +14,14 @@ func WithOptions(options ...string) Opt {
 	}
 }
 
+func WithShmSize(shmSize int64) Opt {
+	return func(c *Config) {
+		c.ShmSize = shmSize
+	}
+}
+
 type Config struct {
 	Tag     string
 	Options []string
+	ShmSize int64
 }