@@ -68,9 +68,10 @@ OptionParser.new do |opts| end.parse! if command = ARGV[0] - if !%w{reload-handlers}.include?(command) + if !%w{reload-handlers stats}.include?(command) puts "Invlalid command. Valid commands are:" puts " reload-handlers" + puts " stats" exit(1) end bin/ernie @@ -2,7 +2,7 @@ -behaviour(gen_server). %% api --export([start_link/1, start/1, lease/0, return/1, reload_assets/0]). +-export([start_link/1, start/1, lease/0, return/1, reload_assets/0, idle_worker_count/0]). %% gen_server callbacks -export([init/1, handle_call/3, handle_cast/2, handle_info/2, @@ -31,6 +31,9 @@ return(Asset) -> reload_assets() -> gen_server:call({global, ?MODULE}, {reload_assets}). +idle_worker_count() -> + gen_server:call({global, ?MODULE}, {idle_worker_count}). + %%==================================================================== %% gen_server callbacks %%==================================================================== @@ -91,6 +94,9 @@ handle_call({return, Asset}, _From, State) -> handle_call({reload_assets}, _From, State) -> Token = make_ref(), {reply, ok, State#state{token = Token}}; +handle_call({idle_worker_count}, _From, State) -> + WorkerCount = queue:len(State#state.assets), + {reply, WorkerCount, State}; handle_call(_Request, _From, State) -> {reply, ok, State}. elib/asset_pool.erl @@ -126,6 +126,14 @@ process_admin(Sock, reload_handlers, _Args, State) -> gen_tcp:send(Sock, term_to_binary({reply, <<"Handlers reloaded.">>})), ok = gen_tcp:close(Sock), State; +process_admin(Sock, stats, _Args, State) -> + IdleWorkers = asset_pool:idle_worker_count(), + IdleWorkersString = list_to_binary([<<"idle workers: ">>, integer_to_list(IdleWorkers), <<"\n">>]), + QueueLength = queue:len(State#state.pending), + QueueLengthString = list_to_binary([<<"pending connections: ">>, integer_to_list(QueueLength), <<"\n">>]), + gen_tcp:send(Sock, term_to_binary({reply, list_to_binary([IdleWorkersString, QueueLengthString])})), + ok = gen_tcp:close(Sock), + State; process_admin(Sock, _Fun, _Args, State) -> gen_tcp:send(Sock, term_to_binary({reply, <<"Admin function not supported.">>})), ok = gen_tcp:close(Sock), elib/ernie_server.erl f4301e0c469dbfd835706d02133c56915d2d3596 Tom Preston-Werner tom@mojombo.com http://github.com/mojombo/ernie/commit/a23e8414645af99774eed7914f5f8e78f67ba5b8 a23e8414645af99774eed7914f5f8e78f67ba5b8 2009-08-20T16:57:00-07:00 2009-08-20T16:57:00-07:00 add stats introspection e19310332f457e91b4821de72d811a61c9b05f44 Tom Preston-Werner tom@mojombo.com