Improve spec of list interleave.
[beam_stats.git] / src / beam_stats_msg_graphite.erl
CommitLineData
a9ed8751
SK
1-module(beam_stats_msg_graphite).
2
3-include("include/beam_stats.hrl").
6e1a5b00 4-include("include/beam_stats_ets_table.hrl").
a9ed8751 5-include("include/beam_stats_msg_graphite.hrl").
4bb8ddfe
SK
6-include("include/beam_stats_process.hrl").
7-include("include/beam_stats_process_ancestry.hrl").
8-include("include/beam_stats_processes.hrl").
a9ed8751
SK
9
10-export_type(
11 [ t/0
12 ]).
13
14-export(
15 [ of_beam_stats/1
ece99ea3 16 , of_beam_stats/2
cdcb989e
PO
17 , to_iolist/1
18 , path_to_iolist/1
ece99ea3 19 , node_id_to_bin/1
a9ed8751
SK
20 ]).
21
22-define(T, #?MODULE).
23
24-type t() ::
25 ?T{}.
26
67948b18
SK
27%% ============================================================================
28%% API
29%% ============================================================================
30
a9ed8751
SK
31-spec of_beam_stats(beam_stats:t()) ->
32 [t()].
33of_beam_stats(#beam_stats{node_id=NodeID}=BeamStats) ->
34 NodeIDBin = node_id_to_bin(NodeID),
35 of_beam_stats(BeamStats, NodeIDBin).
36
37-spec of_beam_stats(beam_stats:t(), binary()) ->
38 [t()].
39of_beam_stats(#beam_stats
40 { timestamp = Timestamp
41 , node_id = _
42 , memory = Memory
a37cd038
SK
43 , io_bytes_in = IOBytesIn
44 , io_bytes_out = IOBytesOut
45 , context_switches = ContextSwitches
46 , reductions = Reductions
47 , run_queue = RunQueue
6e1a5b00 48 , ets = ETS
4bb8ddfe 49 , processes = Processes
a9ed8751
SK
50 },
51 <<NodeID/binary>>
52) ->
a37cd038
SK
53 Ts = Timestamp,
54 N = NodeID,
8fe744e7
SK
55 Msgs =
56 [ cons([N, <<"io">> , <<"bytes_in">> ], IOBytesIn , Ts)
57 , cons([N, <<"io">> , <<"bytes_out">>], IOBytesOut , Ts)
58 , cons([N, <<"context_switches">> ], ContextSwitches, Ts)
59 , cons([N, <<"reductions">> ], Reductions , Ts)
60 , cons([N, <<"run_queue">> ], RunQueue , Ts)
61 | of_memory(Memory, NodeID, Ts)
62 ]
63 ++ of_ets(ETS, NodeID, Ts)
64 ++ of_processes(Processes, NodeID, Ts),
65 lists:map(fun path_prefix_schema_version/1, Msgs).
a9ed8751 66
cdcb989e
PO
67-spec to_iolist(t()) ->
68 iolist().
69to_iolist(
67948b18
SK
70 ?T
71 { path = Path
72 , value = Value
73 , timestamp = Timestamp
74 }
75) ->
cdcb989e 76 PathIOList = path_to_iolist(Path),
67948b18
SK
77 ValueBin = integer_to_binary(Value),
78 TimestampInt = timestamp_to_integer(Timestamp),
79 TimestampBin = integer_to_binary(TimestampInt),
cdcb989e 80 [PathIOList, <<" ">>, ValueBin, <<" ">>, TimestampBin].
67948b18 81
cdcb989e
PO
82-spec path_to_iolist([binary()]) ->
83 iolist().
84path_to_iolist(Path) ->
85 interleave(Path, <<".">>).
9b9299d9 86
ece99ea3
SK
87-spec node_id_to_bin(node()) ->
88 binary().
89node_id_to_bin(NodeID) ->
90 NodeIDBin = atom_to_binary(NodeID, utf8),
91 re:replace(NodeIDBin, "[\@\.]", "_", [global, {return, binary}]).
92
67948b18
SK
93%% ============================================================================
94%% Helpers
95%% ============================================================================
96
8fe744e7
SK
97-spec path_prefix_schema_version(t()) ->
98 t().
99path_prefix_schema_version(?T{}=T) ->
100 path_prefix(T, schema_version()).
101
102-spec path_prefix(t(), binary()) ->
103 t().
104path_prefix(?T{path=Path}=T, <<Prefix/binary>>) ->
105 T?T{path = [Prefix | Path]}.
106
107-spec schema_version() ->
108 binary().
109schema_version() ->
110 <<"beam_stats_v0">>.
111
e1221e42
SK
112-spec interleave([A], A) ->
113 [A].
cdcb989e
PO
114interleave([], _) -> [];
115interleave([X], _) -> [X];
116interleave([X|Xs], Sep) ->
117 [X, Sep | interleave(Xs, Sep)].
67948b18
SK
118
119-spec timestamp_to_integer(erlang:timestamp()) ->
120 non_neg_integer().
121timestamp_to_integer({Megaseconds, Seconds, _}) ->
122 Megaseconds * 1000000 + Seconds.
123
a9ed8751
SK
124-spec of_memory([{atom(), non_neg_integer()}], binary(), erlang:timestamp()) ->
125 [t()].
126of_memory(Memory, <<NodeID/binary>>, Timestamp) ->
127 ComponentToMessage =
128 fun ({Key, Value}) ->
129 KeyBin = atom_to_binary(Key, latin1),
d07a494b 130 cons([NodeID, <<"memory">>, KeyBin], Value, Timestamp)
a9ed8751
SK
131 end,
132 lists:map(ComponentToMessage, Memory).
133
7581255c 134-spec of_ets(beam_stats_ets:t(), binary(), erlang:timestamp()) ->
6e1a5b00
SK
135 [t()].
136of_ets(PerTableStats, <<NodeID/binary>>, Timestamp) ->
137 OfEtsTable = fun (Table) -> of_ets_table(Table, NodeID, Timestamp) end,
c37a5360
SK
138 MsgsNested = lists:map(OfEtsTable, PerTableStats),
139 MsgsFlattened = lists:append(MsgsNested),
140 aggregate_by_path(MsgsFlattened, Timestamp).
6e1a5b00
SK
141
142-spec of_ets_table(beam_stats_ets_table:t(), binary(), erlang:timestamp()) ->
143 [t()].
144of_ets_table(#beam_stats_ets_table
145 { id = ID
146 , name = Name
147 , size = Size
148 , memory = Memory
149 },
150 <<NodeID/binary>>,
151 Timestamp
152) ->
c37a5360
SK
153 IDType =
154 case ID =:= Name
155 of true -> <<"NAMED">>
156 ; false -> <<"TID">>
157 end,
6e1a5b00 158 NameBin = atom_to_binary(Name, latin1),
c37a5360 159 NameAndID = [NameBin, IDType],
6e1a5b00
SK
160 [ cons([NodeID, <<"ets_table">>, <<"size">> | NameAndID], Size , Timestamp)
161 , cons([NodeID, <<"ets_table">>, <<"memory">> | NameAndID], Memory, Timestamp)
162 ].
163
4bb8ddfe
SK
164-spec of_processes(beam_stats_processes:t(), binary(), erlang:timestamp()) ->
165 [t()].
166of_processes(
167 #beam_stats_processes
168 { individual_stats = Processes
169 , count_all = CountAll
170 , count_exiting = CountExiting
171 , count_garbage_collecting = CountGarbageCollecting
172 , count_registered = CountRegistered
173 , count_runnable = CountRunnable
174 , count_running = CountRunning
175 , count_suspended = CountSuspended
176 , count_waiting = CountWaiting
177 },
178 <<NodeID/binary>>,
179 Timestamp
180) ->
181 OfProcess = fun (P) -> of_process(P, NodeID, Timestamp) end,
182 PerProcessMsgsNested = lists:map(OfProcess, Processes),
183 PerProcessMsgsFlattened = lists:append(PerProcessMsgsNested),
697c496d 184 PerProcessMsgsAggregates = aggregate_by_path(PerProcessMsgsFlattened, Timestamp),
4bb8ddfe
SK
185 Ts = Timestamp,
186 N = NodeID,
187 [ cons([N, <<"processes_count_all">> ], CountAll , Ts)
188 , cons([N, <<"processes_count_exiting">> ], CountExiting , Ts)
189 , cons([N, <<"processes_count_garbage_collecting">>], CountGarbageCollecting, Ts)
190 , cons([N, <<"processes_count_registered">> ], CountRegistered , Ts)
191 , cons([N, <<"processes_count_runnable">> ], CountRunnable , Ts)
192 , cons([N, <<"processes_count_running">> ], CountRunning , Ts)
193 , cons([N, <<"processes_count_suspended">> ], CountSuspended , Ts)
194 , cons([N, <<"processes_count_waiting">> ], CountWaiting , Ts)
697c496d 195 | PerProcessMsgsAggregates
4bb8ddfe
SK
196 ].
197
198-spec of_process(beam_stats_process:t(), binary(), erlang:timestamp()) ->
199 [t()].
200of_process(
201 #beam_stats_process
f65a3845 202 { pid = _
4bb8ddfe
SK
203 , memory = Memory
204 , total_heap_size = TotalHeapSize
205 , stack_size = StackSize
206 , message_queue_len = MsgQueueLen
7dbc59b6 207 , reductions = Reductions
4bb8ddfe
SK
208 }=Process,
209 <<NodeID/binary>>,
210 Timestamp
211) ->
212 Origin = beam_stats_process:get_best_known_origin(Process),
213 OriginBin = proc_origin_to_bin(Origin),
4bb8ddfe
SK
214 Ts = Timestamp,
215 N = NodeID,
f65a3845
SK
216 [ cons([N, <<"process_memory">> , OriginBin], Memory , Ts)
217 , cons([N, <<"process_total_heap_size">> , OriginBin], TotalHeapSize , Ts)
218 , cons([N, <<"process_stack_size">> , OriginBin], StackSize , Ts)
219 , cons([N, <<"process_message_queue_len">> , OriginBin], MsgQueueLen , Ts)
7dbc59b6 220 , cons([N, <<"process_reductions">> , OriginBin], Reductions , Ts)
4bb8ddfe
SK
221 ].
222
697c496d
SK
223-spec aggregate_by_path([t()], erlang:timestamp()) ->
224 [t()].
225aggregate_by_path(Msgs, Timestamp) ->
226 Aggregate =
227 fun (?T{path=K, value=V}, ValsByPath) ->
228 dict:update_counter(K, V, ValsByPath)
229 end,
230 ValsByPathDict = lists:foldl(Aggregate, dict:new(), Msgs),
231 ValsByPathList = dict:to_list(ValsByPathDict),
232 [cons(Path, Value, Timestamp) || {Path, Value} <- ValsByPathList].
233
4bb8ddfe
SK
234-spec proc_origin_to_bin(beam_stats_process:best_known_origin()) ->
235 binary().
236proc_origin_to_bin({registered_name, Name}) ->
f65a3845
SK
237 NameBin = atom_to_binary(Name, utf8),
238 <<"named--", NameBin/binary>>;
4bb8ddfe
SK
239proc_origin_to_bin({ancestry, Ancestry}) ->
240 #beam_stats_process_ancestry
241 { raw_initial_call = InitCallRaw
242 , otp_initial_call = InitCallOTPOpt
243 , otp_ancestors = AncestorsOpt
244 } = Ancestry,
245 Blank = <<"NONE">>,
246 InitCallOTPBinOpt = hope_option:map(InitCallOTPOpt , fun mfa_to_bin/1),
247 InitCallOTPBin = hope_option:get(InitCallOTPBinOpt, Blank),
248 AncestorsBinOpt = hope_option:map(AncestorsOpt , fun ancestors_to_bin/1),
249 AncestorsBin = hope_option:get(AncestorsBinOpt , Blank),
250 InitCallRawBin = mfa_to_bin(InitCallRaw),
f65a3845
SK
251 << "spawned-via--"
252 , InitCallRawBin/binary
4bb8ddfe
SK
253 , "--"
254 , InitCallOTPBin/binary
255 , "--"
256 , AncestorsBin/binary
257 >>.
258
259ancestors_to_bin([]) ->
260 <<>>;
261ancestors_to_bin([A | Ancestors]) ->
262 ABin = ancestor_to_bin(A),
263 case ancestors_to_bin(Ancestors)
264 of <<>> ->
265 ABin
266 ; <<AncestorsBin/binary>> ->
267 <<ABin/binary, "-", AncestorsBin/binary>>
268 end.
269
270ancestor_to_bin(A) when is_atom(A) ->
271 atom_to_binary(A, utf8);
272ancestor_to_bin(A) when is_pid(A) ->
10bc7b71 273 <<"PID">>.
4bb8ddfe
SK
274
275-spec mfa_to_bin(mfa()) ->
276 binary().
277mfa_to_bin({Module, Function, Arity}) ->
278 ModuleBin = atom_to_binary(Module , utf8),
279 FunctionBin = atom_to_binary(Function, utf8),
280 ArityBin = erlang:integer_to_binary(Arity),
281 <<ModuleBin/binary, "-", FunctionBin/binary, "-", ArityBin/binary>>.
282
a37cd038
SK
283-spec cons([binary()], integer(), erlang:timestamp()) ->
284 t().
285cons(Path, Value, Timestamp) ->
286 ?T
287 { path = Path
288 , value = Value
289 , timestamp = Timestamp
290 }.
This page took 0.040631 seconds and 4 git commands to generate.