git.m6w6.name Git - m6w6/ext-http/blob - tests/HttpRequestPool_007.phpt

   1 --TEST--
   2 HttpRequestPool chain with libevent
   3 --SKIPIF--
   4 <?php
   5 include 'skip.inc';
   6 checkcls('HttpRequest');
   7 checkcls('HttpRequestPool');
   8 skipif(!http_support(HTTP_SUPPORT_EVENTS), "need libevent support");
   9 ?>
  10 --FILE--
  11 <?php
  12
  13 echo "-TEST\n";
  14
  15 set_time_limit(0);
  16 ini_set('error_reporting', E_ALL);
  17 ini_set('html_errors', 0);
  18
  19 class Request extends HttpRequest
  20 {
  21         public $pool;
  22
  23         function onFinish()
  24         {
  25                 $this->pool->detach($this);
  26
  27                 $u = $this->getUrl();
  28                 $c = $this->getResponseCode();
  29                 $b = $this->getResponseBody();
  30
  31                 printf("%d %s %d\n", $c, $u, strlen($b));
  32
  33                 if ($c == 200 && $this->pool->dir) {
  34                         file_put_contents($this->pool->all[$u], $b);
  35                 }
  36
  37                 if ($a = each($this->pool->rem)) {
  38                         list($url, $file) = $a;
  39                         $r = new Request(
  40                                 $url,
  41                                 HttpRequest::METH_GET,
  42                                 array(
  43                                         'redirect'      => 5,
  44                                         'compress'      => GZIP,
  45                                         'timeout'       => TOUT,
  46                                         'connecttimeout' => TOUT,
  47                                         'lastmodified' => is_file($file)?filemtime($file):0
  48                                 )
  49                         );
  50                         $r->pool = $this->pool;
  51                         $this->pool->attach($r);
  52                 }
  53         }
  54
  55 }
  56
  57 class Pool extends HttpRequestPool
  58 {
  59         public $all;
  60         public $rem;
  61         public $dir;
  62
  63         public final function __construct($urls_file = 'urls.txt', $cache_dir = 'HttpRequestPool_cache')
  64         {
  65                 $this->dir = (is_dir($cache_dir) or @mkdir($cache_dir)) ? $cache_dir : null;
  66
  67                 $urls = file($urls_file);
  68                 shuffle($urls);
  69                 foreach (array_map('trim', $urls) as $url) {
  70                         $this->all[$url] = $this->dir ? $this->dir .'/'. md5($url) : null;
  71                 }
  72
  73                 if (RMAX) {
  74                         $now = array_slice($this->all, 0, RMAX);
  75                         $this->rem = array_slice($this->all, RMAX);
  76                 } else {
  77                         $now = $urls;
  78                         $this->rem = array();
  79                 }
  80
  81                 $this->enableEvents();
  82
  83                 foreach ($now as $url => $file) {
  84                         $r = new Request(
  85                                 $url,
  86                                 HttpRequest::METH_GET,
  87                                 array(
  88                                         'redirect'      => 5,
  89                                         'compress'  => GZIP,
  90                                         'timeout'       => TOUT,
  91                                         'connecttimeout' => TOUT,
  92                                         'lastmodified' => is_file($file)?filemtime($file):0
  93                                 )
  94                         );
  95                         $r->pool = $this;
  96                         $this->attach($r);
  97                 }
  98
  99                 $this->send();
 100         }
 101 }
 102
 103 define('GZIP', true);
 104 define('TOUT', 50);
 105 define('RMAX', 10);
 106 chdir(dirname(__FILE__));
 107
 108 $time = microtime(true);
 109 $pool = new Pool();
 110 printf("Elapsed: %0.3fs\n", microtime(true)-$time);
 111
 112 echo "Done\n";
 113 ?>
 114 --EXPECTF--
 115 %aTEST
 116 %d %s %d
 117 %d %s %d
 118 %d %s %d
 119 %d %s %d
 120 %d %s %d
 121 %d %s %d
 122 %d %s %d
 123 %d %s %d
 124 %d %s %d
 125 %d %s %d
 126 %d %s %d
 127 %d %s %d
 128 %d %s %d
 129 %d %s %d
 130 %d %s %d
 131 %d %s %d
 132 %d %s %d
 133 %d %s %d
 134 %d %s %d
 135 %d %s %d
 136 %d %s %d
 137 %d %s %d
 138 %d %s %d
 139 %d %s %d
 140 %d %s %d
 141 %d %s %d
 142 %d %s %d
 143 %d %s %d
 144 %d %s %d
 145 %d %s %d
 146 %d %s %d
 147 %d %s %d
 148 %d %s %d
 149 %d %s %d
 150 %d %s %d
 151 %d %s %d
 152 %d %s %d
 153 %d %s %d
 154 %d %s %d
 155 %d %s %d
 156 %d %s %d
 157 %d %s %d
 158 %d %s %d
 159 %d %s %d
 160 %d %s %d
 161 %d %s %d
 162 %d %s %d
 163 %d %s %d
 164 %d %s %d
 165 Elapsed: %fs
 166 Done