|
|
@@ -25,10 +25,10 @@
|
|
|
-export([
|
|
|
running/0,
|
|
|
lookup_state/1,
|
|
|
+ get_backend_status/2,
|
|
|
make_resource_id/1,
|
|
|
create_resource/3,
|
|
|
- update_resource/3,
|
|
|
- call/1
|
|
|
+ update_resource/3
|
|
|
]).
|
|
|
|
|
|
-export([
|
|
|
@@ -39,20 +39,21 @@
|
|
|
propagated_post_config_update/5
|
|
|
]).
|
|
|
|
|
|
--import(emqx_dashboard_sso, [provider/1]).
|
|
|
+-import(emqx_dashboard_sso, [provider/1, format/1]).
|
|
|
|
|
|
-define(MOD_TAB, emqx_dashboard_sso).
|
|
|
-define(MOD_KEY_PATH, [dashboard, sso]).
|
|
|
--define(CALL_TIMEOUT, timer:seconds(10)).
|
|
|
-define(MOD_KEY_PATH(Sub), [dashboard, sso, Sub]).
|
|
|
-define(RESOURCE_GROUP, <<"emqx_dashboard_sso">>).
|
|
|
+-define(NO_ERROR, <<>>).
|
|
|
-define(DEFAULT_RESOURCE_OPTS, #{
|
|
|
start_after_created => false
|
|
|
}).
|
|
|
|
|
|
-record(?MOD_TAB, {
|
|
|
backend :: atom(),
|
|
|
- state :: map()
|
|
|
+ state :: undefined | map(),
|
|
|
+ last_error = ?NO_ERROR :: term()
|
|
|
}).
|
|
|
|
|
|
%%------------------------------------------------------------------------------
|
|
|
@@ -62,17 +63,44 @@ start_link() ->
|
|
|
gen_server:start_link({local, ?MODULE}, ?MODULE, [], []).
|
|
|
|
|
|
running() ->
|
|
|
- maps:fold(
|
|
|
+ SSO = emqx:get_config(?MOD_KEY_PATH, #{}),
|
|
|
+ lists:filtermap(
|
|
|
fun
|
|
|
- (Type, #{enable := true}, Acc) ->
|
|
|
- [Type | Acc];
|
|
|
- (_Type, _Cfg, Acc) ->
|
|
|
- Acc
|
|
|
+ (#{backend := Backend, enable := true}) ->
|
|
|
+ case lookup(Backend) of
|
|
|
+ undefined ->
|
|
|
+ false;
|
|
|
+ #?MOD_TAB{last_error = ?NO_ERROR} ->
|
|
|
+ {true, Backend};
|
|
|
+ _ ->
|
|
|
+ false
|
|
|
+ end;
|
|
|
+ (_) ->
|
|
|
+ false
|
|
|
end,
|
|
|
- [],
|
|
|
- emqx:get_config(?MOD_KEY_PATH)
|
|
|
+ maps:values(SSO)
|
|
|
).
|
|
|
|
|
|
+get_backend_status(Backend, false) ->
|
|
|
+ #{
|
|
|
+ backend => Backend,
|
|
|
+ enable => false,
|
|
|
+ running => false,
|
|
|
+ last_error => ?NO_ERROR
|
|
|
+ };
|
|
|
+get_backend_status(Backend, _) ->
|
|
|
+ case lookup(Backend) of
|
|
|
+ undefined ->
|
|
|
+ #{
|
|
|
+ backend => Backend,
|
|
|
+ enable => true,
|
|
|
+ running => false,
|
|
|
+ last_error => <<"Resource not found">>
|
|
|
+ };
|
|
|
+ Data ->
|
|
|
+ maps:merge(#{backend => Backend, enable => true}, do_get_backend_status(Data))
|
|
|
+ end.
|
|
|
+
|
|
|
update(Backend, Config) ->
|
|
|
update_config(Backend, {?FUNCTION_NAME, Backend, Config}).
|
|
|
delete(Backend) ->
|
|
|
@@ -98,7 +126,7 @@ create_resource(ResourceId, Module, Config) ->
|
|
|
Config,
|
|
|
?DEFAULT_RESOURCE_OPTS
|
|
|
),
|
|
|
- start_resource_if_enabled(ResourceId, Result, Config, fun clean_when_start_failed/1).
|
|
|
+ start_resource_if_enabled(ResourceId, Result, Config).
|
|
|
|
|
|
update_resource(ResourceId, Module, Config) ->
|
|
|
Result = emqx_resource:recreate_local(
|
|
|
@@ -106,14 +134,6 @@ update_resource(ResourceId, Module, Config) ->
|
|
|
),
|
|
|
start_resource_if_enabled(ResourceId, Result, Config).
|
|
|
|
|
|
-call(Req) ->
|
|
|
- try
|
|
|
- gen_server:call(?MODULE, Req, ?CALL_TIMEOUT)
|
|
|
- catch
|
|
|
- exit:{timeout, _} ->
|
|
|
- {error, <<"Update backend failed: timeout">>}
|
|
|
- end.
|
|
|
-
|
|
|
%%------------------------------------------------------------------------------
|
|
|
%% gen_server callbacks
|
|
|
%%------------------------------------------------------------------------------
|
|
|
@@ -133,9 +153,6 @@ init([]) ->
|
|
|
start_backend_services(),
|
|
|
{ok, #{}}.
|
|
|
|
|
|
-handle_call({update_config, Req, NewConf}, _From, State) ->
|
|
|
- Result = on_config_update(Req, NewConf),
|
|
|
- {reply, Result, State};
|
|
|
handle_call(_Request, _From, State) ->
|
|
|
Reply = ok,
|
|
|
{reply, Reply, State}.
|
|
|
@@ -170,12 +187,14 @@ start_backend_services() ->
|
|
|
msg => "start_sso_backend_successfully",
|
|
|
backend => Backend
|
|
|
}),
|
|
|
- ets:insert(?MOD_TAB, #?MOD_TAB{backend = Backend, state = State});
|
|
|
+ update_state(Backend, State);
|
|
|
{error, Reason} ->
|
|
|
+ SafeReason = emqx_utils:redact(Reason),
|
|
|
+ update_last_error(Backend, SafeReason),
|
|
|
?SLOG(error, #{
|
|
|
msg => "start_sso_backend_failed",
|
|
|
backend => Backend,
|
|
|
- reason => Reason
|
|
|
+ reason => SafeReason
|
|
|
})
|
|
|
end
|
|
|
end,
|
|
|
@@ -183,39 +202,38 @@ start_backend_services() ->
|
|
|
).
|
|
|
|
|
|
update_config(Backend, UpdateReq) ->
|
|
|
+ %% we always make sure the valid configuration will update successfully,
|
|
|
+ %% ignore the runtime error during its update
|
|
|
case emqx_conf:update(?MOD_KEY_PATH(Backend), UpdateReq, #{override_to => cluster}) of
|
|
|
- {ok, UpdateResult} ->
|
|
|
- #{post_config_update := #{?MODULE := Result}} = UpdateResult,
|
|
|
- ?SLOG(info, #{
|
|
|
- msg => "update_sso_successfully",
|
|
|
- backend => Backend,
|
|
|
- result => Result
|
|
|
- }),
|
|
|
- Result;
|
|
|
- {error, Reason} ->
|
|
|
+ {ok, _UpdateResult} ->
|
|
|
+ case lookup(Backend) of
|
|
|
+ undefined ->
|
|
|
+ ok;
|
|
|
+ #?MOD_TAB{state = State, last_error = ?NO_ERROR} ->
|
|
|
+ {ok, State};
|
|
|
+ Data ->
|
|
|
+ {error, Data#?MOD_TAB.last_error}
|
|
|
+ end;
|
|
|
+ {error, Reason} = Error ->
|
|
|
+ SafeReason = emqx_utils:redact(Reason),
|
|
|
?SLOG(error, #{
|
|
|
msg => "update_sso_failed",
|
|
|
backend => Backend,
|
|
|
- reason => Reason
|
|
|
+ reason => SafeReason
|
|
|
}),
|
|
|
- {error,
|
|
|
- case Reason of
|
|
|
- {_Stage, _Mod, Reason2} ->
|
|
|
- Reason2;
|
|
|
- _ ->
|
|
|
- Reason
|
|
|
- end}
|
|
|
+ Error
|
|
|
end.
|
|
|
|
|
|
pre_config_update(_, {update, _Backend, Config}, _OldConf) ->
|
|
|
- {ok, Config};
|
|
|
+ {ok, maybe_write_certs(Config)};
|
|
|
pre_config_update(_, {delete, _Backend}, undefined) ->
|
|
|
throw(not_exists);
|
|
|
pre_config_update(_, {delete, _Backend}, _OldConf) ->
|
|
|
{ok, null}.
|
|
|
|
|
|
post_config_update(_, UpdateReq, NewConf, _OldConf, _AppEnvs) ->
|
|
|
- call({update_config, UpdateReq, NewConf}).
|
|
|
+ _ = on_config_update(UpdateReq, NewConf),
|
|
|
+ ok.
|
|
|
|
|
|
propagated_post_config_update(
|
|
|
?MOD_KEY_PATH(BackendBin) = Path, _UpdateReq, undefined, OldConf, AppEnvs
|
|
|
@@ -241,26 +259,30 @@ on_config_update({update, Backend, _RawConfig}, Config) ->
|
|
|
case lookup(Backend) of
|
|
|
undefined ->
|
|
|
on_backend_updated(
|
|
|
+ Backend,
|
|
|
emqx_dashboard_sso:create(Provider, Config),
|
|
|
fun(State) ->
|
|
|
- ets:insert(?MOD_TAB, #?MOD_TAB{backend = Backend, state = State})
|
|
|
+ update_state(Backend, State)
|
|
|
end
|
|
|
);
|
|
|
Data ->
|
|
|
+ update_last_error(Backend, ?NO_ERROR),
|
|
|
on_backend_updated(
|
|
|
+ Backend,
|
|
|
emqx_dashboard_sso:update(Provider, Config, Data#?MOD_TAB.state),
|
|
|
fun(State) ->
|
|
|
- ets:insert(?MOD_TAB, Data#?MOD_TAB{state = State})
|
|
|
+ update_state(Backend, State)
|
|
|
end
|
|
|
)
|
|
|
end;
|
|
|
on_config_update({delete, Backend}, _NewConf) ->
|
|
|
case lookup(Backend) of
|
|
|
undefined ->
|
|
|
- {error, not_exists};
|
|
|
+ on_backend_updated(Backend, {error, not_exists}, undefined);
|
|
|
Data ->
|
|
|
Provider = provider(Backend),
|
|
|
on_backend_updated(
|
|
|
+ Backend,
|
|
|
emqx_dashboard_sso:destroy(Provider, Data#?MOD_TAB.state),
|
|
|
fun() ->
|
|
|
ets:delete(?MOD_TAB, Backend)
|
|
|
@@ -276,15 +298,12 @@ lookup(Backend) ->
|
|
|
undefined
|
|
|
end.
|
|
|
|
|
|
-start_resource_if_enabled(ResourceId, Result, Config) ->
|
|
|
- start_resource_if_enabled(ResourceId, Result, Config, undefined).
|
|
|
-
|
|
|
-start_resource_if_enabled(
|
|
|
- ResourceId, {ok, _} = Result, #{enable := true}, CleanWhenStartFailed
|
|
|
-) ->
|
|
|
+%% to avoid resource leakage the resource start will never affect the update result,
|
|
|
+%% so the resource_id will always be recorded
|
|
|
+start_resource_if_enabled(ResourceId, {ok, _} = Result, #{enable := true, backend := Backend}) ->
|
|
|
case emqx_resource:start(ResourceId) of
|
|
|
ok ->
|
|
|
- Result;
|
|
|
+ ok;
|
|
|
{error, Reason} ->
|
|
|
SafeReason = emqx_utils:redact(Reason),
|
|
|
?SLOG(error, #{
|
|
|
@@ -292,31 +311,21 @@ start_resource_if_enabled(
|
|
|
resource_id => ResourceId,
|
|
|
reason => SafeReason
|
|
|
}),
|
|
|
- erlang:is_function(CleanWhenStartFailed) andalso
|
|
|
- CleanWhenStartFailed(ResourceId),
|
|
|
- {error, emqx_dashboard_sso:format(["Start backend failed, Reason: ", SafeReason])}
|
|
|
- end;
|
|
|
-start_resource_if_enabled(_ResourceId, Result, _Config, _) ->
|
|
|
+ update_last_error(Backend, SafeReason),
|
|
|
+ ok
|
|
|
+ end,
|
|
|
+ Result;
|
|
|
+start_resource_if_enabled(_ResourceId, Result, _Config) ->
|
|
|
Result.
|
|
|
|
|
|
-%% ensure the backend creation is atomic, clean the corresponding resource when necessary,
|
|
|
-%% when creating a backend fails, nothing will be inserted into the SSO table,
|
|
|
-%% thus the resources created by backend will leakage.
|
|
|
-%% Although we can treat start failure as successful,
|
|
|
-%% and insert the resource data into the SSO table,
|
|
|
-%% it may be strange for users: it succeeds, but can't be used.
|
|
|
-clean_when_start_failed(ResourceId) ->
|
|
|
- _ = emqx_resource:remove_local(ResourceId),
|
|
|
- ok.
|
|
|
-
|
|
|
-%% this first level `ok` is for emqx_config_handler, and the second level is for the caller
|
|
|
-on_backend_updated({ok, State} = Ok, Fun) ->
|
|
|
+on_backend_updated(_Backend, {ok, State} = Ok, Fun) ->
|
|
|
Fun(State),
|
|
|
- {ok, Ok};
|
|
|
-on_backend_updated(ok, Fun) ->
|
|
|
+ Ok;
|
|
|
+on_backend_updated(_Backend, ok, Fun) ->
|
|
|
Fun(),
|
|
|
- {ok, ok};
|
|
|
-on_backend_updated(Error, _) ->
|
|
|
+ ok;
|
|
|
+on_backend_updated(Backend, {error, Reason} = Error, _) ->
|
|
|
+ update_last_error(Backend, Reason),
|
|
|
Error.
|
|
|
|
|
|
bin(A) when is_atom(A) -> atom_to_binary(A, utf8);
|
|
|
@@ -331,3 +340,50 @@ add_handler() ->
|
|
|
|
|
|
remove_handler() ->
|
|
|
ok = emqx_conf:remove_handler(?MOD_KEY_PATH('?')).
|
|
|
+
|
|
|
+maybe_write_certs(#{<<"backend">> := Backend} = Conf) ->
|
|
|
+ Dir = certs_path(Backend),
|
|
|
+ Provider = provider(Backend),
|
|
|
+ emqx_dashboard_sso:convert_certs(Provider, Dir, Conf).
|
|
|
+
|
|
|
+certs_path(Backend) ->
|
|
|
+ filename:join(["sso", Backend]).
|
|
|
+
|
|
|
+update_state(Backend, State) ->
|
|
|
+ Data = ensure_backend_data(Backend),
|
|
|
+ ets:insert(?MOD_TAB, Data#?MOD_TAB{state = State}).
|
|
|
+
|
|
|
+update_last_error(Backend, LastError) ->
|
|
|
+ Data = ensure_backend_data(Backend),
|
|
|
+ ets:insert(?MOD_TAB, Data#?MOD_TAB{last_error = LastError}).
|
|
|
+
|
|
|
+ensure_backend_data(Backend) ->
|
|
|
+ case ets:lookup(?MOD_TAB, Backend) of
|
|
|
+ [Data] ->
|
|
|
+ Data;
|
|
|
+ [] ->
|
|
|
+ #?MOD_TAB{backend = Backend}
|
|
|
+ end.
|
|
|
+
|
|
|
+do_get_backend_status(#?MOD_TAB{state = #{resource_id := ResourceId}}) ->
|
|
|
+ case emqx_resource_manager:lookup(ResourceId) of
|
|
|
+ {ok, _Group, #{status := connected}} ->
|
|
|
+ #{running => true, last_error => ?NO_ERROR};
|
|
|
+ {ok, _Group, #{status := Status}} ->
|
|
|
+ #{
|
|
|
+ running => false,
|
|
|
+ last_error => format([<<"Resource not valid, status: ">>, Status])
|
|
|
+ };
|
|
|
+ {error, not_found} ->
|
|
|
+ #{
|
|
|
+ running => false,
|
|
|
+ last_error => <<"Resource not found">>
|
|
|
+ }
|
|
|
+ end;
|
|
|
+do_get_backend_status(#?MOD_TAB{last_error = ?NO_ERROR}) ->
|
|
|
+ #{running => true, last_error => ?NO_ERROR};
|
|
|
+do_get_backend_status(#?MOD_TAB{last_error = LastError}) ->
|
|
|
+ #{
|
|
|
+ running => false,
|
|
|
+ last_error => format([LastError])
|
|
|
+ }.
|