Add support for a custom db path

2 files changed, 4 insertions, 2 deletions

diff --git a/db.go b/db.go
index 9de5e2d..b70e885 100644
--- a/db.go
+++ b/db.go
@@ -17,7 +17,7 @@ type User struct {
 }
 
 func InitAndGetDB(driverName, dataSourceName string) (*sql.DB, error) {
-	db, err := sql.Open("sqlite3", "./db/test.db")
+	db, err := sql.Open("sqlite3", dataSourceName)
 	if err != nil {
 		return nil, err
 	}
diff --git a/main.go b/main.go
index 15e7281..04bcf7c 100644
--- a/main.go
+++ b/main.go
@@ -629,9 +629,11 @@ func upload(w http.ResponseWriter, r *http.Request) {
 
 func main() {
 	p := flag.Int("p", 8080, "the port to bind to")
+	dbPath := flag.String("d", "./db/test.db", "the db to connect to")
 	flag.Parse()
 	var err error
-	db, err = InitAndGetDB("sqlite3", "./db/test.db")
+	log.Printf("Connecting to db: %s\n", *dbPath)
+	db, err = InitAndGetDB("sqlite3", *dbPath)
 	if err != nil {
 		panic(err)
 	}