DNS解析测试脚本

test1.erl

%%%-------------------------------------------------------------------
%%% @author nieyong
%%% @copyright (C) 2015
%%% @doc
%%% DNS解析测试
%%% @end
%%%-------------------------------------------------------------------
-module(test1).
-export([start/2]).

%%
%% 传入域名列表和每一个域名执行时间
%% 使用方式：test1:start(["www.baidu.com", "www.taobao.com", "www.qq.com", "www.sina.com.cn", "www.youku.com"], 1000).
%% 
start(Hosts, Num)  when is_integer(Num), Num > 0 ->
	StartTime = now_timestamp_milliseconds(),
	PID = spawn(fun() -> loop(0, 0, Num * length(Hosts), StartTime) end),
	[spawn(fun() -> get_host_fn(Host, N, PID), ok end) ||  N <- lists:seq(1, Num), Host <- Hosts],
	ok;
start(_Host, _Other) -> 
	io:format("Invalid Argument~n"),
	ok.	

get_host_fn(Host, _Id, PID) ->
	case inet_res:gethostbyname(Host) of
        	 {ok, _Hostent} ->
                          PID ! success;
                          %%io:format("~p times ~p : ~p~n", [Id, Host, Hostent]);
             {error, _Reason} -> PID ! failure %%, throw(Reason) %%throw(Reason)
        end,
	ok.

%% 等待处理结果，输出最终测试报告
loop(Succ, Fail, Total, StartTime) ->
	case (Succ + Fail) == Total of
		true ->
			UseTime = now_timestamp_milliseconds() - StartTime,
			io:format("Summary DNS Query: [Total : ~p, Success : ~p, Failure : ~p, UseTime ~w seconds]~n", [Total, Succ, Fail, UseTime/1000]),
			self() ! quit;
		false ->
			ok
	end,

	receive
		success -> 
			NewSucc = Succ + 1,
			loop(NewSucc, Fail, Total, StartTime);
		failure	->
			NewFail = Fail + 1,
			loop(Succ, NewFail, Total, StartTime);
		_	->
			true
	end.

%%
%% 当前时间戳，以毫秒表示
%%
now_timestamp_milliseconds() ->
  {Mega, Sec, Micro} = os:timestamp(),
  (Mega*1000000 + Sec)*1000 + round(Micro/1000).