PageRenderTime 45ms CodeModel.GetById 16ms RepoModel.GetById 1ms app.codeStats 0ms

/redis_cluster.h

https://gitlab.com/oytunistrator/phpredis
C Header | 283 lines | 246 code | 19 blank | 18 comment | 16 complexity | fbac7c6343f36aee4e33cb9991bd7165 MD5 | raw file
    

  1. #ifndef REDIS_CLUSTER_H
    2. 

  2. #define REDIS_CLUSTER_H
    3. 

  3. 

  4. #include "cluster_library.h"
    5. 

  5. #include <php.h>
    6. 

  6. #include <stddef.h>
    7. 

  7. 

  8. /* Redis cluster hash slots and N-1 which we'll use to find it */
    9. 

  9. #define REDIS_CLUSTER_SLOTS 16384
    10. 

  10. #define REDIS_CLUSTER_MOD   (REDIS_CLUSTER_SLOTS-1)
    11. 

  11. 

  12. /* Get attached object context */
    13. 

  13. #define GET_CONTEXT() \
    14. 

  14.     ((redisCluster*)zend_object_store_get_object(getThis() TSRMLS_CC))
    15. 

  15. 

  16. /* Command building/processing is identical for every command */
    17. 

  17. #define CLUSTER_BUILD_CMD(name, c, cmd, cmd_len, slot) \
    18. 

  18.     redis_##name##_cmd(INTERNAL_FUNCTION_PARAM_PASSTHRU, c->flags, &cmd, \
    19. 

  19.                        &cmd_len, &slot)
    20. 

  20. 

  21. /* Append information required to handle MULTI commands to the tail of our MULTI 
    22. 

  22.  * linked list. */
    23. 

  23. #define CLUSTER_ENQUEUE_RESPONSE(c, slot, cb, ctx) \
    24. 

  24.     clusterFoldItem *_item; \
    25. 

  25.     _item = emalloc(sizeof(clusterFoldItem)); \
    26. 

  26.     _item->callback = cb; \
    27. 

  27.     _item->slot = slot; \
    28. 

  28.     _item->ctx = ctx; \
    29. 

  29.     _item->next = NULL; \
    30. 

  30.     if(c->multi_head == NULL) { \
    31. 

  31.         c->multi_head = _item; \
    32. 

  32.         c->multi_curr = _item; \
    33. 

  33.     } else { \
    34. 

  34.         c->multi_curr->next = _item; \
    35. 

  35.         c->multi_curr = _item; \
    36. 

  36.     } \
    37. 

  37. 

  38. /* Simple macro to free our enqueued callbacks after we EXEC */
    39. 

  39. #define CLUSTER_FREE_QUEUE(c) \
    40. 

  40.     clusterFoldItem *_item = c->multi_head, *_tmp; \
    41. 

  41.     while(_item) { \
    42. 

  42.         _tmp = _item->next; \
    43. 

  43.         efree(_item); \
    44. 

  44.         _item = _tmp; \
    45. 

  45.     } \
    46. 

  46.     c->multi_head = c->multi_curr = NULL; \
    47. 

  47. 

  48. /* Reset anything flagged as MULTI */
    49. 

  49. #define CLUSTER_RESET_MULTI(c) \
    50. 

  50.     redisClusterNode **_node; \
    51. 

  51.     for(zend_hash_internal_pointer_reset(c->nodes); \
    52. 

  52.         zend_hash_get_current_data(c->nodes, (void**)&_node); \
    53. 

  53.         zend_hash_move_forward(c->nodes)) \
    54. 

  54.     { \
    55. 

  55.         (*_node)->sock->watching = 0; \
    56. 

  56.         (*_node)->sock->mode     = ATOMIC; \
    57. 

  57.     } \
    58. 

  58.     c->flags->watching = 0; \
    59. 

  59.     c->flags->mode     = ATOMIC; \
    60. 

  60. 

  61. /* Simple 1-1 command -> response macro */
    62. 

  62. #define CLUSTER_PROCESS_CMD(cmdname, resp_func, readcmd) \
    63. 

  63.     redisCluster *c = GET_CONTEXT(); \
    64. 

  64.     c->readonly = CLUSTER_IS_ATOMIC(c) && readcmd; \
    65. 

  65.     char *cmd; int cmd_len; short slot; void *ctx=NULL; \
    66. 

  66.     if(redis_##cmdname##_cmd(INTERNAL_FUNCTION_PARAM_PASSTHRU,c->flags, &cmd, \
    67. 

  67.                              &cmd_len, &slot, &ctx)==FAILURE) { \
    68. 

  68.         RETURN_FALSE; \
    69. 

  69.     } \
    70. 

  70.     if(cluster_send_command(c,slot,cmd,cmd_len TSRMLS_CC)<0 || c->err!=NULL) {\
    71. 

  71.         efree(cmd); \
    72. 

  72.         RETURN_FALSE; \
    73. 

  73.     } \
    74. 

  74.     efree(cmd); \
    75. 

  75.     if(c->flags->mode == MULTI) { \
    76. 

  76.         CLUSTER_ENQUEUE_RESPONSE(c, slot, resp_func, ctx); \
    77. 

  77.         RETURN_ZVAL(getThis(), 1, 0); \
    78. 

  78.     } \
    79. 

  79.     resp_func(INTERNAL_FUNCTION_PARAM_PASSTHRU, c, ctx); 
    80. 

  80.         
    81. 

  81. /* More generic processing, where only the keyword differs */
    82. 

  82. #define CLUSTER_PROCESS_KW_CMD(kw, cmdfunc, resp_func, readcmd) \
    83. 

  83.     redisCluster *c = GET_CONTEXT(); \
    84. 

  84.     c->readonly = CLUSTER_IS_ATOMIC(c) && readcmd; \
    85. 

  85.     char *cmd; int cmd_len; short slot; void *ctx=NULL; \
    86. 

  86.     if(cmdfunc(INTERNAL_FUNCTION_PARAM_PASSTHRU, c->flags, kw, &cmd, &cmd_len,\
    87. 

  87.                &slot,&ctx)==FAILURE) { \
    88. 

  88.         RETURN_FALSE; \
    89. 

  89.     } \
    90. 

  90.     if(cluster_send_command(c,slot,cmd,cmd_len TSRMLS_CC)<0 || c->err!=NULL) { \
    91. 

  91.         efree(cmd); \
    92. 

  92.         RETURN_FALSE; \
    93. 

  93.     } \
    94. 

  94.     efree(cmd); \
    95. 

  95.     if(c->flags->mode == MULTI) { \
    96. 

  96.         CLUSTER_ENQUEUE_RESPONSE(c, slot, resp_func, ctx); \
    97. 

  97.         RETURN_ZVAL(getThis(), 1, 0); \
    98. 

  98.     } \
    99. 

  99.     resp_func(INTERNAL_FUNCTION_PARAM_PASSTHRU, c, ctx); 
    100. 

  100. 

  101. /* For the creation of RedisCluster specific exceptions */
    102. 

  102. PHP_REDIS_API zend_class_entry *rediscluster_get_exception_base(int root TSRMLS_DC);
    103. 

  103. 

  104. /* Create cluster context */
    105. 

  105. zend_object_value create_cluster_context(zend_class_entry *class_type 
    106. 

  106.                                          TSRMLS_DC);
    107. 

  107. 

  108. /* Free cluster context struct */
    109. 

  109. void free_cluster_context(void *object TSRMLS_DC);
    110. 

  110. 

  111. /* Inittialize our class with PHP */
    112. 

  112. void init_rediscluster(TSRMLS_D);
    113. 

  113. 

  114. /* RedisCluster method implementation */
    115. 

  115. PHP_METHOD(RedisCluster, __construct);
    116. 

  116. PHP_METHOD(RedisCluster, close);
    117. 

  117. PHP_METHOD(RedisCluster, get);
    118. 

  118. PHP_METHOD(RedisCluster, set);
    119. 

  119. PHP_METHOD(RedisCluster, mget);
    120. 

  120. PHP_METHOD(RedisCluster, mset);
    121. 

  121. PHP_METHOD(RedisCluster, msetnx);
    122. 

  122. PHP_METHOD(RedisCluster, mset);
    123. 

  123. PHP_METHOD(RedisCluster, del);
    124. 

  124. PHP_METHOD(RedisCluster, dump);
    125. 

  125. PHP_METHOD(RedisCluster, setex);
    126. 

  126. PHP_METHOD(RedisCluster, psetex);
    127. 

  127. PHP_METHOD(RedisCluster, setnx);
    128. 

  128. PHP_METHOD(RedisCluster, getset);
    129. 

  129. PHP_METHOD(RedisCluster, exists);
    130. 

  130. PHP_METHOD(RedisCluster, keys);
    131. 

  131. PHP_METHOD(RedisCluster, type);
    132. 

  132. PHP_METHOD(RedisCluster, persist);
    133. 

  133. PHP_METHOD(RedisCluster, lpop);
    134. 

  134. PHP_METHOD(RedisCluster, rpop);
    135. 

  135. PHP_METHOD(RedisCluster, spop);
    136. 

  136. PHP_METHOD(RedisCluster, rpush);
    137. 

  137. PHP_METHOD(RedisCluster, lpush);
    138. 

  138. PHP_METHOD(RedisCluster, blpop);
    139. 

  139. PHP_METHOD(RedisCluster, brpop);
    140. 

  140. PHP_METHOD(RedisCluster, rpushx);
    141. 

  141. PHP_METHOD(RedisCluster, lpushx);
    142. 

  142. PHP_METHOD(RedisCluster, linsert);
    143. 

  143. PHP_METHOD(RedisCluster, lindex);
    144. 

  144. PHP_METHOD(RedisCluster, lrem);
    145. 

  145. PHP_METHOD(RedisCluster, brpoplpush);
    146. 

  146. PHP_METHOD(RedisCluster, rpoplpush);
    147. 

  147. PHP_METHOD(RedisCluster, llen);
    148. 

  148. PHP_METHOD(RedisCluster, scard);
    149. 

  149. PHP_METHOD(RedisCluster, smembers);
    150. 

  150. PHP_METHOD(RedisCluster, sismember);
    151. 

  151. PHP_METHOD(RedisCluster, sadd);
    152. 

  152. PHP_METHOD(RedisCluster, saddarray);
    153. 

  153. PHP_METHOD(RedisCluster, srem);
    154. 

  154. PHP_METHOD(RedisCluster, sunion);
    155. 

  155. PHP_METHOD(RedisCluster, sunionstore);
    156. 

  156. PHP_METHOD(RedisCluster, sinter);
    157. 

  157. PHP_METHOD(RedisCluster, sinterstore);
    158. 

  158. PHP_METHOD(RedisCluster, sdiff);
    159. 

  159. PHP_METHOD(RedisCluster, sdiffstore);
    160. 

  160. PHP_METHOD(RedisCluster, strlen);
    161. 

  161. PHP_METHOD(RedisCluster, ttl);
    162. 

  162. PHP_METHOD(RedisCluster, pttl);
    163. 

  163. PHP_METHOD(RedisCluster, zcard);
    164. 

  164. PHP_METHOD(RedisCluster, zscore);
    165. 

  165. PHP_METHOD(RedisCluster, zcount);
    166. 

  166. PHP_METHOD(RedisCluster, zrem);
    167. 

  167. PHP_METHOD(RedisCluster, zremrangebyscore);
    168. 

  168. PHP_METHOD(RedisCluster, zrank);
    169. 

  169. PHP_METHOD(RedisCluster, zrevrank);
    170. 

  170. PHP_METHOD(RedisCluster, zadd);
    171. 

  171. PHP_METHOD(RedisCluster, zincrby);
    172. 

  172. PHP_METHOD(RedisCluster, hlen);
    173. 

  173. PHP_METHOD(RedisCluster, hget);
    174. 

  174. PHP_METHOD(RedisCluster, hkeys);
    175. 

  175. PHP_METHOD(RedisCluster, hvals);
    176. 

  176. PHP_METHOD(RedisCluster, hmget);
    177. 

  177. PHP_METHOD(RedisCluster, hmset);
    178. 

  178. PHP_METHOD(RedisCluster, hdel);
    179. 

  179. PHP_METHOD(RedisCluster, hgetall);
    180. 

  180. PHP_METHOD(RedisCluster, hexists);
    181. 

  181. PHP_METHOD(RedisCluster, hincrby);
    182. 

  182. PHP_METHOD(RedisCluster, hincrbyfloat);
    183. 

  183. PHP_METHOD(RedisCluster, hset);
    184. 

  184. PHP_METHOD(RedisCluster, hsetnx);
    185. 

  185. PHP_METHOD(RedisCluster, incr);
    186. 

  186. PHP_METHOD(RedisCluster, decr);
    187. 

  187. PHP_METHOD(RedisCluster, incrby);
    188. 

  188. PHP_METHOD(RedisCluster, decrby);
    189. 

  189. PHP_METHOD(RedisCluster, incrbyfloat);
    190. 

  190. PHP_METHOD(RedisCluster, expire);
    191. 

  191. PHP_METHOD(RedisCluster, expireat);
    192. 

  192. PHP_METHOD(RedisCluster, pexpire);
    193. 

  193. PHP_METHOD(RedisCluster, pexpireat);
    194. 

  194. PHP_METHOD(RedisCluster, append);
    195. 

  195. PHP_METHOD(RedisCluster, getbit);
    196. 

  196. PHP_METHOD(RedisCluster, setbit);
    197. 

  197. PHP_METHOD(RedisCluster, bitop);
    198. 

  198. PHP_METHOD(RedisCluster, bitpos);
    199. 

  199. PHP_METHOD(RedisCluster, bitcount);
    200. 

  200. PHP_METHOD(RedisCluster, lget);
    201. 

  201. PHP_METHOD(RedisCluster, getrange);
    202. 

  202. PHP_METHOD(RedisCluster, ltrim);
    203. 

  203. PHP_METHOD(RedisCluster, lrange);
    204. 

  204. PHP_METHOD(RedisCluster, zremrangebyrank);
    205. 

  205. PHP_METHOD(RedisCluster, publish);
    206. 

  206. PHP_METHOD(RedisCluster, lset);
    207. 

  207. PHP_METHOD(RedisCluster, rename);
    208. 

  208. PHP_METHOD(RedisCluster, renamenx);
    209. 

  209. PHP_METHOD(RedisCluster, pfcount);
    210. 

  210. PHP_METHOD(RedisCluster, pfadd);
    211. 

  211. PHP_METHOD(RedisCluster, pfmerge);
    212. 

  212. PHP_METHOD(RedisCluster, restore);
    213. 

  213. PHP_METHOD(RedisCluster, setrange);
    214. 

  214. PHP_METHOD(RedisCluster, smove);
    215. 

  215. PHP_METHOD(RedisCluster, srandmember);
    216. 

  216. PHP_METHOD(RedisCluster, zrange);
    217. 

  217. PHP_METHOD(RedisCluster, zrevrange);
    218. 

  218. PHP_METHOD(RedisCluster, zrangebyscore);
    219. 

  219. PHP_METHOD(RedisCluster, zrevrangebyscore);
    220. 

  220. PHP_METHOD(RedisCluster, zrangebylex);
    221. 

  221. PHP_METHOD(RedisCluster, zrevrangebylex);
    222. 

  222. PHP_METHOD(RedisCluster, zlexcount);
    223. 

  223. PHP_METHOD(RedisCluster, zremrangebylex);
    224. 

  224. PHP_METHOD(RedisCluster, zunionstore);
    225. 

  225. PHP_METHOD(RedisCluster, zinterstore);
    226. 

  226. PHP_METHOD(RedisCluster, sort);
    227. 

  227. PHP_METHOD(RedisCluster, object);
    228. 

  228. PHP_METHOD(RedisCluster, subscribe);
    229. 

  229. PHP_METHOD(RedisCluster, psubscribe);
    230. 

  230. PHP_METHOD(RedisCluster, unsubscribe);
    231. 

  231. PHP_METHOD(RedisCluster, punsubscribe);
    232. 

  232. PHP_METHOD(RedisCluster, eval);
    233. 

  233. PHP_METHOD(RedisCluster, evalsha);
    234. 

  234. PHP_METHOD(RedisCluster, info);
    235. 

  235. PHP_METHOD(RedisCluster, cluster);
    236. 

  236. PHP_METHOD(RedisCluster, client);
    237. 

  237. PHP_METHOD(RedisCluster, config);
    238. 

  238. PHP_METHOD(RedisCluster, pubsub);
    239. 

  239. PHP_METHOD(RedisCluster, script);
    240. 

  240. PHP_METHOD(RedisCluster, slowlog);
    241. 

  241. PHP_METHOD(RedisCluster, command);
    242. 

  242. 

  243. /* SCAN and friends */
    244. 

  244. PHP_METHOD(RedisCluster, scan);
    245. 

  245. PHP_METHOD(RedisCluster, zscan);
    246. 

  246. PHP_METHOD(RedisCluster, hscan);
    247. 

  247. PHP_METHOD(RedisCluster, sscan);
    248. 

  248. 

  249. /* Transactions */
    250. 

  250. PHP_METHOD(RedisCluster, multi);
    251. 

  251. PHP_METHOD(RedisCluster, exec);
    252. 

  252. PHP_METHOD(RedisCluster, discard);
    253. 

  253. PHP_METHOD(RedisCluster, watch);
    254. 

  254. PHP_METHOD(RedisCluster, unwatch);
    255. 

  255. 

  256. /* Commands we direct to a node */
    257. 

  257. PHP_METHOD(RedisCluster, save);
    258. 

  258. PHP_METHOD(RedisCluster, bgsave);
    259. 

  259. PHP_METHOD(RedisCluster, flushdb);
    260. 

  260. PHP_METHOD(RedisCluster, flushall);
    261. 

  261. PHP_METHOD(RedisCluster, dbsize);
    262. 

  262. PHP_METHOD(RedisCluster, bgrewriteaof);
    263. 

  263. PHP_METHOD(RedisCluster, lastsave);
    264. 

  264. PHP_METHOD(RedisCluster, role);
    265. 

  265. PHP_METHOD(RedisCluster, time);
    266. 

  266. PHP_METHOD(RedisCluster, randomkey);
    267. 

  267. PHP_METHOD(RedisCluster, ping);
    268. 

  268. PHP_METHOD(RedisCluster, echo);
    269. 

  269. PHP_METHOD(RedisCluster, rawcommand);
    270. 

  270. 

  271. /* Introspection */
    272. 

  272. PHP_METHOD(RedisCluster, getmode);
    273. 

  273. PHP_METHOD(RedisCluster, getlasterror);
    274. 

  274. PHP_METHOD(RedisCluster, clearlasterror);
    275. 

  275. PHP_METHOD(RedisCluster, getoption);
    276. 

  276. PHP_METHOD(RedisCluster, setoption);
    277. 

  277. PHP_METHOD(RedisCluster, _prefix);
    278. 

  278. PHP_METHOD(RedisCluster, _serialize);
    279. 

  279. PHP_METHOD(RedisCluster, _unserialize);
    280. 

  280. PHP_METHOD(RedisCluster, _masters);
    281. 

  281. PHP_METHOD(RedisCluster, _redir);
    282. 

  282. 

  283. #endif
    284. 

  284.