Merge pull request #540 from trungutt/fix/sqlite-database-locking

Attempt to fix SQLite database locking issues with concurrent access

Author: trungutt

pkg/memory/database/sqlite/sqlite.go

@@ -14,11 +14,20 @@ type MemoryDatabase struct {
 }
 
 func NewMemoryDatabase(path string) (database.Database, error) {
-	db, err := sql.Open("sqlite", path)
+	// Add query parameters for better concurrency handling
+	// _busy_timeout: Wait up to 5 seconds if database is locked
+	// _journal_mode=WAL: Enable Write-Ahead Logging for better concurrent access
+	db, err := sql.Open("sqlite", path+"?_busy_timeout=5000&_journal_mode=WAL")
 	if err != nil {
 		return nil, err
 	}
 
+	// Configure connection pool to serialize writes (SQLite limitation)
+	// This prevents "database is locked" errors from concurrent writes
+	db.SetMaxOpenConns(1)
+	db.SetMaxIdleConns(1)
+	db.SetConnMaxLifetime(0)
+
 	_, err = db.ExecContext(context.Background(), "CREATE TABLE IF NOT EXISTS memories (id TEXT PRIMARY KEY, created_at TEXT, memory TEXT)")
 	if err != nil {
 		return nil, err

pkg/session/store.go

@@ -42,11 +42,20 @@ type SQLiteSessionStore struct {
 
 // NewSQLiteSessionStore creates a new SQLite session store
 func NewSQLiteSessionStore(path string) (Store, error) {
-	db, err := sql.Open("sqlite", path)
+	// Add query parameters for better concurrency handling
+	// _busy_timeout: Wait up to 5 seconds if database is locked
+	// _journal_mode=WAL: Enable Write-Ahead Logging for better concurrent access
+	db, err := sql.Open("sqlite", path+"?_busy_timeout=5000&_journal_mode=WAL")
 	if err != nil {
 		return nil, err
 	}
 
+	// Configure connection pool to serialize writes (SQLite limitation)
+	// This prevents "database is locked" errors from concurrent writes
+	db.SetMaxOpenConns(1)
+	db.SetMaxIdleConns(1)
+	db.SetConnMaxLifetime(0)
+
 	_, err = db.ExecContext(context.Background(), `
 		CREATE TABLE IF NOT EXISTS sessions (
 			id TEXT PRIMARY KEY,