runtest

#!/usr/bin/env escript
%% -*- erlang -*-
%%! +K true

-mode(compile).

main([LogFile, ClientsStr, SecondsStr]) ->
    main([LogFile, "localhost", "8000", ClientsStr, SecondsStr]);
main([LogFile, Server, PortStr, ClientsStr, SecondsStr]) ->
    Port = list_to_integer(PortStr),
    Clients = list_to_integer(ClientsStr),
    Seconds = list_to_integer(SecondsStr),

    code:add_paths([
                    "deps/eleveldb/ebin",
                    "ebin"]),
    
    io:format("Running test with ~p clients for ~p seconds~n"
              "Data File: ~p~n",
              [Clients, Seconds, LogFile]),

    {ok, _} = wsdemo_logger:start_link(LogFile),
    {ok, _} = wsdemo_stats:start_link(Server, Port, Clients),
    
    spawn(fun() ->
                  report_loop(now(), Seconds)
          end),
    erlang:send_after(timer:seconds(Seconds), self(), done),

    receive 
        done ->
            wsdemo_stats:stop(),
            wsdemo_logger:close(),
            done
    end;

main(_) ->
    usage().


usage() ->
    io:format("usage: ~n"
     "runtest LogFile::string() ClientCount::integer() Seconds::integer() | ~n"
     "runtest LogFile::string() Server::string() Port::integer() ClientCount::integer() Seconds::integer()~n", []).

report_loop(Start, Seconds) ->
    case whereis(wsdemo_logger) of
        undefined ->
	   timer:sleep(10),
           report_loop(Start, Seconds);
        Pid ->
	   Elapsed = trunc(timer:now_diff(erlang:now(), Start) / 1000000),
           {message_queue_len, Len} = process_info(Pid, message_queue_len),
           io:format("logger queue: ~w Elapsed: ~ws of ~ws ~n", [Len,Elapsed, Seconds]),
           timer:sleep(30000),
           report_loop(Start, Seconds)
    end.