./ct_report/coverage/mongoose_instrument_exometer.COVER.html

1 -module(mongoose_instrument_exometer).
2
3 -behaviour(mongoose_instrument).
4
5 -define(PREFIXES, {?MODULE, prefixes}).
6
7 -export([config_spec/0, start/0, stop/0, set_up/3, handle_event/4]).
8
9 -include("mongoose.hrl").
10 -include("mongoose_config_spec.hrl").
11
12 -spec config_spec() -> mongoose_config_spec:config_section().
13 config_spec() ->
14 101 #section{items = #{<<"all_metrics_are_global">> => #option{type = boolean}},
15 defaults = #{<<"all_metrics_are_global">> => false}}.
16
17 -spec start() -> ok.
18 start() ->
19 101 AllGlobal = mongoose_config:get_opt([instrumentation, exometer, all_metrics_are_global]),
20 101 Prefixes = [{HostType, make_host_type_prefix(HostType, AllGlobal)}
21 101 || HostType <- ?ALL_HOST_TYPES],
22 101 persistent_term:put(?PREFIXES, maps:from_list(Prefixes)).
23
24 -spec stop() -> ok.
25 stop() ->
26 
:-(
 persistent_term:erase(?PREFIXES),
27 
:-(
 ok.
28
29 -spec set_up(mongoose_instrument:event_name(), mongoose_instrument:labels(),
30 mongoose_instrument:config()) -> boolean().
31 set_up(EventName, Labels, #{metrics := Metrics}) ->
32 
:-(
 maps:foreach(fun(MetricName, MetricType) ->
33 
:-(
 set_up_metric(EventName, Labels, MetricName, MetricType)
34 end, Metrics),
35 
:-(
 true;
36 set_up(_EventName, _Labels, #{}) ->
37 
:-(
 false.
38
39 -spec handle_event(mongoose_instrument:event_name(), mongoose_instrument:labels(),
40 mongoose_instrument:config(), mongoose_instrument:measurements()) -> ok.
41 handle_event(EventName, Labels, #{metrics := Metrics}, Measurements) ->
42 
:-(
 maps:foreach(fun(MetricName, MetricType) ->
43 
:-(
 handle_metric_event(EventName, Labels, MetricName, MetricType, Measurements)
44 end, Metrics).
45
46 -spec set_up_metric(mongoose_instrument:event_name(), mongoose_instrument:labels(),
47 mongoose_instrument:metric_name(), mongoose_instrument:metric_type()) ->
48 ok.
49 set_up_metric(EventName, Labels, MetricName, MetricType) ->
50 
:-(
 Name = exometer_metric_name(EventName, Labels, MetricName),
51 
:-(
 try exometer:new(Name, MetricType)
52 catch
53 
:-(
 error:exists -> ok = exometer:reset(Name)
54 end.
55
56 -spec handle_metric_event(mongoose_instrument:event_name(), mongoose_instrument:labels(),
57 mongoose_instrument:metric_name(), mongoose_instrument:metric_type(),
58 mongoose_instrument:measurements()) -> ok.
59 handle_metric_event(EventName, Labels, MetricName, MetricType, Measurements) ->
60 
:-(
 case Measurements of
61 #{MetricName := MetricValue} ->
62 
:-(
 Name = exometer_metric_name(EventName, Labels, MetricName),
63 
:-(
 update_metric(Name, MetricType, MetricValue);
64 #{} ->
65 
:-(
 ok
66 end.
67
68 -spec update_metric(exometer:name(), spiral | histogram, integer()) -> ok.
69 update_metric(Name, gauge, Value) when is_integer(Value) ->
70 
:-(
 ok = exometer:update(Name, Value);
71 update_metric(Name, spiral, Value) when is_integer(Value), Value >= 0 ->
72 
:-(
 ok = exometer:update(Name, Value);
73 update_metric(Name, histogram, Value) when is_integer(Value) ->
74 
:-(
 ok = exometer:update(Name, Value).
75
76 %% This logic will need extending if we add more labels
77 -spec exometer_metric_name(mongoose_instrument:event_name(), mongoose_instrument:labels(),
78 mongoose_instrument:metric_name()) -> exometer:name().
79 exometer_metric_name(EventName, #{host_type := HostType}, MetricName) ->
80 
:-(
 [get_host_type_prefix(HostType), EventName, MetricName].
81
82 -spec get_host_type_prefix(mongooseim:host_type()) -> global | binary().
83 get_host_type_prefix(HostType) ->
84 
:-(
 #{HostType := Prefix} = persistent_term:get(?PREFIXES),
85 
:-(
 Prefix.
86
87 -spec make_host_type_prefix(mongooseim:host_type(), boolean()) -> global | binary().
88 make_host_type_prefix(_HostType, true) ->
89 
:-(
 global;
90 make_host_type_prefix(HostType, false) ->
91 541 binary:replace(HostType, <<" ">>, <<"_">>, [global]).
Line Hits Source