Take DATABASE_URL env for pg connection

Author: st0012

plugin.gb

@@ -12,8 +12,14 @@ db_user = ENV["POSTGRES_USER"] || "postgres"
 db_name = ENV["POSTGRES_DB"]   || "goby_test"
 db_host = ENV["POSTGRES_HOST"] || "0.0.0.0"
 
+pg_host = ENV["DATABASE_URL"]
+
+if pg_host.nil?
+  pg_host = String.fmt("user=%s dbname=%s host=%s sslmode=disable", db_user, db_name, db_host)
+end
+
 # Open PG connection using sqlx package's Open function
-conn, err = plugin.go_func("Open", "postgres", String.fmt("user=%s dbname=%s host=%s sslmode=disable", db_user, db_name, db_host))
+conn, err = plugin.go_func("Open", "postgres", pg_host)
 
 if err
   puts(err.go_func("Error"))