Support connecting with a unix domain socket

Fixes #285

Author: Ramblurr

pkg/database/connection.go

@@ -2,6 +2,7 @@ package database
 
 import (
 	"fmt"
+	"strings"
 
 	mysql "github.com/go-sql-driver/mysql"
 )
@@ -17,8 +18,13 @@ func (c Connection) MySQL() string {
 	config := mysql.NewConfig()
 	config.User = c.User
 	config.Passwd = c.Pass
-	config.Net = "tcp"
-	config.Addr = fmt.Sprintf("%s:%d", c.Host, c.Port)
+	if strings.HasPrefix(c.Host, "/") {
+		config.Net = "unix"
+		config.Addr = c.Host
+	} else {
+		config.Net = "tcp"
+		config.Addr = fmt.Sprintf("%s:%d", c.Host, c.Port)
+	}
 	config.ParseTime = true
 	return config.FormatDSN()
 }

pkg/database/schemas.go

@@ -17,7 +17,7 @@ func init() {
 }
 
 func GetSchemas(dbconn Connection) ([]string, error) {
-	db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@tcp(%s:%d)/", dbconn.User, dbconn.Pass, dbconn.Host, dbconn.Port))
+	db, err := sql.Open("mysql", dbconn.MySQL())
 	if err != nil {
 		return nil, fmt.Errorf("failed to open connection to database: %v", err)
 	}