You've already forked qlg.tsgz.moe
Init Repo
This commit is contained in:
root
460
vendor/swoole/swoole_process_pool.c
vendored
Executable file
460
vendor/swoole/swoole_process_pool.c
vendored
Executable file
@ -0,0 +1,460 @@
|
||||
/*
|
||||
+----------------------------------------------------------------------+
|
||||
| Swoole |
|
||||
+----------------------------------------------------------------------+
|
||||
| This source file is subject to version 2.0 of the Apache license, |
|
||||
| that is bundled with this package in the file LICENSE, and is |
|
||||
| available through the world-wide-web at the following url: |
|
||||
| http://www.apache.org/licenses/LICENSE-2.0.html |
|
||||
| If you did not receive a copy of the Apache2.0 license and are unable|
|
||||
| to obtain it through the world-wide-web, please send a note to |
|
||||
| license@swoole.com so we can mail you a copy immediately. |
|
||||
+----------------------------------------------------------------------+
|
||||
| Author: Tianfeng Han <mikan.tenny@gmail.com> |
|
||||
+----------------------------------------------------------------------+
|
||||
*/
|
||||
|
||||
#include "php_swoole.h"
|
||||
|
||||
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_process_pool_void, 0, 0, 0)
|
||||
ZEND_END_ARG_INFO()
|
||||
|
||||
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_process_pool_construct, 0, 0, 1)
|
||||
ZEND_ARG_INFO(0, worker_num)
|
||||
ZEND_ARG_INFO(0, ipc_type)
|
||||
ZEND_ARG_INFO(0, msgqueue_key)
|
||||
ZEND_END_ARG_INFO()
|
||||
|
||||
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_process_pool_on, 0, 0, 2)
|
||||
ZEND_ARG_INFO(0, event_name)
|
||||
ZEND_ARG_INFO(0, callback)
|
||||
ZEND_END_ARG_INFO()
|
||||
|
||||
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_process_pool_listen, 0, 0, 1)
|
||||
ZEND_ARG_INFO(0, host)
|
||||
ZEND_ARG_INFO(0, port)
|
||||
ZEND_ARG_INFO(0, backlog)
|
||||
ZEND_END_ARG_INFO()
|
||||
|
||||
ZEND_BEGIN_ARG_INFO_EX(arginfo_swoole_process_pool_write, 0, 0, 1)
|
||||
ZEND_ARG_INFO(0, data)
|
||||
ZEND_END_ARG_INFO()
|
||||
|
||||
static PHP_METHOD(swoole_process_pool, __construct);
|
||||
static PHP_METHOD(swoole_process_pool, __destruct);
|
||||
static PHP_METHOD(swoole_process_pool, on);
|
||||
static PHP_METHOD(swoole_process_pool, listen);
|
||||
static PHP_METHOD(swoole_process_pool, write);
|
||||
static PHP_METHOD(swoole_process_pool, start);
|
||||
|
||||
static const zend_function_entry swoole_process_pool_methods[] =
|
||||
{
|
||||
PHP_ME(swoole_process_pool, __construct, arginfo_swoole_process_pool_construct, ZEND_ACC_PUBLIC | ZEND_ACC_CTOR)
|
||||
PHP_ME(swoole_process_pool, __destruct, arginfo_swoole_process_pool_void, ZEND_ACC_PUBLIC | ZEND_ACC_DTOR)
|
||||
PHP_ME(swoole_process_pool, on, arginfo_swoole_process_pool_on, ZEND_ACC_PUBLIC)
|
||||
PHP_ME(swoole_process_pool, listen, arginfo_swoole_process_pool_listen, ZEND_ACC_PUBLIC)
|
||||
PHP_ME(swoole_process_pool, write, arginfo_swoole_process_pool_write, ZEND_ACC_PUBLIC)
|
||||
PHP_ME(swoole_process_pool, start, arginfo_swoole_process_pool_void, ZEND_ACC_PUBLIC)
|
||||
PHP_FE_END
|
||||
};
|
||||
|
||||
typedef struct
|
||||
{
|
||||
zval *onWorkerStart;
|
||||
zval *onWorkerStop;
|
||||
zval *onMessage;
|
||||
zval _onWorkerStart;
|
||||
zval _onWorkerStop;
|
||||
zval _onMessage;
|
||||
} process_pool_property;
|
||||
|
||||
static zend_class_entry swoole_process_pool_ce;
|
||||
static zend_class_entry *swoole_process_pool_class_entry_ptr;
|
||||
static swProcessPool *current_pool;
|
||||
|
||||
void swoole_process_pool_init(int module_number TSRMLS_DC)
|
||||
{
|
||||
SWOOLE_INIT_CLASS_ENTRY(swoole_process_pool_ce, "swoole_process_pool", "Swoole\\Process\\Pool", swoole_process_pool_methods);
|
||||
swoole_process_pool_class_entry_ptr = zend_register_internal_class(&swoole_process_pool_ce TSRMLS_CC);
|
||||
SWOOLE_CLASS_ALIAS(swoole_process_pool, "Swoole\\Process\\Pool");
|
||||
}
|
||||
|
||||
static void php_swoole_process_pool_onWorkerStart(swProcessPool *pool, int worker_id)
|
||||
{
|
||||
SWOOLE_GET_TSRMLS;
|
||||
|
||||
zval *zobject = (zval *) pool->ptr;
|
||||
zval *zworker_id;
|
||||
zval *retval = NULL;
|
||||
|
||||
SW_MAKE_STD_ZVAL(zworker_id);
|
||||
ZVAL_LONG(zworker_id, worker_id);
|
||||
|
||||
zval **args[2];
|
||||
args[0] = &zobject;
|
||||
args[1] = &zworker_id;
|
||||
|
||||
process_pool_property *pp = swoole_get_property(zobject, 0);
|
||||
if (pp->onWorkerStart == NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
if (SwooleG.main_reactor)
|
||||
{
|
||||
SwooleG.main_reactor->free(SwooleG.main_reactor);
|
||||
SwooleG.main_reactor = NULL;
|
||||
swTraceLog(SW_TRACE_PHP, "destroy reactor");
|
||||
}
|
||||
if (sw_call_user_function_ex(EG(function_table), NULL, pp->onWorkerStart, &retval, 2, args, 0, NULL TSRMLS_CC) == FAILURE)
|
||||
{
|
||||
swoole_php_fatal_error(E_WARNING, "onWorkerStart handler error.");
|
||||
}
|
||||
if (EG(exception))
|
||||
{
|
||||
zend_exception_error(EG(exception), E_ERROR TSRMLS_CC);
|
||||
}
|
||||
if (retval)
|
||||
{
|
||||
sw_zval_ptr_dtor(&retval);
|
||||
}
|
||||
if (SwooleG.main_reactor)
|
||||
{
|
||||
php_swoole_event_wait();
|
||||
SwooleG.running = 0;
|
||||
}
|
||||
}
|
||||
|
||||
static void php_swoole_process_pool_onMessage(swProcessPool *pool, char *data, uint32_t length)
|
||||
{
|
||||
SWOOLE_GET_TSRMLS;
|
||||
|
||||
zval *zobject = (zval *) pool->ptr;
|
||||
zval *zdata;
|
||||
zval *retval;
|
||||
|
||||
SW_MAKE_STD_ZVAL(zdata);
|
||||
SW_ZVAL_STRINGL(zdata, data, length, 1);
|
||||
|
||||
zval **args[2];
|
||||
args[0] = &zobject;
|
||||
args[1] = &zdata;
|
||||
|
||||
process_pool_property *pp = swoole_get_property(zobject, 0);
|
||||
|
||||
if (sw_call_user_function_ex(EG(function_table), NULL, pp->onMessage, &retval, 2, args, 0, NULL TSRMLS_CC) == FAILURE)
|
||||
{
|
||||
swoole_php_fatal_error(E_WARNING, "onWorkerStart handler error.");
|
||||
}
|
||||
if (EG(exception))
|
||||
{
|
||||
zend_exception_error(EG(exception), E_ERROR TSRMLS_CC);
|
||||
}
|
||||
sw_zval_ptr_dtor(&zdata);
|
||||
if (retval)
|
||||
{
|
||||
sw_zval_ptr_dtor(&retval);
|
||||
}
|
||||
}
|
||||
|
||||
static void php_swoole_process_pool_onWorkerStop(swProcessPool *pool, int worker_id)
|
||||
{
|
||||
SWOOLE_GET_TSRMLS;
|
||||
|
||||
zval *zobject = (zval *) pool->ptr;
|
||||
zval *zworker_id;
|
||||
zval *retval = NULL;
|
||||
|
||||
SW_MAKE_STD_ZVAL(zworker_id);
|
||||
ZVAL_LONG(zworker_id, worker_id);
|
||||
|
||||
zval **args[2];
|
||||
args[0] = &zobject;
|
||||
args[1] = &zworker_id;
|
||||
|
||||
process_pool_property *pp = swoole_get_property(zobject, 0);
|
||||
if (pp->onWorkerStop == NULL)
|
||||
{
|
||||
return;
|
||||
}
|
||||
if (sw_call_user_function_ex(EG(function_table), NULL, pp->onWorkerStop, &retval, 2, args, 0, NULL TSRMLS_CC) == FAILURE)
|
||||
{
|
||||
swoole_php_fatal_error(E_WARNING, "onWorkerStop handler error.");
|
||||
}
|
||||
if (EG(exception))
|
||||
{
|
||||
zend_exception_error(EG(exception), E_ERROR TSRMLS_CC);
|
||||
}
|
||||
if (retval)
|
||||
{
|
||||
sw_zval_ptr_dtor(&retval);
|
||||
}
|
||||
}
|
||||
|
||||
static void php_swoole_process_pool_signal_hanlder(int sig)
|
||||
{
|
||||
switch (sig)
|
||||
{
|
||||
case SIGTERM:
|
||||
SwooleG.running = 0;
|
||||
break;
|
||||
case SIGUSR1:
|
||||
case SIGUSR2:
|
||||
current_pool->reloading = 1;
|
||||
current_pool->reload_init = 0;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
static PHP_METHOD(swoole_process_pool, __construct)
|
||||
{
|
||||
long worker_num;
|
||||
long ipc_type = SW_IPC_NONE;
|
||||
long msgq_key = 0;
|
||||
|
||||
//only cli env
|
||||
if (!SWOOLE_G(cli))
|
||||
{
|
||||
swoole_php_fatal_error(E_ERROR, "swoole_process_pool only can be used in PHP CLI mode.");
|
||||
RETURN_FALSE;
|
||||
}
|
||||
|
||||
if (SwooleG.serv)
|
||||
{
|
||||
swoole_php_fatal_error(E_ERROR, "swoole_process_pool cannot use in server process.");
|
||||
RETURN_FALSE;
|
||||
}
|
||||
|
||||
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "l|ll", &worker_num, &ipc_type, &msgq_key) == FAILURE)
|
||||
{
|
||||
RETURN_FALSE;
|
||||
}
|
||||
|
||||
if (worker_num <= 0)
|
||||
{
|
||||
zend_throw_exception_ex(swoole_exception_class_entry_ptr, errno TSRMLS_CC, "invalid worker_num");
|
||||
RETURN_FALSE;
|
||||
}
|
||||
|
||||
swProcessPool *pool = emalloc(sizeof(swProcessPool));
|
||||
if (swProcessPool_create(pool, worker_num, 0, (key_t) msgq_key, ipc_type) < 0)
|
||||
{
|
||||
zend_throw_exception_ex(swoole_exception_class_entry_ptr, errno TSRMLS_CC, "failed to create process pool");
|
||||
RETURN_FALSE;
|
||||
}
|
||||
|
||||
if (ipc_type > 0)
|
||||
{
|
||||
if (swProcessPool_set_protocol(pool, 0, SW_BUFFER_INPUT_SIZE) < 0)
|
||||
{
|
||||
zend_throw_exception_ex(swoole_exception_class_entry_ptr, errno TSRMLS_CC, "failed to create process pool");
|
||||
RETURN_FALSE;
|
||||
}
|
||||
}
|
||||
|
||||
pool->ptr = sw_zval_dup(getThis());
|
||||
|
||||
process_pool_property *pp = emalloc(sizeof(process_pool_property));
|
||||
bzero(pp, sizeof(process_pool_property));
|
||||
swoole_set_property(getThis(), 0, pp);
|
||||
swoole_set_object(getThis(), pool);
|
||||
}
|
||||
|
||||
static PHP_METHOD(swoole_process_pool, on)
|
||||
{
|
||||
char *name;
|
||||
zend_size_t l_name;
|
||||
zval *callback;
|
||||
|
||||
swProcessPool *pool = swoole_get_object(getThis());
|
||||
|
||||
if (pool->started > 0)
|
||||
{
|
||||
swoole_php_fatal_error(E_WARNING, "process pool is started. unable to register event callback function.");
|
||||
RETURN_FALSE;
|
||||
}
|
||||
|
||||
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "sz", &name, &l_name, &callback) == FAILURE)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
if (!php_swoole_is_callable(callback))
|
||||
{
|
||||
RETURN_FALSE;
|
||||
}
|
||||
|
||||
process_pool_property *pp = swoole_get_property(getThis(), 0);
|
||||
|
||||
if (strncasecmp("WorkerStart", name, l_name) == 0)
|
||||
{
|
||||
if (pp->onWorkerStart)
|
||||
{
|
||||
sw_zval_ptr_dtor(&pp->onWorkerStart);
|
||||
}
|
||||
pp->onWorkerStart = callback;
|
||||
sw_zval_add_ref(&callback);
|
||||
sw_copy_to_stack(pp->onWorkerStart, pp->_onWorkerStart);
|
||||
RETURN_TRUE;
|
||||
}
|
||||
else if (strncasecmp("Message", name, l_name) == 0)
|
||||
{
|
||||
if (pool->ipc_mode == SW_IPC_NONE)
|
||||
{
|
||||
swoole_php_fatal_error(E_WARNING, "cannot set onMessage event with ipc_type=0.");
|
||||
RETURN_TRUE;
|
||||
}
|
||||
if (pp->onMessage)
|
||||
{
|
||||
sw_zval_ptr_dtor(&pp->onMessage);
|
||||
}
|
||||
pp->onMessage = callback;
|
||||
sw_zval_add_ref(&callback);
|
||||
sw_copy_to_stack(pp->onMessage, pp->_onMessage);
|
||||
RETURN_TRUE;
|
||||
}
|
||||
else if (strncasecmp("WorkerStop", name, l_name) == 0)
|
||||
{
|
||||
if (pp->onWorkerStop)
|
||||
{
|
||||
sw_zval_ptr_dtor(&pp->onWorkerStop);
|
||||
}
|
||||
pp->onWorkerStop = callback;
|
||||
sw_zval_add_ref(&callback);
|
||||
sw_copy_to_stack(pp->onWorkerStop, pp->_onWorkerStop);
|
||||
RETURN_TRUE;
|
||||
}
|
||||
else
|
||||
{
|
||||
swoole_php_error(E_WARNING, "unknown event type[%s]", name);
|
||||
RETURN_TRUE;
|
||||
}
|
||||
}
|
||||
|
||||
static PHP_METHOD(swoole_process_pool, listen)
|
||||
{
|
||||
char *host;
|
||||
zend_size_t l_host;
|
||||
long port;
|
||||
long backlog = 2048;
|
||||
|
||||
swProcessPool *pool = swoole_get_object(getThis());
|
||||
|
||||
if (pool->started > 0)
|
||||
{
|
||||
swoole_php_fatal_error(E_WARNING, "process pool is started. unable to listen.");
|
||||
RETURN_FALSE;
|
||||
}
|
||||
|
||||
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s|ll", &host, &l_host, &port, &backlog) == FAILURE)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
if (pool->ipc_mode != SW_IPC_SOCKET)
|
||||
{
|
||||
swoole_php_fatal_error(E_WARNING, "unsupported ipc type[%d].", pool->ipc_mode);
|
||||
RETURN_FALSE;
|
||||
}
|
||||
|
||||
SwooleG.reuse_port = 0;
|
||||
int ret;
|
||||
//unix socket
|
||||
if (strncasecmp("unix:/", host, 6) == 0)
|
||||
{
|
||||
ret = swProcessPool_create_unix_socket(pool, host + 5, backlog);
|
||||
}
|
||||
else
|
||||
{
|
||||
ret = swProcessPool_create_tcp_socket(pool, host, port, backlog);
|
||||
}
|
||||
SW_CHECK_RETURN(ret);
|
||||
}
|
||||
|
||||
static PHP_METHOD(swoole_process_pool, write)
|
||||
{
|
||||
char *data;
|
||||
zend_size_t length;
|
||||
|
||||
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &data, &length) == FAILURE)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
swProcessPool *pool = swoole_get_object(getThis());
|
||||
if (pool->ipc_mode != SW_IPC_SOCKET)
|
||||
{
|
||||
swoole_php_fatal_error(E_WARNING, "unsupported ipc type[%d].", pool->ipc_mode);
|
||||
RETURN_FALSE;
|
||||
}
|
||||
if (length == 0)
|
||||
{
|
||||
RETURN_FALSE;
|
||||
}
|
||||
SW_CHECK_RETURN(swProcessPool_response(pool, data, length));
|
||||
}
|
||||
|
||||
static PHP_METHOD(swoole_process_pool, start)
|
||||
{
|
||||
swProcessPool *pool = swoole_get_object(getThis());
|
||||
if (pool->started)
|
||||
{
|
||||
swoole_php_fatal_error(E_WARNING, "process pool is started. unable to execute swoole_process_pool->start.");
|
||||
RETURN_FALSE;
|
||||
}
|
||||
|
||||
process_pool_property *pp = swoole_get_property(getThis(), 0);
|
||||
|
||||
SwooleG.use_signalfd = 0;
|
||||
|
||||
swSignal_add(SIGTERM, php_swoole_process_pool_signal_hanlder);
|
||||
|
||||
if (pool->ipc_mode > SW_IPC_NONE)
|
||||
{
|
||||
pool->onMessage = php_swoole_process_pool_onMessage;
|
||||
}
|
||||
else
|
||||
{
|
||||
if (pp->onWorkerStart == NULL)
|
||||
{
|
||||
swoole_php_fatal_error(E_ERROR, "require onWorkerStart callback");
|
||||
RETURN_FALSE;
|
||||
}
|
||||
}
|
||||
|
||||
pool->onWorkerStart = php_swoole_process_pool_onWorkerStart;
|
||||
pool->onWorkerStop = php_swoole_process_pool_onWorkerStop;
|
||||
|
||||
if (swProcessPool_start(pool) < 0)
|
||||
{
|
||||
RETURN_FALSE;
|
||||
}
|
||||
|
||||
current_pool = pool;
|
||||
|
||||
swProcessPool_wait(pool);
|
||||
swProcessPool_shutdown(pool);
|
||||
}
|
||||
|
||||
static PHP_METHOD(swoole_process_pool, __destruct)
|
||||
{
|
||||
swProcessPool *pool = swoole_get_object(getThis());
|
||||
sw_zval_free(pool->ptr);
|
||||
efree(pool);
|
||||
|
||||
process_pool_property *pp = swoole_get_property(getThis(), 0);
|
||||
if (pp->onWorkerStart)
|
||||
{
|
||||
sw_zval_ptr_dtor(&pp->onWorkerStart);
|
||||
}
|
||||
if (pp->onMessage)
|
||||
{
|
||||
sw_zval_ptr_dtor(&pp->onMessage);
|
||||
}
|
||||
if (pp->onWorkerStop)
|
||||
{
|
||||
sw_zval_ptr_dtor(&pp->onWorkerStop);
|
||||
}
|
||||
efree(pp);
|
||||
}
|
||||
Reference in New Issue
Block a user