Add INI setting to choose session consistent hash (ketama or ketama_weighted)

Author: sodabrew

php_memcached.c

@@ -323,6 +323,20 @@ PHP_INI_MH(OnUpdateSessionPrefixString)
 	return OnUpdateString(entry, new_value, mh_arg1, mh_arg2, mh_arg3, stage);
 }
 
+static
+PHP_INI_MH(OnUpdateConsistentHash)
+{
+	if (!strcmp(ZSTR_VAL(new_value), "ketama")) {
+        MEMC_SESS_INI(consistent_hash_type) = MEMCACHED_BEHAVIOR_KETAMA;
+	} else if (!strcmp(ZSTR_VAL(new_value), "ketama_weighted")) {
+        MEMC_SESS_INI(consistent_hash_type) = MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED;
+	} else {
+		return FAILURE;
+	}
+
+	return OnUpdateString(entry, new_value, mh_arg1, mh_arg2, mh_arg3, stage);
+}
+
 #define MEMC_INI_ENTRY(key, default_value, update_fn, gkey) \
 	STD_PHP_INI_ENTRY("memcached."key, default_value, PHP_INI_ALL, update_fn, memc.gkey, zend_php_memcached_globals, php_memcached_globals)
 
@@ -357,6 +371,7 @@ PHP_INI_BEGIN()
 	MEMC_SESSION_INI_BOOL ("binary_protocol",        "1",          OnUpdateBool,           binary_protocol_enabled)
 #endif
 	MEMC_SESSION_INI_BOOL ("consistent_hash",        "1",          OnUpdateBool,           consistent_hash_enabled)
+	MEMC_SESSION_INI_BOOL ("consistent_hash_type",   "ketama",     OnUpdateConsistentHash, consistent_hash_type)
 	MEMC_SESSION_INI_ENTRY("number_of_replicas",     "0",          OnUpdateLongGEZero,     number_of_replicas)
 	MEMC_SESSION_INI_BOOL ("randomize_replica_read", "0",          OnUpdateBool,           randomize_replica_read_enabled)
 	MEMC_SESSION_INI_BOOL ("remove_failed_servers",  "0",          OnUpdateBool,           remove_failed_servers_enabled)
@@ -4263,6 +4278,7 @@ PHP_GINIT_FUNCTION(php_memcached)
 	php_memcached_globals->session.lock_expiration = 30;
 	php_memcached_globals->session.binary_protocol_enabled = 1;
 	php_memcached_globals->session.consistent_hash_enabled = 1;
+	php_memcached_globals->session.consistent_hash_type = MEMCACHED_BEHAVIOR_KETAMA;
 	php_memcached_globals->session.number_of_replicas = 0;
 	php_memcached_globals->session.server_failure_limit = 1;
 	php_memcached_globals->session.randomize_replica_read_enabled = 1;

php_memcached.h

@@ -42,14 +42,6 @@ PHP_MEMCACHED_API zend_class_entry *php_memc_get_exception_base(int root);
 extern zend_module_entry memcached_module_entry;
 #define phpext_memcached_ptr &memcached_module_entry
 
-#ifdef ZTS
-#define MEMC_G(v) TSRMG(php_memcached_globals_id, zend_php_memcached_globals *, memc.v)
-#define MEMC_SERVER_G(v) TSRMG(php_memcached_globals_id, zend_php_memcached_globals *, server.v)
-#else
-#define MEMC_G(v) (php_memcached_globals.memc.v)
-#define MEMC_SERVER_G(v) (php_memcached_globals.server.v)
-#endif
-
 #endif /* PHP_MEMCACHED_H */
 
 /*

php_memcached_private.h

@@ -79,6 +79,19 @@ typedef unsigned long int uint32_t;
 #  define GC_SET_REFCOUNT(p, rc)  GC_REFCOUNT(p) = rc
 #endif
 
+/* Globals accessor macros */
+#ifdef ZTS
+#  define MEMC_G(v) TSRMG(php_memcached_globals_id, zend_php_memcached_globals *, memc.v)
+#  define MEMC_SERVER_G(v) TSRMG(php_memcached_globals_id, zend_php_memcached_globals *, server.v)
+#  define MEMC_SESS_INI(v) TSRMG(php_memcached_globals_id, zend_php_memcached_globals *, session.v)
+#else
+#  define MEMC_G(v) (php_memcached_globals.memc.v)
+#  define MEMC_SERVER_G(v) (php_memcached_globals.server.v)
+#  define MEMC_SESS_INI(v) (php_memcached_globals.session.v)
+#endif
+
+#define MEMC_SESS_STR_INI(vv) ((MEMC_SESS_INI(vv) && *MEMC_SESS_INI(vv)) ? MEMC_SESS_INI(vv) : NULL)
+
 /****************************************
   Structures and definitions
 ****************************************/
@@ -156,6 +169,7 @@ ZEND_BEGIN_MODULE_GLOBALS(php_memcached)
 
 		zend_bool binary_protocol_enabled;
 		zend_bool consistent_hash_enabled;
+		zend_bool consistent_hash_type;
 
 		zend_long server_failure_limit;
 		zend_long number_of_replicas;

php_memcached_session.c

@@ -43,14 +43,6 @@ typedef struct  {
 # define MAX(a,b) (((a)>(b))?(a):(b))
 #endif
 
-#ifdef ZTS
-#define MEMC_SESS_INI(v) TSRMG(php_memcached_globals_id, zend_php_memcached_globals *, session.v)
-#else
-#define MEMC_SESS_INI(v) (php_memcached_globals.session.v)
-#endif
-
-#define MEMC_SESS_STR_INI(vv) ((MEMC_SESS_INI(vv) && *MEMC_SESS_INI(vv)) ? MEMC_SESS_INI(vv) : NULL)
-
 static
 	int le_memc_sess;
 
@@ -202,7 +194,7 @@ zend_bool s_configure_from_ini_values(memcached_st *memc, zend_bool silent)
 	}
 
 	if (MEMC_SESS_INI(consistent_hash_enabled)) {
-		check_set_behavior(MEMCACHED_BEHAVIOR_KETAMA, 1);
+		check_set_behavior(MEMC_SESS_INI(consistent_hash_type), 1);
 	}
 
 	if (MEMC_SESS_INI(server_failure_limit)) {