snapshot of https://github.com/ory/keto.git (master, 2026-09-16; blobs > 131072B omitted)

oryx/dbal/dsn.go

@@ -0,0 +1,47 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package dbal
+
+import (
+	"fmt"
+	"os"
+	"regexp"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+var sqliteMemoryRegexp = regexp.MustCompile(`^sqlite://file:.+\?.*&?mode=memory($|&.*)|sqlite://(file:)?:memory:\?.*$|^(:memory:|memory)$`)
+
+// IsMemorySQLite returns true if a given DSN string is pointing to a SQLite database.
+//
+// SQLite can be written in different styles depending on the use case
+// - just in memory
+// - shared connection
+// - shared but unique in the same process
+// see: https://sqlite.org/inmemorydb.html
+func IsMemorySQLite(dsn string) bool { return sqliteMemoryRegexp.MatchString(dsn) }
+
+// NewSQLiteTestDatabase creates a new, unique SQLite database
+// which is shared amongst all callers and identified by an individual file name.
+// The database file is created in the system's temporary directory, and not actively
+// removed to allow debugging in case of test failures.
+func NewSQLiteTestDatabase(t testing.TB) string {
+	fn, err := os.MkdirTemp("", "sqlite-test-db-*")
+	require.NoError(t, err)
+	return NewSQLiteDatabase(fn)
+}
+
+// NewSQLiteDatabase creates a new on-disk, unique SQLite database
+// which is shared amongst all callers and identified by an individual file name.
+// This is sometimes necessary over a in-memory database, for example when multiple tests/goroutines run in parallel
+// and access the same table.
+// This would result in a locking error from SQLite when running in-memory.
+// Additionally, shared cache mode is deprecated and discouraged, and the problem is better solved with the WAL,
+// according to official docs.
+//
+// In tests, use [NewSQLiteTestDatabase] instead.
+func NewSQLiteDatabase(name string) string {
+	return fmt.Sprintf("sqlite://%s/db.sqlite?_fk=true&_busy_timeout=100000&_pragma=journal_mode(WAL)&_time_format=sqlite", name)
+}