make database handler public method

tutipay · Sep 21, 2022 · 6df1a56 · 6df1a56
1 parent e4dd524
commit 6df1a56
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -40,4 +40,9 @@ location /ws {
      proxy_set_header Host $host;
 }
 ```
+This might not be needed, since we do in `ServeWs` endpoint.
+
+### Notes about the database
+
+There's no simple way of using a database.
 
diff --git a/db.go b/db.go
@@ -28,8 +28,8 @@ type Message struct {
 	IsDelivered bool   `db:"is_delivered" json:"is_delivered,omitempty"`
 }
 
-func openDb() (*sqlx.DB, error) {
-	db, err := sqlx.Connect("sqlite3", "test.db")
+func OpenDb(name string) (*sqlx.DB, error) {
+	db, err := sqlx.Connect("sqlite3", name)
 	if err != nil {
 		log.Printf("error in opening db: %v", err)
 		return nil, err

diff --git a/db_test.go b/db_test.go
@@ -7,7 +7,7 @@ import (
 	_ "github.com/mattn/go-sqlite3"
 )
 
-var testDb, _ = openDb()
+var testDb, _ = OpenDb("test.db")
 
 func TestChat_insert(t *testing.T) {