diff --git a/libfic/db.go b/libfic/db.go index 41ad70af..eb049ff6 100644 --- a/libfic/db.go +++ b/libfic/db.go @@ -5,6 +5,7 @@ import ( _ "github.com/go-sql-driver/mysql" "log" "os" + "strings" "time" ) @@ -19,11 +20,15 @@ func DSNGenerator() string { db_db := "fic" if v, exists := os.LookupEnv("MYSQL_HOST"); exists { - db_host = "tcp(" + v + ":" - if p, exists := os.LookupEnv("MYSQL_PORT"); exists { - db_host += p + ")" + if strings.HasPrefix(v, "/") { + db_host = "unix(" + v + ")" } else { - db_host += "3306)" + db_host = "tcp(" + v + ":" + if p, exists := os.LookupEnv("MYSQL_PORT"); exists { + db_host += p + ")" + } else { + db_host += "3306)" + } } } if v, exists := os.LookupEnv("MYSQL_PASSWORD"); exists {