Add RedisArray::_continuum method

yatsukhnenko · yatsukhnenko · commit 71922bf1dd30 · 2018-12-22T15:31:51.000+02:00
diff --git a/redis_array.c b/redis_array.c
@@ -104,6 +104,7 @@ ZEND_END_ARG_INFO()
 zend_function_entry redis_array_functions[] = {
      PHP_ME(RedisArray, __call, arginfo_call, ZEND_ACC_PUBLIC)
      PHP_ME(RedisArray, __construct, arginfo_ctor, ZEND_ACC_PUBLIC)
+     PHP_ME(RedisArray, _continuum, arginfo_void, ZEND_ACC_PUBLIC)
      PHP_ME(RedisArray, _distributor, arginfo_void, ZEND_ACC_PUBLIC)
      PHP_ME(RedisArray, _function, arginfo_void, ZEND_ACC_PUBLIC)
      PHP_ME(RedisArray, _hosts, arginfo_void, ZEND_ACC_PUBLIC)
@@ -639,6 +640,38 @@ PHP_METHOD(RedisArray, _rehash)
     }
 }
 
+PHP_METHOD(RedisArray, _continuum)
+{
+    int i;
+    zval *object;
+    RedisArray *ra;
+
+    if (zend_parse_method_parameters(ZEND_NUM_ARGS() TSRMLS_CC, getThis(), "O",
+                &object, redis_array_ce) == FAILURE) {
+        RETURN_FALSE;
+    }
+
+    if ((ra = redis_array_get(object TSRMLS_CC)) == NULL) {
+        RETURN_FALSE;
+    }
+
+    array_init(return_value);
+    if (ra->continuum) {
+        for (i = 0; i < ra->continuum->nb_points; ++i) {
+            zval zv, *z_tmp = &zv;
+#if (PHP_MAJOR_VERSION < 7)
+            MAKE_STD_ZVAL(z_tmp);
+#endif
+
+            array_init(z_tmp);
+            add_assoc_long(z_tmp, "index", ra->continuum->points[i].index);
+            add_assoc_long(z_tmp, "value", ra->continuum->points[i].value);
+            add_next_index_zval(return_value, z_tmp);
+        }
+    }
+}
+
+
 static void
 multihost_distribute_call(RedisArray *ra, zval *return_value, zval *z_fun, int argc, zval *argv TSRMLS_DC)
 {
diff --git a/redis_array.h b/redis_array.h
@@ -16,6 +16,7 @@ PHP_METHOD(RedisArray, _instance);
 PHP_METHOD(RedisArray, _function);
 PHP_METHOD(RedisArray, _distributor);
 PHP_METHOD(RedisArray, _rehash);
+PHP_METHOD(RedisArray, _continuum);
 
 PHP_METHOD(RedisArray, select);
 PHP_METHOD(RedisArray, info);
