4 files changed, 139 insertions, 0 deletions
diff --git a/statsserver/ebin/statsserver.app b/statsserver/ebin/statsserver.app
new file mode 100644
index 0000000..9c642ed
--- /dev/null
+++ b/statsserver/ebin/statsserver.app
@@ -0,0 +1,13 @@
+%%% Copyright (c) 2017, NORDUnet A/S.
+%%% See LICENSE for licensing information.
+
+%%% Application resource file for statsserver, see app(5).
+
+{application, statsserver,
+ [{description, "Plop statsserver"},
+  {vsn, "0.10.1"},
+  {modules, [statsserver_app, statsserver_sup, statsserver]},
+  {applications, [kernel, stdlib, lager, plop]},
+  {registered, [statsserver_sup, statsserver]},
+  {mod, {statsserver_app, []}}
+ ]}.
diff --git a/statsserver/src/statsserver.erl b/statsserver/src/statsserver.erl
new file mode 100644
index 0000000..1d42b27
--- /dev/null
+++ b/statsserver/src/statsserver.erl
@@ -0,0 +1,71 @@
+%%% Copyright (c) 2014-2016, NORDUnet A/S.
+%%% See LICENSE for licensing information.
+
+%%% @doc Frontend node API
+
+-module(statsserver).
+-export([init_module/0]).
+%% API (URL)
+-export([request/4]).
+
+-define(APPURL_PLOP_STATUS, "plop/v1/status").
+
+request(post, ?APPURL_PLOP_STATUS, Service, Input) ->
+    case (catch mochijson2:decode(Input)) of
+        {error, E} ->
+            html("bad input:", E);
+        Entries when is_list(Entries) ->
+            lists:foreach(fun ({struct, PropList}) ->
+                                  Target = proplists:get_value(<<"target">>, PropList),
+                                  Variable = proplists:get_value(<<"key">>, PropList),
+                                  Status = proplists:get_value(<<"value">>, PropList),
+                                  set_status(Service, Target, Variable, Status)
+                          end, Entries)
+    end,
+    success({[{result, <<"ok">>}]});
+request(get, "", "status", Input) ->
+    Now = erlang:monotonic_time(millisecond),
+    Variables = [{struct, [
+                           {service, list_to_binary(Service)},
+                           {target, Target},
+                           {variable, Variable},
+                           {status, Status},
+                           {age, (Now - Timestamp) / 1000}
+                          ]}  || {{Service, Target, Variable}, Status, Timestamp} <- get_status()],
+    success({[{result, Variables}]}).
+
+
+success(Data) ->
+    {200, [{"Content-Type", "text/json"}, {"Access-Control-Allow-Origin", "*"}], mochijson2:encode(Data)}.
+
+html(Text, Input) ->
+    {400, [{"Content-Type", "text/html"}],
+     io_lib:format(
+       "<html><body><p>~n" ++
+           "~s~n" ++
+           "~p~n" ++
+           "</body></html>~n", [Text, Input])}.
+
+-define(STATUSDB_TABLE, statsserver_statusdb).
+
+init_module() ->
+    create_statusdb().
+
+create_statusdb() ->
+    case ets:info(?STATUSDB_TABLE) of
+	undefined ->
+	    ok;
+	_ ->
+	    ets:delete(?STATUSDB_TABLE)
+    end,
+    ets:new(?STATUSDB_TABLE, [set, public, named_table]).
+
+get_status() ->
+    [E || [E] <- ets:match(?STATUSDB_TABLE, '$1')].
+
+set_status(Service, Target, Variable, Status) ->
+    lager:info("status: ~p ~p ~p ~p", [Service, Target, Variable, Status]),
+    Timestamp = erlang:monotonic_time(millisecond),
+    true = ets:insert(?STATUSDB_TABLE,
+                      {{Service, Target, Variable}, Status, Timestamp}),
+    ok.
diff --git a/statsserver/src/statsserver_app.erl b/statsserver/src/statsserver_app.erl
new file mode 100644
index 0000000..6caf2b7
--- /dev/null
+++ b/statsserver/src/statsserver_app.erl
@@ -0,0 +1,13 @@
+%%% Copyright (c) 2017, NORDUnet A/S.
+%%% See LICENSE for licensing information.
+
+-module(statsserver_app).
+-behaviour(application).
+-export([start/2, stop/1]).
+
+start(normal, Args) ->
+    statsserver:init_module(),
+    statsserver_sup:start_link(Args).
+
+stop(_State) ->
+    ok.
diff --git a/statsserver/src/statsserver_sup.erl b/statsserver/src/statsserver_sup.erl
new file mode 100644
index 0000000..6b92e35
--- /dev/null
+++ b/statsserver/src/statsserver_sup.erl
@@ -0,0 +1,42 @@
+%%% Copyright (c) 2017, NORDUnet A/S.
+%%% See LICENSE for licensing information.
+
+-module(statsserver_sup).
+-behaviour(supervisor).
+
+-export([start_link/1, init/1]).
+
+start_link(_Args) ->
+    supervisor:start_link({local, ?MODULE}, ?MODULE, []).
+
+gen_http_config(Config, SSLOptions, SSLFlag) ->
+    {ChildName, IpAddress, Port, Module} = Config,
+    {ok, IPv4Address} = inet:parse_ipv4strict_address(IpAddress),
+    WebConfig = [{ip, IPv4Address},
+                 {port, Port},
+                 {ssl, SSLFlag},
+                 {acceptor_pool_size,
+                  application:get_env(catlfish, http_server_pool_size, 16)},
+                 {ssl_opts, SSLOptions}
+                ],
+    {ChildName,
+     {catlfish_web, start, [WebConfig, Module, ChildName]},
+     permanent, 5000, worker, dynamic}.
+
+
+init([]) ->
+    SSLOptions =
+        [{certfile, application:get_env(plop, https_certfile, none)},
+         {keyfile, application:get_env(plop, https_keyfile, none)},
+         {cacertfile, application:get_env(plop, https_cacertfile, none)}],
+    Servers =
+        lists:map(fun (Config) ->
+                          gen_http_config(Config, SSLOptions, true)
+                  end, application:get_env(plop, https_servers, [])) ++
+        lists:map(fun (Config) ->
+                          gen_http_config(Config, SSLOptions, false)
+                  end, application:get_env(plop, http_servers, [])),
+    lager:debug("Starting servers ~p", [Servers]),
+    {ok,
+     {{one_for_one, 3, 10},
+      Servers}}.