Modifying Handler.Open to retry opening urlstr with sqlite3 when it a…

…ppears to be a local file
kod3r · Mar 4, 2017 · 3e0f981 · 3e0f981
1 parent d2136f6
commit 3e0f981
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -48,6 +48,6 @@ various queries.
 </p>
 
 # TODO
-* Fix --command/-c execution
+* Better handling of local files (such as unix domain sockets) for Handler.Open
 * All the various \\d* commands from `psql`
 * SQL completion
diff --git a/handler.go b/handler.go
@@ -71,7 +71,14 @@ func (h *Handler) Open(urlstr string) error {
 
 	// parse dsn
 	h.u, err = dburl.Parse(urlstr)
-	if err != nil {
+	switch {
+	case err == dburl.ErrInvalidDatabaseScheme:
+		if _, err = os.Stat(urlstr); err == nil {
+			// it is a file, so reattempt to open it with sqlite3
+			return h.Open("sqlite3:" + urlstr)
+		}
+
+	case err != nil:
 		return err
 	}