Erlcron 源码分析（2）

上一篇介绍了如何使用，这次看看具体实现

用github将项目拉到本地，看下项目具体的结构

可以看到项目是rebar3管理，遵循opt原则，集成travis ci 做集成测试。

项目启动

ecrn_app.erl 运行start/2

start(_StartType, _StartArgs) ->
    case ecrn_sup:start_link() of
        {ok, Pid} ->
            setup(),
            {ok, Pid};
        Error ->
            Error
    end.

ecrn_sup.erl 执行start_link/0

start_link() ->
    supervisor:start_link({local, ?SERVER}, ?MODULE, []).

%%%===================================================================
%%% Supervisor callbacks
%%%===================================================================

%% @private
init([]) ->
    RestartStrategy = one_for_one,
    MaxRestarts = 3,
    MaxSecondsBetweenRestarts = 10,
    SupFlags = {RestartStrategy,
                MaxRestarts,
                MaxSecondsBetweenRestarts},

    ChildSup =  {ecrn_cron_sup, {ecrn_cron_sup, start_link, []},
                 permanent, 1000, supervisor, [ecrn_cron_sup]},
    RegistrationServer  =  {ecrn_reg_server, {ecrn_reg, start_link, []},
                            permanent, 1000, worker, [ecrn_reg]},
    BroadcastServer  =  {ecrn_control, {ecrn_control, start_link, []},
                         permanent, 1000, worker, [ecrn_control]},


    {ok, {SupFlags, [ChildSup, RegistrationServer, BroadcastServer]}}.

查看源码可以看出ecrn_app:start_link() 分别启动了三个子进程分别是
ecrn_cron_sup、ecrn_reg_server、ecrn_control
3.三个启动模块

我们先看ecrn_cron_sup启动都做了些什么

start_link() ->
    supervisor:start_link({local, ?SERVER}, ?MODULE, []).
init([]) ->
    RestartStrategy = simple_one_for_one,
    MaxRestarts = 1000,
    MaxSecondsBetweenRestarts = 3600,

    SupFlags = {RestartStrategy, MaxRestarts, MaxSecondsBetweenRestarts},

    Restart = transient,
    Shutdown = 2000,
    Type = worker,

    AChild = {ecrn_agent, {ecrn_agent, start_link, []},
              Restart, Shutdown, Type, [ecrn_agent]},

    {ok, {SupFlags, [AChild]}}.

又去启动了ecrn_agent，真是一层套一层，不过这也遵循了otp原则，这里可以看到启动策略使用了simple_one_for_one，需要时创建子进程，初步考虑应当就是任务执行的最终模块了

那接着我们再去看第二个启动的ecrn_reg

-behaviour(gen_server).

start_link() ->
    gen_server:start_link({local, ?SERVER}, ?MODULE, [], []).

init([]) ->
    {ok, #state{registered=dict:new()}}.

一个gen_server模块，创建了一个进程字典，不过先不管他，看最后一个启动的模块

ecrn_control

-behaviour(gen_server).

start_link() ->
    gen_server:start_link({local, ?SERVER}, ?MODULE, [], []).

init([]) ->
    DateTime = erlang:localtime(),
    {ok, #state{reference_datetime=DateTime,
                datetime_at_reference=ecrn_util:epoch_seconds()}}.

也是一个gen_server 模块，state里放了启动时间，不过是两种不同格式

3.三个进程启动完了后在ecrn_app.erl 执行setup()

setup() ->
    case application:get_env(erlcron, crontab) of 
        {ok, Crontab} ->
            lists:foreach(fun(CronJob) ->
                erlcron:cron(CronJob) 
            end, Crontab);
        undefined ->
            ok
    end.

这里看到用了一个lists:foreach/2 循环执行在sys.config里配置的需要定时执行的任务。

具体如何做到循环执行的就要看erlcron:cron/1是如何处理列表里job的了

Erlcron 源码分析（2）

项目启动

推荐阅读更多精彩内容