Memcache拡張モジュールを透過的に扱えるパッチを書いてみた。
PHPなんだから透過的にmemcachedとか扱えないとまずいでしょうということで。
要はphp.iniに
# 以下の書式が利用可能 # hostname # hostname:tcpport # hostname:tcpport:udpport # [hostname] # [hostname]:tcpport # [hostname]:tcpport:udpport memcache.auto_connect_hosts = 10.1.4.1, 10.1.4.2:11211, [::::::10.1.4.3]::11211
と書いておくと、
<?php $_MEMCACHE['test'] = 'test'; // Memcache::set() 相当 var_dump($_MEMCACHE['test']); // Memcache::get() 相当 ?>
のように、オートグローバル変数 $_MEMCACHE を操作した際に自動的に接続を行ってmemcachedにアクセスできるという話。
以下より memcache-3.0.2.tar.gz を落としてきて、パッチを当ててからビルドしてください。
http://pecl.php.net/package/memcache
追記:
$_MEMCACHE の中身はどうなってるんだという話。var_dump($_MEMCACHE);してみると分かるとおり、単なるMemcacheオブジェクトで、 配列アクセス「[]」がオーバーロードされているだけです。
diff -ur memcache-3.0.2~/memcache.c memcache-3.0.2/memcache.c --- memcache-3.0.2~/memcache.c 2008-09-12 05:03:23.000000000 +0900 +++ memcache-3.0.2/memcache.c 2008-11-19 10:37:50.000000000 +0900 @@ -123,6 +123,32 @@ }; /* }}} */ +/* {{{ mmc_object_handlers */ +static zend_object_handlers mmc_object_handlers = { + ZEND_OBJECTS_STORE_HANDLERS, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, + NULL, +}; +/* }}} */ + #ifdef COMPILE_DL_MEMCACHE ZEND_GET_MODULE(memcache) #endif @@ -224,6 +250,7 @@ PHP_INI_BEGIN() STD_PHP_INI_ENTRY("memcache.allow_failover", "1", PHP_INI_ALL, OnUpdateLong, allow_failover, zend_memcache_globals, memcache_globals) STD_PHP_INI_ENTRY("memcache.max_failover_attempts", "20", PHP_INI_ALL, OnUpdateFailoverAttempts, max_failover_attempts, zend_memcache_globals, memcache_globals) + STD_PHP_INI_ENTRY("memcache.auto_connect_hosts", "localhost:11211", PHP_INI_ALL, OnUpdateString, auto_connect_hosts, zend_memcache_globals, memcache_globals) STD_PHP_INI_ENTRY("memcache.default_port", "11211", PHP_INI_ALL, OnUpdateLong, default_port, zend_memcache_globals, memcache_globals) STD_PHP_INI_ENTRY("memcache.chunk_size", "32768", PHP_INI_ALL, OnUpdateChunkSize, chunk_size, zend_memcache_globals, memcache_globals) STD_PHP_INI_ENTRY("memcache.protocol", "ascii", PHP_INI_ALL, OnUpdateProtocol, protocol, zend_memcache_globals, memcache_globals) @@ -244,6 +271,9 @@ static void _mmc_server_list_dtor(zend_rsrc_list_entry * TSRMLS_DC); static void php_mmc_set_failure_callback(mmc_pool_t *, zval *, zval * TSRMLS_DC); static void php_mmc_failure_callback(mmc_pool_t *, mmc_t *, void * TSRMLS_DC); +static zend_bool _php_mmc_auto_global_cb(char *name, uint name_len TSRMLS_DC); +static zval *_php_mmc_read_dimension(zval *object, zval *offset, int type TSRMLS_DC); +static void _php_mmc_write_dimension(zval *object, zval *offset, zval *value TSRMLS_DC); /* }}} */ /* {{{ php_memcache_init_globals() @@ -260,7 +290,10 @@ PHP_MINIT_FUNCTION(memcache) { zend_class_entry ce; - + mmc_object_handlers = *zend_get_std_object_handlers(); + mmc_object_handlers.read_dimension = _php_mmc_read_dimension; + mmc_object_handlers.write_dimension = _php_mmc_write_dimension; + INIT_CLASS_ENTRY(ce, "MemcachePool", php_memcache_pool_class_functions); memcache_pool_ce = zend_register_internal_class(&ce TSRMLS_CC); @@ -286,6 +319,8 @@ REGISTER_LONG_CONSTANT("MEMCACHE_HAVE_SESSION", 0, CONST_CS | CONST_PERSISTENT); #endif + zend_register_auto_global("_MEMCACHE", sizeof("_MEMCACHE") - 1, _php_mmc_auto_global_cb); + return SUCCESS; } /* }}} */ @@ -523,7 +558,7 @@ long value = 1, defval = 0, exptime = 0; mmc_request_t *request; void *value_handler_param[3]; - + if (mmc_object == NULL) { if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "Oz|lll", &mmc_object, memcache_pool_ce, &keys, &value, &defval, &exptime) == FAILURE) { return; @@ -986,6 +1021,238 @@ } /* }}} */ +static zend_bool _php_mmc_auto_global_cb(char *name, uint name_len TSRMLS_DC) /* {{{ */ +{ + zval *mmc_object = NULL; + if (!MEMCACHE_G(auto_connect_hosts)) { + return FALSE; + } + { + int list_id; + zend_object_value ov; + mmc_pool_t *pool = mmc_pool_new(TSRMLS_C); + pool->failure_callback_param = &php_mmc_failure_callback; + pool->failure_callback = &php_mmc_failure_callback; + + ALLOC_INIT_ZVAL(mmc_object); + object_init_ex(mmc_object, memcache_ce); + list_id = zend_list_insert(pool, le_memcache_pool); + add_property_resource(mmc_object, "connection", list_id); + Z_OBJ_HT_P(mmc_object) = &mmc_object_handlers; + } + { + const char *host; + int host_len; + int tcp_port; + int udp_port; + { + const char *p, *pt; + p = MEMCACHE_G(auto_connect_hosts); + for (;;) { + while (*p != '\0' && isspace(*(unsigned char *)p)) { + ++p; + } + if (*p == '\0') { + break; + } + if (*p == '[') { + host = ++p; + while (*p != '\0' && *p != ']') { + ++p; + } + if (*p == '\0') { + pt = host; + } else { + pt = p; + ++p; + while (*p != '\0' && isspace(*(unsigned char *)p)) { + ++p; + } + if (*p != '\0' && *p != ':' && *p != ',') { + pt = host; + } + } + } else { + host = p; + while (*p != '\0' && *p != ':' && *p != ',') { + ++p; + } + pt = p; + while (pt > host && isspace(*(unsigned char *)(pt - 1))) { + --pt; + } + } + host_len = pt - host; + tcp_port = MEMCACHE_G(default_port); + udp_port = 0; + if (*p == ':') { + ++p; + while (*p != '\0' && isspace(*(unsigned char *)p)) { + ++p; + } + if (*p != '\0') { + const char *tcp_port_str = p; + while (*p != '\0' && *p != ':' && *p != ',') { + ++p; + } + pt = p; + while (pt > tcp_port_str && isspace(*(unsigned char *)(pt - 1))) { + --pt; + } + if (pt != tcp_port_str) { + errno = 0; + tcp_port = strtol(tcp_port_str, NULL, 10); + switch (errno) { + case EINVAL: case ERANGE: + tcp_port = -1; + break; + } + } + } + if (*p == ':') { + ++p; + while (*p != '\0' && isspace(*(unsigned char *)p)) { + ++p; + } + if (*p != '\0') { + const char *udp_port_str = p; + while (*p != '\0' && *p != ',') { + ++p; + } + pt = p; + while (pt > udp_port_str && isspace(*(unsigned char *)(pt - 1))) { + --pt; + } + if (pt != udp_port_str) { + errno = 0; + udp_port = strtol(udp_port_str, NULL, 10); + switch (errno) { + case EINVAL: case ERANGE: + udp_port = -1; + break; + } + } + if (*p == ',') { + ++p; + } + } + } else if (*p == ',') { + ++p; + } + } else if (*p == ',') { + ++p; + } + if (host_len == 0) { + php_error_docref(NULL TSRMLS_CC, E_WARNING, "memcache.auto_connect_hosts contains an invalid host name"); + goto fail; + } + if (tcp_port < 0 || udp_port < 0) { + php_error_docref(NULL TSRMLS_CC, E_WARNING, "memcache.auto_connect_hosts contains an invalid port number"); + goto fail; + } + + if (!php_mmc_pool_addserver(mmc_object, host, host_len, tcp_port, 0, 1, 1, MMC_DEFAULT_TIMEOUT, MMC_DEFAULT_RETRY, 1, NULL TSRMLS_CC)) { + goto fail; + } + } + } + } + zend_hash_update(&EG(symbol_table), "_MEMCACHE", sizeof("_MEMCACHE"), &mmc_object, sizeof(zval *), NULL); + return 0; +fail: + if (mmc_object) { + zval_ptr_dtor(&mmc_object); + } + return 1; +} +/* }}} */ + +static zval *_php_mmc_read_dimension(zval *mmc_object, zval *offset, int type TSRMLS_DC) /* {{{ */ +{ + mmc_pool_t *pool; + void *value_handler_param[3], *failover_handler_param[2]; + zval *return_value; + if (!mmc_get_pool(mmc_object, &pool TSRMLS_CC) || !pool->num_servers) { + return NULL; + } + + ALLOC_INIT_ZVAL(return_value); + ZVAL_FALSE(return_value); + + value_handler_param[0] = return_value; + value_handler_param[1] = NULL; + value_handler_param[2] = NULL; + + { + mmc_request_t *request; + + /* allocate request */ + request = mmc_pool_request_get( + pool, MMC_PROTO_UDP, + mmc_value_handler_single, value_handler_param, + mmc_pool_failover_handler, NULL TSRMLS_CC); + + if (mmc_prepare_key(offset, request->key, &(request->key_len)) != MMC_OK) { + mmc_pool_release(pool, request); + php_error_docref(NULL TSRMLS_CC, E_WARNING, "Invalid key"); + zval_ptr_dtor(&return_value); + return NULL; + } + + pool->protocol->get(request, MMC_OP_GET, offset, request->key, request->key_len); + + /* schedule request */ + if (mmc_pool_schedule_key(pool, request->key, request->key_len, request, 1 TSRMLS_CC) != MMC_OK) { + zval_ptr_dtor(&return_value); + return NULL; + } + } + + /* execute all requests */ + mmc_pool_run(pool TSRMLS_CC); + + return return_value; +} +/* }}} */ + +static void _php_mmc_write_dimension(zval *mmc_object, zval *offset, zval *value TSRMLS_DC) /* {{{ */ +{ + mmc_pool_t *pool; + mmc_request_t *request; + zval return_value; + + if (!mmc_get_pool(mmc_object, &pool TSRMLS_CC) || !pool->num_servers) { + return; + } + + { + mmc_request_t *request; + /* allocate request */ + request = mmc_pool_request(pool, MMC_PROTO_TCP, mmc_stored_handler, &return_value, mmc_pool_failover_handler, NULL TSRMLS_CC); + + if (mmc_prepare_key(offset, request->key, &(request->key_len)) != MMC_OK) { + php_error_docref(NULL TSRMLS_CC, E_WARNING, "Invalid key"); + mmc_pool_release(pool, request); + return; + } + + /* assemble command */ + if (pool->protocol->store(pool, request, MMC_OP_SET, request->key, request->key_len, 0, 0, 0, value TSRMLS_CC) != MMC_OK) { + mmc_pool_release(pool, request); + return; + } + + /* schedule request */ + if (mmc_pool_schedule_key(pool, request->key, request->key_len, request, MEMCACHE_G(redundancy) TSRMLS_CC) != MMC_OK) { + return; + } + } + + /* execute all requests */ + mmc_pool_run(pool TSRMLS_CC); +} +/* }}} */ + /* ---------------- module functions ---------------- */ diff -ur memcache-3.0.2~/memcache_pool.h memcache-3.0.2/memcache_pool.h --- memcache-3.0.2~/memcache_pool.h 2008-09-12 05:03:23.000000000 +0900 +++ memcache-3.0.2/memcache_pool.h 2008-11-18 09:43:24.000000000 +0900 @@ -348,6 +348,7 @@ long max_failover_attempts; long redundancy; long session_redundancy; + char* auto_connect_hosts; ZEND_END_MODULE_GLOBALS(memcache) #ifdef ZTS
