85 lines
2.9 KiB
Erlang
85 lines
2.9 KiB
Erlang
%%%-------------------------------------------------------------------
|
|
%%% @author anlicheng
|
|
%%% @copyright (C) 2025, <COMPANY>
|
|
%%% @doc
|
|
%%%
|
|
%%% @end
|
|
%%% Created : 18. 4月 2025 16:42
|
|
%%%-------------------------------------------------------------------
|
|
-module(efka_service_sup).
|
|
-author("anlicheng").
|
|
-include("efka_tables.hrl").
|
|
|
|
-behaviour(supervisor).
|
|
|
|
%% API
|
|
-export([start_link/0]).
|
|
-export([start_service/1, stop_service/1]).
|
|
|
|
%% Supervisor callbacks
|
|
-export([init/1]).
|
|
|
|
-define(SERVER, ?MODULE).
|
|
|
|
%%%===================================================================
|
|
%%% API functions
|
|
%%%===================================================================
|
|
|
|
%% @doc Starts the supervisor
|
|
-spec(start_link() -> {ok, Pid :: pid()} | ignore | {error, Reason :: term()}).
|
|
start_link() ->
|
|
supervisor:start_link({local, ?SERVER}, ?MODULE, []).
|
|
|
|
%%%===================================================================
|
|
%%% Supervisor callbacks
|
|
%%%===================================================================
|
|
|
|
%% @private
|
|
%% @doc Whenever a supervisor is started using supervisor:start_link/[2,3],
|
|
%% this function is called by the new process to find out about
|
|
%% restart strategy, maximum restart frequency and child
|
|
%% specifications.
|
|
init([]) ->
|
|
SupFlags = #{strategy => one_for_one, intensity => 1000, period => 3600},
|
|
%% 简化逻辑，只启动需要运行的微服务
|
|
{ok, Services} = service_model:get_running_services(),
|
|
ServiceIds = lists:map(fun(#service{service_id = ServiceId}) -> ServiceId end, Services),
|
|
lager:debug("[efka_service_sup] will start services: ~p", [ServiceIds]),
|
|
|
|
Specs = lists:map(fun(ServiceId) -> child_spec(ServiceId) end, Services),
|
|
|
|
{ok, {SupFlags, Specs}}.
|
|
|
|
%%%===================================================================
|
|
%%% Internal functions
|
|
%%%===================================================================
|
|
|
|
-spec start_service(ServiceId :: binary()) -> {ok, Pid :: pid()} | {error, Reason :: any()}.
|
|
start_service(ServiceId) when is_binary(ServiceId) ->
|
|
case supervisor:start_child(?MODULE, child_spec(ServiceId)) of
|
|
{ok, Pid} when is_pid(Pid) ->
|
|
{ok, Pid};
|
|
{error, {'already_started', Pid}} when is_pid(Pid) ->
|
|
{ok, Pid};
|
|
{error, Error} ->
|
|
{error, Error}
|
|
end.
|
|
|
|
-spec stop_service(ServiceId :: binary()) -> ok.
|
|
stop_service(ServiceId) when is_binary(ServiceId) ->
|
|
ChildId = efka_service:get_name(ServiceId),
|
|
supervisor:terminate_child(?MODULE, ChildId),
|
|
supervisor:delete_child(?MODULE, ChildId).
|
|
|
|
child_spec(#service{service_id = ServiceId}) when is_binary(ServiceId) ->
|
|
child_spec(ServiceId);
|
|
child_spec(ServiceId) when is_binary(ServiceId) ->
|
|
Name = efka_service:get_name(ServiceId),
|
|
#{
|
|
id => Name,
|
|
start => {efka_service, start_link, [Name, ServiceId]},
|
|
restart => permanent,
|
|
shutdown => 5000,
|
|
type => worker,
|
|
modules => ['efka_service']
|
|
}. |