Swoole源码学习记录(十四)――Server模块详解(下)(二)

2015-01-27 14:07:47 · 作者: · 浏览: 364
{ swWarn(onTimer is null. Can not use timer.); return SW_ERR; } //timer no init if (SwooleG.timer.fd == 0) { if (swTimer_create(&SwooleG.timer, interval, SwooleG.use_timer_pipe) < 0) { return SW_ERR; } if (swIsMaster()) { serv->connection_list[SW_SERVER_TIMER_FD_INDEX].fd = SwooleG.timer.fd; } if (SwooleG.use_timer_pipe) { SwooleG.main_reactor->setHandle(SwooleG.main_reactor, SW_FD_TIMER, swTimer_event_handler); SwooleG.main_reactor->add(SwooleG.main_reactor, SwooleG.timer.fd, SW_FD_TIMER); } SwooleG.timer.onTimer = swServer_onTimer; } return swTimer_add(&SwooleG.timer, interval);

源码解释: 首先判定是否有onTimer回调,如果没有则返回一个Error。随后,如果没有初始化timer计时器,则调用swTimer_create函数创建计时器。如果当前进程是master进程,将timer的fd添加到connection_list中。如果timer指定使用了pipe管道,则将timer的fd添加到SwooleG的main_reactor中。最后,调用swTimer_add添加timer。

swServer_tcp_send

该函数用于发送TCP数据。函数原型如下:

// Server.h 443h
int swServer_tcp_send(swServer *serv, int fd, void *data, int length);
参数 说明
swServer *serv swServer对象
int fd 发送的socket描述符
void *data 需要发送的数据
int length 数据长度

函数核心源码:

    // Server.c 788-858h
    swSendData _send;
    swFactory *factory = &(serv->factory);

#ifndef SW_WORKER_SEND_CHUNK
    /**
     * More than the output buffer
     */
    if (length >= serv->buffer_output_size)
    {
        swWarn(More than the output buffer size[%d], please use the sendfile., serv->buffer_output_size);
        return SW_ERR;
    }
    else
    {
        _send.info.fd = fd;
        _send.info.type = SW_EVENT_TCP;
        _send.data = data;

        if (length >= SW_BUFFER_SIZE)
        {
            _send.length = length;
        }
        else
        {
            _send.info.len = length;
            _send.length = 0;
        }
        return factory->finish(factory, &_send);
    }
#else
    char buffer[SW_BUFFER_SIZE];
    int trunk_num = (length / SW_BUFFER_SIZE) + 1;
    int send_n = 0, i, ret;

    swConnection *conn = swServer_connection_get(serv, fd);
    if (conn == NULL || conn->active == 0)
    {
        swWarn(Connection[%d] has been closed., fd);
        return SW_ERR;
    }

    for (i = 0; i < trunk_num; i++)
    {
        //last chunk
        if (i == (trunk_num - 1))
        {
            send_n = length % SW_BUFFER_SIZE;
            if (send_n == 0)
                break;
        }
        else
        {
            send_n = SW_BUFFER_SIZE;
        }
        memcpy(buffer, data + SW_BUFFER_SIZE * i, send_n);
        _send.info.len = send_n;
        ret = factory->finish(factory, &_send);

#ifdef SW_WORKER_SENDTO_YIELD
        if ((i % SW_WORKER_SENDTO_YIELD) == (SW_WORKER_SENDTO_YIELD - 1))
        {
            swYield();
        }
#endif
    }
    return ret;
#endif

源码解释: 如果没有定义SW_WORKER_SEND_CHUNK宏,执行如下操作: 如果数据长度大于swServer的输出缓存大小,则报错。否则,设置swSendData的相关属性,调用swServer中的factory的finish函数将数据发出。 如果定义了SW_WORKER_SEND_CHUNK宏,执行如下操作: 首先根据数据长度length计算出需要多少个trunk。随后,获取fd对应的swConnecton,如果找不到connection或者connection已经关闭,则报错。随后,将数据划分为一个个trunk的长度,放进swSendData中后调用swServer中的factory的finish函数将数据发出。

swServer_reload

该函数用于通知Manager进程重启全部的Worker进程。函数原型如下:

// Server.h 443h
int swServer_reload(swServer *serv);
参数 说明
swServer *serv swServer对象

函数核心源码:

    // Server.c 1009-1017h
    int manager_pid = swServer_get_manager_pid(serv);
    if (manager_pid > 0)
    {
        return kill(manager_pid, SIGUSR1);
    }
    return SW_ERR;

源码解释: 通过kill函数向Manager进程发送SIGUSR1信号,该信号的行为是杀死并重启全部的Worker。

剩下的swServer的操作函数较为简单,在此不再贴出源码进行分析。

2.Server相关结构体分析

下面分析一些声明在Server.h中的结构体。

swPackage

该结构体已被swEventData替代。

swPackage_task

声明:

// Server.h 420-424h
typedef struct
{
    int length;
    char tmpfile[sizeof(SW_TASK_TMP_FILE)];
} swPackage_task;
成员 说明
int length 数据长度
char tmpfile[sizeof(SW_TASK_TMP_FILE)] 临时文件的文件名

说明: swPackage_task用于封装内容较大的task包(超过8K),tmpfile指向一个由mkstemp函数(如果开启了HAVE_MKOSTEMP选项,则为mkostemp函数)创造的临时文件,所有的数据会被暂时存放在这个文件里。

swPackage_response

声明:

// Server.h 426-430h
typedef struct
{
    int le