hphp/test/slow/ext_redis/scan.php

   1 <?php
   2
   3 include (__DIR__ . '/redis.inc');
   4
   5 $r = NewRedisTestInstance();
   6 ExtRedisScan::$prefix = GetTestKeyName(__FILE__) . ':';
   7 $r->setOption(Redis::OPT_PREFIX, ExtRedisScan::$prefix);
   8 $r->setOption(Redis::OPT_SCAN, Redis::SCAN_RETRY);
   9 $ret = $r->delete('scan');
  10 $ret = $r->mset(array('key:one' => 'one', 'key:two' => 'two',
  11                'key:three' => 'three','key:four' => 'four'));
  12
  13 /*
  14  * The PHP5 extension we're patterning after doesn't try to turn
  15  * OPT_PREFIX into a pattern when doing scan(), so scan returns
  16  * global state. Filter it ourselves.
  17  */
  18 function performScan($fn){
  19         $cursor = null;
  20         $returns = array();
  21         while (($retval = $fn($cursor))){
  22                 foreach ($retval as $key){
  23
  24                         if (strstr($key, ExtRedisScan::$prefix) !== false){
  25                                 $returns []= substr($key, strlen(ExtRedisScan::$prefix));
  26                         }
  27                 }
  28         }
  29         // Normalize return order
  30         sort($returns);
  31         return $returns;
  32 }
  33
  34 try {
  35         $ret = performScan(function(&$cursor) use($r){
  36                 // Scan without patterns.
  37                 return $r->scan($cursor);
  38         });
  39         var_dump($ret);
  40
  41         $ret = performScan(function(&$cursor) use($r){
  42                 // catch key:two and key:three
  43
  44                 return $r->scan($cursor, ExtRedisScan::$prefix . 'key:t*');
  45         });
  46         var_dump($ret);
  47
  48         $ret = performScan(function(&$cursor) use($r){
  49                 // nothing.
  50
  51                 return $r->scan($cursor, ExtRedisScan::$prefix . 'nokey:t*');
  52         });
  53         var_dump($ret);
  54 } finally {
  55         $r->delete('key:one');
  56         $r->delete('key:two');
  57         $r->delete('key:three');
  58         $r->delete('key:four');
  59 }
  60
  61 abstract final class ExtRedisScan {
  62   public static $prefix;
  63 }