mirror of
https://gitee.com/dgiiot/dgiot.git
synced 2024-12-12 12:16:01 +08:00
89 lines
2.7 KiB
Erlang
89 lines
2.7 KiB
Erlang
%%--------------------------------------------------------------------
|
|
%% Copyright (c) 2019-2021 EMQ Technologies Co., Ltd. All Rights Reserved.
|
|
%%
|
|
%% Licensed under the Apache License, Version 2.0 (the "License");
|
|
%% you may not use this file except in compliance with the License.
|
|
%% You may obtain a copy of the License at
|
|
%%
|
|
%% http://www.apache.org/licenses/LICENSE-2.0
|
|
%%
|
|
%% Unless required by applicable law or agreed to in writing, software
|
|
%% distributed under the License is distributed on an "AS IS" BASIS,
|
|
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
%% See the License for the specific language governing permissions and
|
|
%% limitations under the License.
|
|
%%--------------------------------------------------------------------
|
|
|
|
-module(emqx_alarm_handler).
|
|
|
|
-behaviour(gen_event).
|
|
|
|
-include("emqx.hrl").
|
|
-include("logger.hrl").
|
|
|
|
-logger_header("[Alarm Handler]").
|
|
|
|
%% gen_event callbacks
|
|
-export([ init/1
|
|
, handle_event/2
|
|
, handle_call/2
|
|
, handle_info/2
|
|
, terminate/2
|
|
]).
|
|
|
|
-export([ load/0
|
|
, unload/0
|
|
]).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% API
|
|
%%--------------------------------------------------------------------
|
|
|
|
load() ->
|
|
gen_event:swap_handler(alarm_handler, {alarm_handler, swap}, {?MODULE, []}).
|
|
|
|
%% on the way shutting down, give it back to OTP
|
|
unload() ->
|
|
gen_event:swap_handler(alarm_handler, {?MODULE, swap}, {alarm_handler, []}).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% gen_event callbacks
|
|
%%--------------------------------------------------------------------
|
|
|
|
init({_Args, {alarm_handler, _ExistingAlarms}}) ->
|
|
{ok, []};
|
|
|
|
init(_) ->
|
|
{ok, []}.
|
|
|
|
handle_event({set_alarm, {system_memory_high_watermark, []}}, State) ->
|
|
emqx_alarm:activate(high_system_memory_usage, #{high_watermark => emqx_os_mon:get_sysmem_high_watermark()}),
|
|
{ok, State};
|
|
|
|
handle_event({set_alarm, {process_memory_high_watermark, Pid}}, State) ->
|
|
emqx_alarm:activate(high_process_memory_usage, #{pid => list_to_binary(pid_to_list(Pid)),
|
|
high_watermark => emqx_os_mon:get_procmem_high_watermark()}),
|
|
{ok, State};
|
|
|
|
handle_event({clear_alarm, system_memory_high_watermark}, State) ->
|
|
emqx_alarm:deactivate(high_system_memory_usage),
|
|
{ok, State};
|
|
|
|
handle_event({clear_alarm, process_memory_high_watermark}, State) ->
|
|
emqx_alarm:deactivate(high_process_memory_usage),
|
|
{ok, State};
|
|
|
|
handle_event(_, State) ->
|
|
{ok, State}.
|
|
|
|
handle_info(_, State) ->
|
|
{ok, State}.
|
|
|
|
handle_call(_Query, State) ->
|
|
{ok, {error, bad_query}, State}.
|
|
|
|
terminate(swap, _State) ->
|
|
{emqx_alarm_handler, []};
|
|
terminate(_, _) ->
|
|
ok.
|