MultiHttp/src/MultiRequest.php at 0c4864e573e8a2fcad2f14f02b32ecb666468624 · sinacms/MultiHttp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
<?php

namespace MultiHttp;

use MultiHttp\Exception\InvalidArgumentException;

/**
 *
 * @author  qiangjian@staff.sina.com.cn sunsky303@gmail.com
 * Date: 2017/6/9
 * Time: 15:09
 * @version $Id: $
 * @since 1.0
 * @copyright Sina Corp.
 */
class MultiRequest
{
    protected static $requestPool = array();
    protected static $multiHandler;
    private static $instance;

    protected function __construct()
    {
    }

    public static function create()
    {
        if (!(self::$instance instanceof self)) {
            self::$instance = new self;
        }
        self::prepare();
        return self::$instance;
    }

    protected static function prepare()
    {
        self::$multiHandler = curl_multi_init();
    }

    public function add($method, $uri, $payload, array $options = array())
    {
        $options = array(
                'method' => $method,
                'url' => $uri,
                'data' => $payload,
            ) + $options;
        $this->addOptions(array($options));
        return $this;
    }

    /**
     * @param array $URLOptions
     * example: array(array('url'=>'http://localhost:9999/','timeout'=>1, 'method'=>'POST', 'data'=>'aa=bb&c=d'))
     * @return $this
     */
    public function addOptions(array $URLOptions)
    {
        foreach ($URLOptions as $options) {
            $request = Request::create()->addOptions($options)->applyOptions();
            if (isset($options['callback'])) {
                $request->onEnd($options['callback']);
            }
            $this->import($request);
        }
        return $this;
    }

    public function import(Request $request)
    {
        if (!is_resource($request->curlHandle)) {
            throw new InvalidArgumentException('Request curl handle is not initialized');
        }
        curl_multi_add_handle(self::$multiHandler, $request->curlHandle);
        self::$requestPool[] = $request;
        return $this;
    }

    /**
     * @return array(Response)
     */
    public function execute()
    {
        $sleepTime = 1000;//microsecond, prevent  CPU 100%
        do {
            curl_multi_exec(self::$multiHandler, $active);
            // bug in PHP 5.3.18+ where curl_multi_select can return -1
            // https://bugs.php.net/bug.php?id=63411
            if (curl_multi_select(self::$multiHandler) == -1) {
                usleep($sleepTime);
            }
            usleep($sleepTime);
        } while ($active);
        $return = array();
        foreach (self::$requestPool as $request) {
            $response = $request->makeResponse(true);
            $func = $response->request->endCallback();
            if (isset($func)) {
                $func($response);
            }
            $return[] = $response;
            curl_multi_remove_handle(self::$multiHandler, $request->curlHandle);
            curl_close($request->curlHandle);
        }
        curl_multi_close(self::$multiHandler);
        self::$requestPool = array();
        return $return;
    }

}