Apply memory limits per ncruces suggestion

ncruces/go-sqlite3#148 (comment)
mtlynch · Sep 12, 2024 · 632e08a · 632e08a
1 parent 4c6a0bf
commit 632e08a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/store/sqlite/sqlite.go b/store/sqlite/sqlite.go
@@ -5,9 +5,11 @@ import (
 	"log"
 	"time"
 
+	"github.com/ncruces/go-sqlite3"
 	"github.com/ncruces/go-sqlite3/driver"
 	_ "github.com/ncruces/go-sqlite3/embed"
 	"github.com/ncruces/go-sqlite3/ext/blobio"
+	"github.com/tetratelabs/wazero"
 
 	"github.com/mtlynch/picoshare/v2/picoshare"
 )
@@ -51,6 +53,10 @@ func New(path string, optimizeForLitestream bool) Store {
 		}
 	}
 
+	// Try to prevent memory exhaustion.
+	// https://github.com/ncruces/go-sqlite3/issues/148#issuecomment-2344857749
+	sqlite3.RuntimeConfig = wazero.NewRuntimeConfig().WithMemoryLimitPages(512)
+
 	applyMigrations(ctx)
 
 	return Store{