clickhouse.cpp

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include "php.h"
#include "php_clickhouse.h"
#include "bridge.h"

#include "ext/standard/info.h"
#include "Zend/zend_API.h"

extern "C" {

int clickhouse_obj_res_num; 
	
ZEND_BEGIN_ARG_INFO_EX(arginfo___construct, 0, 0, 0)
	ZEND_ARG_INFO(0, host)
	ZEND_ARG_INFO(0, username)
	ZEND_ARG_INFO(0, password)
	ZEND_ARG_INFO(0, default_database)
	ZEND_ARG_INFO(0, port)
	ZEND_ARG_INFO(0, compression)
ZEND_END_ARG_INFO()	
		
ZEND_BEGIN_ARG_INFO_EX(arginfo_select, 0, 0, 2)
	ZEND_ARG_INFO(0, query)
	ZEND_ARG_INFO(0, callback)
ZEND_END_ARG_INFO()	
		
ZEND_BEGIN_ARG_INFO_EX(arginfo_execute, 0, 0, 1)
	ZEND_ARG_INFO(0, query)
ZEND_END_ARG_INFO()	
		
ZEND_BEGIN_ARG_INFO_EX(arginfo_ping, 0, 0, 0)
ZEND_END_ARG_INFO()	
		
ZEND_BEGIN_ARG_INFO_EX(arginfo_resetConnection, 0, 0, 0)
ZEND_END_ARG_INFO()	
		
zend_class_entry *clickhouse_ce;	

zend_function_entry clickhouse_functions[] = {
    PHP_ME(ClickHouse, __construct, arginfo___construct, ZEND_ACC_PUBLIC)
	PHP_ME(ClickHouse, select, arginfo_select, ZEND_ACC_PUBLIC)
	PHP_ME(ClickHouse, execute, arginfo_execute, ZEND_ACC_PUBLIC)
	PHP_ME(ClickHouse, ping, arginfo_ping, ZEND_ACC_PUBLIC)
	PHP_ME(ClickHouse, resetConnection, arginfo_resetConnection, ZEND_ACC_PUBLIC)
    PHP_FE_END
};

zend_module_entry clickhouse_module_entry = {
#if ZEND_MODULE_API_NO >= 20010901
    STANDARD_MODULE_HEADER,
#endif
    PHP_CLICKHOUSE_EXTNAME,
    clickhouse_functions,
    PHP_MINIT(clickhouse),
    PHP_MSHUTDOWN(clickhouse),
    PHP_RINIT(clickhouse),
    PHP_RSHUTDOWN(clickhouse),
    PHP_MINFO(clickhouse),
#if ZEND_MODULE_API_NO >= 20010901
    PHP_CLICKHOUSE_VERSION,
#endif
    STANDARD_MODULE_PROPERTIES
};

#ifdef COMPILE_DL_CLICKHOUSE
ZEND_GET_MODULE(clickhouse)
#endif

PHP_RINIT_FUNCTION(clickhouse) 
{
    return SUCCESS;
}

PHP_RSHUTDOWN_FUNCTION(clickhouse) 
{
    return SUCCESS;
}

static void clickhouse_obj_res_dtor(zend_resource *rsrc)
{
    chc_destruct((void*)rsrc->ptr);
}

PHP_MINIT_FUNCTION(clickhouse) 
{
	/* Register class name */
    zend_class_entry tmp_ce;
    INIT_CLASS_ENTRY(tmp_ce, "ClickHouse", clickhouse_functions);
    clickhouse_ce = zend_register_internal_class(&tmp_ce TSRMLS_CC);
	zend_declare_property_null(clickhouse_ce, "connection", sizeof("connection") - 1, ZEND_ACC_PRIVATE TSRMLS_CC);
	/* Register connection object as resource */
	clickhouse_obj_res_num = zend_register_list_destructors_ex(clickhouse_obj_res_dtor, NULL, "client", module_number);
    return SUCCESS;
}

PHP_MSHUTDOWN_FUNCTION(clickhouse)
{
    return SUCCESS;
}

PHP_MINFO_FUNCTION(clickhouse)
{
    php_info_print_table_start();
    php_info_print_table_header(2, "ClickHouse support", "enabled");
    php_info_print_table_row(2, "ClickHouse extension version", PHP_CLICKHOUSE_VERSION);
    php_info_print_table_row(2, "Build date", __DATE__);
#ifdef OVEROPTIMIZATION
    php_info_print_table_row(2, "Over-optimization", "enabled");
#else
	php_info_print_table_row(2, "Over-optimization", "disabled");
#endif
    php_info_print_table_end();

    php_info_print_table_start();
    php_info_print_table_row(1, "ClickHouse Module realized by Bartosz Derleta.");
    php_info_print_table_end();
}

PHP_METHOD(ClickHouse, __construct) 
{
	char *host = NULL, *user = NULL, *password = NULL, *default_database = NULL;
	size_t host_len = 0, user_len = 0, password_len = 0, default_database_len = 0;
	zend_long port = 9000;
	zend_bool port_null = false, compression = true;

	ZEND_PARSE_PARAMETERS_START_EX(ZEND_PARSE_PARAMS_THROW, 0, 6)
		Z_PARAM_OPTIONAL
		Z_PARAM_STRING_EX(host, host_len, 1, 0) // Z_PARAM_STRING_OR_NULL
		Z_PARAM_STRING_EX(user, user_len, 1, 0) // Z_PARAM_STRING_OR_NULL
		Z_PARAM_STRING_EX(password, password_len, 1, 0) // Z_PARAM_STRING_OR_NULL
		Z_PARAM_STRING_EX(default_database, default_database_len, 1, 0) // Z_PARAM_STRING_OR_NULL
		Z_PARAM_LONG_EX(port, port_null, 1, 0) // Z_PARAM_LONG_OR_NULL
		Z_PARAM_BOOL(compression)
	ZEND_PARSE_PARAMETERS_END();
	
	void* ch_object = chc_construct(host, user, password, default_database, port_null ? 9000 : port, compression);
	zend_resource *res_client = zend_register_resource(ch_object, clickhouse_obj_res_num);
	zval zv_client;
	ZVAL_RES(&zv_client, res_client);
	zend_update_property(clickhouse_ce, getThis(), "connection", sizeof("connection") - 1, &zv_client TSRMLS_CC);
}

PHP_METHOD(ClickHouse, select) 
{
	char *query;
	size_t query_len, total;
	zval *zv_client;	
	
	zend_fcall_info fci = {0};
	zend_fcall_info_cache fci_cache;
	
	ZEND_PARSE_PARAMETERS_START(2, 2)
	        Z_PARAM_STRING(query, query_len)
	        Z_PARAM_FUNC_EX(fci, fci_cache, 1, 0)
	ZEND_PARSE_PARAMETERS_END();	
	
	zv_client = zend_read_property(clickhouse_ce, getThis(), "connection", sizeof("connection") - 1, 1, NULL TSRMLS_CC);
	ZEND_ASSERT(Z_TYPE_P(zv_client) == IS_RESOURCE);
	void* ch_object = (void*)zend_fetch_resource(Z_RES_P(zv_client), "client", clickhouse_obj_res_num);
	total = chc_select(ch_object, query, &fci, &fci_cache);

	RETURN_LONG(total);
}

PHP_METHOD(ClickHouse, execute) 
{
	char *query;
	size_t query_len;
	zval *zv_client;	
	
	ZEND_PARSE_PARAMETERS_START(1, 1)
	        Z_PARAM_STRING(query, query_len)
	ZEND_PARSE_PARAMETERS_END();	
	
	zv_client = zend_read_property(clickhouse_ce, getThis(), "connection", sizeof("connection") - 1, 1, NULL TSRMLS_CC);
	ZEND_ASSERT(Z_TYPE_P(zv_client) == IS_RESOURCE);
	void* ch_object = (void*)zend_fetch_resource(Z_RES_P(zv_client), "client", clickhouse_obj_res_num);
	chc_execute(ch_object, query);
	
	RETURN_TRUE;
}

PHP_METHOD(ClickHouse, ping) 
{
	zval *zv_client;	
	
	ZEND_PARSE_PARAMETERS_START(0, 0)
	ZEND_PARSE_PARAMETERS_END();	
	
	zv_client = zend_read_property(clickhouse_ce, getThis(), "connection", sizeof("connection") - 1, 1, NULL TSRMLS_CC);
	ZEND_ASSERT(Z_TYPE_P(zv_client) == IS_RESOURCE);
	void* ch_object = (void*)zend_fetch_resource(Z_RES_P(zv_client), "client", clickhouse_obj_res_num);
	chc_ping(ch_object);

	RETURN_TRUE;
}

PHP_METHOD(ClickHouse, resetConnection) 
{
	zval *zv_client;	
	
	ZEND_PARSE_PARAMETERS_START(0, 0)
	ZEND_PARSE_PARAMETERS_END();	
	
	zv_client = zend_read_property(clickhouse_ce, getThis(), "connection", sizeof("connection") - 1, 1, NULL TSRMLS_CC);
	ZEND_ASSERT(Z_TYPE_P(zv_client) == IS_RESOURCE);
	void* ch_object = (void*)zend_fetch_resource(Z_RES_P(zv_client), "client", clickhouse_obj_res_num);
	chc_reset_connection(ch_object);

	RETURN_TRUE;
}

}