| 1 |
|
%%============================================================================== |
| 2 |
|
%% Copyright 2019 Erlang Solutions Ltd. |
| 3 |
|
%% |
| 4 |
|
%% Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 |
|
%% you may not use this file except in compliance with the License. |
| 6 |
|
%% You may obtain a copy of the License at |
| 7 |
|
%% |
| 8 |
|
%% http://www.apache.org/licenses/LICENSE-2.0 |
| 9 |
|
%% |
| 10 |
|
%% Unless required by applicable law or agreed to in writing, software |
| 11 |
|
%% distributed under the License is distributed on an "AS IS" BASIS, |
| 12 |
|
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 |
|
%% See the License for the specific language governing permissions and |
| 14 |
|
%% limitations under the License. |
| 15 |
|
%%============================================================================== |
| 16 |
|
-module(mongoose_metrics_probe). |
| 17 |
|
-behaviour(exometer_probe). |
| 18 |
|
|
| 19 |
|
-include("mongoose_logger.hrl"). |
| 20 |
|
|
| 21 |
|
%% exometer_entry callbacks |
| 22 |
|
%% exometer_probe callbacks |
| 23 |
|
-export([ |
| 24 |
|
behaviour/0, |
| 25 |
|
probe_init/3, |
| 26 |
|
probe_terminate/1, |
| 27 |
|
probe_get_value/2, |
| 28 |
|
probe_get_datapoints/1, |
| 29 |
|
probe_update/2, |
| 30 |
|
probe_reset/1, |
| 31 |
|
probe_sample/1, |
| 32 |
|
probe_setopts/3, |
| 33 |
|
probe_handle_msg/2, |
| 34 |
|
probe_code_change/3 |
| 35 |
|
]). |
| 36 |
|
|
| 37 |
|
-ignore_xref([behaviour_info/1]). |
| 38 |
|
|
| 39 |
|
-type datapoint() :: atom(). |
| 40 |
|
-callback datapoints() -> [datapoint()]. |
| 41 |
|
-callback sample() -> #{datapoint() => integer()}. |
| 42 |
|
|
| 43 |
|
-record(state, { |
| 44 |
|
callback_module :: atom(), |
| 45 |
|
data = #{} :: #{datapoint() => integer()}, |
| 46 |
|
ref :: reference() | undefined |
| 47 |
|
}). |
| 48 |
|
|
| 49 |
|
-spec behaviour() -> exometer:behaviour(). |
| 50 |
|
behaviour() -> |
| 51 |
532 |
probe. |
| 52 |
|
|
| 53 |
|
probe_init(_Name, _Type, Opts) -> |
| 54 |
532 |
Mod = case proplists:get_value(callback_module, Opts) of |
| 55 |
532 |
M when is_atom(M) -> M; |
| 56 |
:-( |
M -> error({invalid_callback_module, M}) |
| 57 |
|
end, |
| 58 |
532 |
{ok, #state{callback_module = Mod, |
| 59 |
|
data = #{}}}. |
| 60 |
|
|
| 61 |
528 |
probe_terminate(_) -> ok. |
| 62 |
|
|
| 63 |
|
probe_get_value(DPs, #state{data = Data} = S) -> |
| 64 |
666 |
{ok, maps:to_list(maps:with(DPs, Data)), S}. |
| 65 |
|
|
| 66 |
|
probe_get_datapoints(#state{callback_module = Mod}) -> |
| 67 |
666 |
{ok, Mod:datapoints()}. |
| 68 |
|
|
| 69 |
|
probe_update(_, _) -> |
| 70 |
:-( |
{error, not_supported}. |
| 71 |
|
|
| 72 |
|
probe_reset(S) -> |
| 73 |
:-( |
{ok, S#state{data = #{}}}. |
| 74 |
|
|
| 75 |
|
|
| 76 |
|
probe_sample(#state{callback_module = Mod, ref = undefined} = S) -> |
| 77 |
895 |
{_Pid, Ref} = |
| 78 |
|
spawn_monitor( |
| 79 |
|
fun() -> |
| 80 |
895 |
exit({sample, Mod:sample()}) |
| 81 |
|
end), |
| 82 |
895 |
{ok, S#state{ref = Ref}}; |
| 83 |
|
probe_sample(#state{} = S) -> |
| 84 |
:-( |
{ok, S}. |
| 85 |
|
|
| 86 |
|
probe_setopts(_Entry, _Opts, _S) -> |
| 87 |
:-( |
{error, not_supported}. |
| 88 |
|
|
| 89 |
|
probe_handle_msg({'DOWN', Ref, _, _, {sample, Data}}, #state{ref = Ref} = S) -> |
| 90 |
894 |
{ok, S#state{ref = undefined, data = Data}}; |
| 91 |
|
probe_handle_msg({'DOWN', Ref, _, _, Reason}, #state{callback_module = Mod, ref = Ref} = S) -> |
| 92 |
1 |
?LOG_WARNING(#{what => probe_callback_module_died, callback_module => Mod, |
| 93 |
:-( |
reason => Reason}), |
| 94 |
1 |
{ok, S#state{ref = undefined}}; |
| 95 |
|
probe_handle_msg(_, S) -> |
| 96 |
:-( |
{ok, S}. |
| 97 |
|
|
| 98 |
:-( |
probe_code_change(_, S, _) -> {ok, S}. |