2013-02-10 13:54:51 -06:00
|
|
|
%% Feel free to use, reuse and abuse the code in this file.
|
|
|
|
|
|
|
|
%% @doc Streaming handler.
|
|
|
|
-module(toppage_handler).
|
|
|
|
|
|
|
|
-export([init/3]).
|
|
|
|
-export([rest_init/2]).
|
|
|
|
-export([content_types_provided/2]).
|
|
|
|
-export([streaming_csv/2]).
|
|
|
|
|
|
|
|
init(_Transport, _Req, _Table) ->
|
|
|
|
{upgrade, protocol, cowboy_rest}.
|
|
|
|
|
|
|
|
rest_init(Req, Table) ->
|
|
|
|
{ok, Req, Table}.
|
|
|
|
|
|
|
|
content_types_provided(Req, State) ->
|
|
|
|
{[
|
|
|
|
{{<<"text">>, <<"csv">>, []}, streaming_csv}
|
|
|
|
], Req, State}.
|
|
|
|
|
|
|
|
streaming_csv(Req, Table) ->
|
2014-09-23 16:43:29 +03:00
|
|
|
N = cowboy_req:binding(v1, Req, 1),
|
2013-02-10 13:54:51 -06:00
|
|
|
MS = [{{'$1', '$2', '$3'}, [{'==', '$2', N}], ['$$']}],
|
2014-09-23 16:43:29 +03:00
|
|
|
{{stream, result_streamer(Table, MS)}, Req, Table}.
|
2013-02-10 13:54:51 -06:00
|
|
|
|
|
|
|
result_streamer(Table, MS) ->
|
|
|
|
fun (Socket, Transport) ->
|
2013-09-08 19:50:31 +02:00
|
|
|
send_records(Socket, Transport, ets:select(Table, MS, 1))
|
2013-02-10 13:54:51 -06:00
|
|
|
end.
|
|
|
|
|
|
|
|
send_records(Socket, Transport, {[Rec], Cont}) ->
|
|
|
|
timer:sleep(500),
|
|
|
|
send_line(Socket, Transport, Rec),
|
|
|
|
send_records(Socket, Transport, ets:select(Cont));
|
|
|
|
send_records(_Socket, _Transport, '$end_of_table') ->
|
|
|
|
ok.
|
|
|
|
|
|
|
|
send_line(Socket, Transport, [Key, V1, V2]) ->
|
|
|
|
Transport:send(Socket,
|
|
|
|
[Key, $,, integer_to_list(V1), $,, integer_to_list(V2), $\r, $\n]).
|