119 lines
3.3 KiB
Erlang
119 lines
3.3 KiB
Erlang
%%--------------------------------------------------------------------
|
|
%% Copyright (c) 2017-2022 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_stomp_impl).
|
|
|
|
-behaviour(emqx_gateway_impl).
|
|
|
|
-include_lib("emqx/include/logger.hrl").
|
|
-include_lib("emqx_gateway/include/emqx_gateway.hrl").
|
|
|
|
-import(
|
|
emqx_gateway_utils,
|
|
[
|
|
normalize_config/1,
|
|
start_listeners/4,
|
|
stop_listeners/2
|
|
]
|
|
).
|
|
|
|
%% APIs
|
|
-export([
|
|
reg/0,
|
|
unreg/0
|
|
]).
|
|
|
|
-export([
|
|
on_gateway_load/2,
|
|
on_gateway_update/3,
|
|
on_gateway_unload/2
|
|
]).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% APIs
|
|
%%--------------------------------------------------------------------
|
|
|
|
-spec reg() -> ok | {error, any()}.
|
|
reg() ->
|
|
RegistryOptions = [{cbkmod, ?MODULE}],
|
|
emqx_gateway_registry:reg(stomp, RegistryOptions).
|
|
|
|
-spec unreg() -> ok | {error, any()}.
|
|
unreg() ->
|
|
emqx_gateway_registry:unreg(stomp).
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% emqx_gateway_registry callbacks
|
|
%%--------------------------------------------------------------------
|
|
|
|
on_gateway_load(
|
|
_Gateway = #{
|
|
name := GwName,
|
|
config := Config
|
|
},
|
|
Ctx
|
|
) ->
|
|
Listeners = normalize_config(Config),
|
|
ModCfg = #{
|
|
frame_mod => emqx_stomp_frame,
|
|
chann_mod => emqx_stomp_channel
|
|
},
|
|
case
|
|
start_listeners(
|
|
Listeners, GwName, Ctx, ModCfg
|
|
)
|
|
of
|
|
{ok, ListenerPids} ->
|
|
%% FIXME: How to throw an exception to interrupt the restart logic ?
|
|
%% FIXME: Assign ctx to GwState
|
|
{ok, ListenerPids, _GwState = #{ctx => Ctx}};
|
|
{error, {Reason, Listener}} ->
|
|
throw(
|
|
{badconf, #{
|
|
key => listeners,
|
|
value => Listener,
|
|
reason => Reason
|
|
}}
|
|
)
|
|
end.
|
|
|
|
on_gateway_update(Config, Gateway, GwState = #{ctx := Ctx}) ->
|
|
GwName = maps:get(name, Gateway),
|
|
try
|
|
%% XXX: 1. How hot-upgrade the changes ???
|
|
%% XXX: 2. Check the New confs first before destroy old state???
|
|
on_gateway_unload(Gateway, GwState),
|
|
on_gateway_load(Gateway#{config => Config}, Ctx)
|
|
catch
|
|
Class:Reason:Stk ->
|
|
logger:error(
|
|
"Failed to update ~ts; "
|
|
"reason: {~0p, ~0p} stacktrace: ~0p",
|
|
[GwName, Class, Reason, Stk]
|
|
),
|
|
{error, Reason}
|
|
end.
|
|
|
|
on_gateway_unload(
|
|
_Gateway = #{
|
|
name := GwName,
|
|
config := Config
|
|
},
|
|
_GwState
|
|
) ->
|
|
Listeners = normalize_config(Config),
|
|
stop_listeners(GwName, Listeners).
|