valitydev/psql-migration
14
0
Fork 0
mirror of https://github.com/valitydev/psql-migration.git synced 2024-11-06 01:05:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add connection_opts/2 to expose URL to look up

Browse Source
This commit is contained in:
Marc Nijdam 2020-02-08 21:22:08 -07:00
parent a8d8b9ac2d
commit 2236b5dbce
No known key found for this signature in database
GPG Key ID: 1C84AFC593B1B79D
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/psql_migration.erl
View File

@ -1,7 +1,7 @@
-module(psql_migration).
%% API exports
-export([main/1, open_connection/1, connection_opts/1]).
-export([main/1, open_connection/1, connection_opts/1, connection_opts/2]).
%%====================================================================
%% API functions
@ -157,8 +157,11 @@ with_connection(Args, Fun) ->
end.
connection_opts(Args) ->
connection_opts(Args, "DATABASE_URL").
connection_opts(Args, URLName) ->
envloader:load(dot_env(Args)),
URL = os:getenv("DATABASE_URL"),
URL = os:getenv(URLName),
ParseOpts = [{scheme_defaults, [{postgres, 5432}, {postgresql, 5432}]}],
case http_uri:parse(URL, ParseOpts) of
{error, Error} ->