Add support for a custom db path

1 files changed, 3 insertions, 1 deletions

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)
 	}