From 417daf0028e5e3bb301ba349810c6b49ceba08b6 Mon Sep 17 00:00:00 2001
From: Haelnorr <lockedonohoe@gmail.com>
Date: Mon, 17 Feb 2025 11:53:20 +1100
Subject: [PATCH] Added close method to SafeConn

---
 db/connection.go | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/db/connection.go b/db/connection.go
index 1057dc9..421d4e6 100644
--- a/db/connection.go
+++ b/db/connection.go
@@ -112,13 +112,22 @@ func (conn *SafeConn) Resume() {
 	conn.mux.Unlock()
 }
 
+// Close the database connection
+func (conn *SafeConn) Close() error {
+	conn.mux.Lock()
+	defer conn.mux.Unlock()
+	return conn.db.Close()
+}
+
 // Returns a database connection handle for the Turso DB
-func ConnectToDatabase(dbName string) (*sql.DB, error) {
+func ConnectToDatabase(dbName string) (*SafeConn, error) {
 	file := fmt.Sprintf("file:%s.db", dbName)
 	db, err := sql.Open("sqlite3", file)
-
 	if err != nil {
 		return nil, errors.Wrap(err, "sql.Open")
 	}
-	return db, nil
+
+	conn := &SafeConn{db: db}
+
+	return conn, nil
 }