aboutsummaryrefslogblamecommitdiffstats
path: root/examples/rest_stream_response/src/toppage_handler.erl
blob: 61f5ab92d33fbcd2ea86b6509aab0a99ade3f11c (plain) (tree)
1
2
3
4
5
6
7
8
9




                                                           
                  


                                    
                   
                                  






                                                            
                                           
                                                               
                                                           


                                  
                                                                         











                                                                                   
%% Feel free to use, reuse and abuse the code in this file.

%% @doc Streaming handler.
-module(toppage_handler).

-export([init/2]).
-export([content_types_provided/2]).
-export([streaming_csv/2]).

init(Req, Table) ->
	{cowboy_rest, Req, Table}.

content_types_provided(Req, State) ->
	{[
		{{<<"text">>, <<"csv">>, []}, streaming_csv}
	], Req, State}.

streaming_csv(Req, Table) ->
	N = cowboy_req:binding(v1, Req, 1),
	MS = [{{'$1', '$2', '$3'}, [{'==', '$2', N}], ['$$']}],
	{{stream, result_streamer(Table, MS)}, Req, Table}.

result_streamer(Table, MS) ->
	fun (Socket, Transport) ->
		send_records(Socket, Transport, ets:select(Table, MS, 1))
	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]).