valitydev/woody_erlang
14
0
Fork 0
mirror of https://github.com/valitydev/woody_erlang.git synced 2024-11-06 02:15:19 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add predefined ip pickers (#91)

Browse Source
This commit is contained in:
Andrey Fadeev 2019-04-10 21:07:38 +03:00 committed by GitHub
parent 4f85c9be4d
commit cf9dc0dad5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/woody_resolver.erl
View File

@ -11,17 +11,24 @@
-type resolve_result() :: {Old::parsed_url(), New::parsed_url()}.
-type options() :: #{
ip_picker => {module(), atom()},
ip_picker => ip_picker(),
timeout => timeout()
}.
-type ip_picker() :: {module(), atom()} | predefined_ip_picker().
-type predefined_ip_picker() ::
random |
first.
-export_type([options/0]).
-export_type([ip_picker/0]).
-export_type([predefined_ip_picker/0]).
-export([resolve_url/2]).
-export([resolve_url/3]).
-define(DEFAULT_RESOLVE_TIMEOUT, infinity).
-define(DEFAULT_IP_PICKER, {erlang, hd}).
-define(DEFAULT_IP_PICKER, first).
%%
@ -115,8 +122,15 @@ parse_hostent(HostEnt, Opts) ->
{get_ip(HostEnt, Opts), get_ip_family(HostEnt)}.
get_ip(HostEnt, Opts) ->
{M, F} = maps:get(ip_picker, Opts, ?DEFAULT_IP_PICKER),
erlang:apply(M, F, [HostEnt#hostent.h_addr_list]).
Picker = maps:get(ip_picker, Opts, ?DEFAULT_IP_PICKER),
apply_ip_picker(Picker, HostEnt#hostent.h_addr_list).
apply_ip_picker(first, [Head | _Tail]) ->
Head;
apply_ip_picker(random, AddrList) ->
lists:nth(rand:uniform(length(AddrList)), AddrList);
apply_ip_picker({M, F}, AddrList) ->
erlang:apply(M, F, [AddrList]).
get_ip_family(HostEnt) ->
HostEnt#hostent.h_addrtype.