-module(beam_stats_consumer_csv).
-include("include/beam_stats.hrl").
+-include("beam_stats_logging.hrl").
-behaviour(beam_stats_consumer).
]).
-type option() ::
- {path , file:filename()}
- | {consumption_interval , non_neg_integer()}
+ {consumption_interval , non_neg_integer()}
+ | {path , file:filename()}
.
-record(state,
-spec try_to_write(state(), binary()) ->
state().
try_to_write(#state{file=none, path=Path}=State, _) ->
- io:format("error: file closed: ~s~n", [Path]),
+ ?log_error("Writing to file (~p) failed: no file in state.", [Path]),
State;
try_to_write(#state{file={some, File}}=State, Payload) ->
case file:write(File, Payload)
of ok ->
State
; {error, _}=Error ->
- io:format("error: file:write/2 failed: ~p~n", [Error]),
+ ?log_error("file:write(~p, ~p) -> ~p", [File, Payload, Error]),
% TODO: Maybe schedule retry?
ok = file:close(File),
State#state{file=none}
try_to_open_if_no_file(#state{file={some, _}}=State) ->
State;
try_to_open_if_no_file(#state{file=none, path=Path}=State) ->
- case file:open(Path, [append])
+ Options = [append],
+ case file:open(Path, Options)
of {ok, File} ->
State#state{file = {some, File}}
; {error, _}=Error ->
- io:format("error: file:open/2 failed: ~p~n", [Error]),
+ ?log_error("file:open(~p, ~p) -> ~p", [Path, Options, Error]),
State#state{file = none}
end.
) ->
<<TimestampBin/binary>> = timestamp_to_bin(Timestamp),
<<NodeIDBin/binary>> = node_id_to_bin(NodeID),
- PairToBin = make_pair_to_bin(NodeIDBin, TimestampBin),
+ MemoryPairToBin = make_pair_to_bin(NodeIDBin, TimestampBin, <<"memory">>),
MemoryBinPairs = lists:map(fun atom_int_to_bin_bin/1, Memory),
- MemoryBins = lists:map(PairToBin, MemoryBinPairs),
+ MemoryBins = lists:map(MemoryPairToBin, MemoryBinPairs),
AllBins =
[ MemoryBins
],
TotalMicroSeconds = (TotalWholeSeconds * OneMillion) + ComponentMicro,
TotalMicroSeconds / OneMillion.
--spec make_pair_to_bin(binary(), binary()) ->
+-spec make_pair_to_bin(binary(), binary(), binary()) ->
fun(({binary(), binary()}) -> binary()).
-make_pair_to_bin(<<NodeID/binary>>, <<TimestampBin/binary>>) ->
+make_pair_to_bin(<<NodeID/binary>>, <<TimestampBin/binary>>, <<Type/binary>>) ->
fun ({<<K/binary>>, <<V/binary>>}) ->
<< TimestampBin/binary
, "|"
, NodeID/binary
, "|"
+ , Type/binary
+ , "|"
, K/binary
, "|"
, V/binary