php基于curl_multi*函數(shù)簇實現(xiàn)批量rpc調(diào)用

背景

在web應用中讨韭,為了更好的拆分服務,一個較大的業(yè)務會拆分為多個自服務,每個子服務獨自完成部分功能罪治,眾多子服務一起支撐較大的業(yè)務瘦麸。

這就不可避免的遇到一個復雜的業(yè)務需要串行地調(diào)用多個子服務的場景,特別是一些業(yè)務邏輯上沒有依賴的部分箍鼓,這時候铐殃,如果可以有批量rpc調(diào)用,那就如虎添翼了是整。

好消息是舵盈,鳥哥開發(fā)的Yar就是支持并行rpc調(diào)用的框架筒愚,他是以擴展的形式提供并行RPC功能。據(jù)說在微博內(nèi)部已經(jīng)廣發(fā)使用了,然而楚堤,對于線上已經(jīng)在跑的業(yè)務岔乔,給編譯一個擴展嘿歌,運營成本還是挺高的茄唐。

思路

PHP原生的curl_multi*函數(shù)簇,也可以支持一次發(fā)送多個RPC調(diào)用的作用厨幻,使用方式為:

實現(xiàn)

<?php  
/** 
 * 基于 curl_multi 的并行處理 
 * 
 * 使用示例 
 * $url1 = "http://baidu.com";
 * $url2 = "http://sina.com";
 *
 * $objMulti = new Multi();
 * $objMulti->init();
 * $objMulti->register('url1', $url1);
 * $objMulti->register('url2', $url2);
 * $objMulti->call();
 * $res = $objMulti->getResult('url1');
 * var_dump($res);
 * $res = $objMulti->getResult('url2');
 * var_dump($res);
 */  
class Multi {  
    /* default 500k */
    const MAX_RESPONSE_SIZE        = 512000;    

    /* max redirs */
    const MAX_REDIRS               = 1;

    /*鏈接重試*/
    const MAX_CONNECT_RETRY        = 3;

    /* 默認鏈接超時時間 */
    const MAX_CONNECT_TIMEOUT      = 1000;

    /*默認超時時間*/
    const MAX_TIME_OUT             = 3000;

    /*UTF8 字符編碼*/
    const ENCODE_UTF8              = 'utf-8';

    /* cURL 多個句柄 */
    protected $_obj_multi_handler  = null;       

    /* curl句柄列表 */
    protected $_arr_curl_handler   = array(); 

    /* curl 參數(shù) */
    protected $_arr_curl_options   = array();

    /* 調(diào)用返回的結(jié)果 */
    protected $_arr_result         = array();

    /**
    * @desc   初始化curl_multi句柄
    * @author dayeshisir(dayeshisir@weidian.com)
    * @date   2016-10-09 
    * @param  
    * @return
    * @note
    */
    public function init($config = array()) {
        if (is_null($this->_obj_multi_handler)) {  
            $this->_obj_multi_handler = curl_multi_init();  
        }  

        $str_user_agent = isset($_SERVER['HTTP_USER_AGENT']) ? $_SERVER['HTTP_USER_AGENT'] : '';
        $str_referer    = isset($_SERVER['HTTP_REFERER']) ? $_SERVER['HTTP_REFERER'] : '';

        $this->_arr_curl_options = array(
            CURLOPT_NOSIGNAL          => 1,          //注意棠隐,毫秒超時一定要設置這個
            CURLOPT_FOLLOWLOCATION    => false,
            CURLOPT_HEADER            => false,
            CURLOPT_MAXREDIRS         => self::MAX_REDIRS,
            CURLOPT_RETURNTRANSFER    => true,
            CURLOPT_CONNECTTIMEOUT_MS => self::MAX_CONNECT_TIMEOUT,
            CURLOPT_TIMEOUT_MS        => self::MAX_TIME_OUT,
            CURLOPT_USERAGENT         => $str_user_agent,
            CURLOPT_REFERER           => $str_referer,
            CURLOPT_ENCODING          => '',
        );

        return true;
    }
  
    /** 
     * @desc  注冊待調(diào)用的服務
     * @param key          : string : 用來區(qū)分各個自調(diào)用 
     * @param url          : string : url地址挖藏,可包含問號  
     * @param date         : array  : 入?yún)?shù)組,key-value對炸宵,post或get的參數(shù)
     * @param request_type : string : 請求的類型,例如post,get 
     * @param config       : array  : 附加的配置铃将,具體數(shù)值例如:
     * array( 
     *  CURLOPT_COOKIE     => string 設定HTTP請求中"Cookie: "部分的內(nèi)容奥此。多個cookie用分號分隔,分號后帶一個空格(例如丈牢, "fruit=apple; colour=red")垫桂。 
     *  CURLOPT_REFERER    => string 請求頭中"Referer: "的內(nèi)容 
     *  CURLOPT_USERAGENT  => string "User-Agent: "頭的字符串。如:“Mozilla/4.0 (compatible; MSIE 5.01; Windows NT 5.0)” 
     *  CURLOPT_HTTPHEADER => array  header頭列表,如:array('host:http://abc.com','Content-type: text/plain') 
     *  CURLOPT_TIMEOUT    => int    超時時間,秒 
     * ) 
     * 
     * @return resource 
     */  
    public function register($key, $url, $data = array(),  $config = array(), $request_type = 'post') {  
        if (null === $this->_obj_multi_handler) {
            return false;
        }

        //初始化curl會話  
        $obj_ch = curl_init();  
  
        //curl參數(shù)  
        extract($this->_arr_curl_options);
        $option = $this->_arr_curl_options;
  
        //get 參數(shù)處理(url處理)  
        if ('get' === $request_type && !empty($data)) {  
            $url .= (strpos($url, '?') ? '&' : '?') . http_build_query($data);  
        }  

        $option[CURLOPT_URL] = $url;  
  
        //post 參數(shù)處理  
        if ('post' === $request_type && empty($data)) {  
            $option[CURLOPT_POSTFIELDS] = $data;  
        }  
  
        //其它參數(shù)合并  
        $option += $config;

        //批量設置會話參數(shù)  
        curl_setopt_array($obj_ch, $option);  

        // 添加到curl句柄
        if (0 !== curl_multi_add_handle($this->_obj_multi_handler, $obj_ch)) {
            return false;
        }

        // 如果原來有相同的$key,則刪除之
        if (isset($this->_arr_curl_handler[$key])) {
            curl_multi_remove_handle($this->_obj_multi_handler, $this->_arr_curl_handler[$key]);  //移除原curl句柄  
            curl_close($this->_arr_curl_handler[$key]);                                          //關閉原curl會話  
            unset($this->_arr_curl_handler[$key]);                                               //刪除原$k  
        }

        $this->_arr_curl_handler[$key] = $obj_ch;
  
        return true;  
    }  
  
    /** 
     * @desc 執(zhí)行批量請求,返回批量數(shù)據(jù)
     * @param  null
     * @return array 
     */  
    public function call() {   
        //無批量執(zhí)行句柄茶没,返回空數(shù)組  
        if (is_null($this->_obj_multi_handler)) {  
            return $this->_arr_result;  
        }  
  
        $active = null;
        //執(zhí)行  
        do {  
            $mrc = curl_multi_exec($this->_obj_multi_handler, $active);  
        } while ($mrc == CURLM_CALL_MULTI_PERFORM);  

        while ($active && $mrc == CURLM_OK) {  
            while(curl_multi_exec($this->_obj_multi_handler, $active) === CURLM_CALL_MULTI_PERFORM);

            if (curl_multi_select($this->_obj_multi_handler) != -1) {  
                do {  
                    $mrc = curl_multi_exec($this->_obj_multi_handler, $active);  
                } while ($mrc == CURLM_CALL_MULTI_PERFORM);  
            }  
        }  

        //獲取返回結(jié)果  
        foreach ($this->_arr_curl_handler as $key => $curl_handler) {  
            $strResult = curl_multi_getcontent($curl_handler);  
            $this->_arr_result[$key] = json_decode($strResult, true);
            curl_multi_remove_handle($this->_obj_multi_handler, $curl_handler);//移除句柄  
            curl_close($curl_handler);//關閉會話  
        }  
        curl_multi_close($this->_obj_multi_handler);//關閉批量會話  
  
        //清理數(shù)據(jù),恢復初始狀態(tài)辱揭,為下一次使用做好準備  
        $this->_obj_multi_handler = null;
        $this->_arr_curl_handler  = array();
  
        return $this->_arr_result;  
    } 

    /**
    * @desc   獲取結(jié)果
    * @param  key : string : 用來區(qū)分子服務的鍵值
    * @return array 
    */
    public function getResult($key) {
        return isset($this->_arr_result[$key]) ? $this->_arr_result[$key] : array();
    } 
}  

$url1 = "http://www.baidu.com";
$url2 = "http://www.sina.com.cn";

$objMulti = new Multi();
$objMulti->init();
$objMulti->register('url1', $url1);
$objMulti->register('url2', $url2);
$objMulti->call();
$res = $objMulti->getResult('url1');
var_dump($res);
$res = $objMulti->getResult('url2');
var_dump($res);
最后編輯于
?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市馋没,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖鉴扫,帶你破解...
    沈念sama閱讀 218,122評論 6 505
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件姐赡,死亡現(xiàn)場離奇詭異,居然都是意外死亡悉抵,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,070評論 3 395
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人汹押,你說我怎么就攤上這事矿筝。” “怎么了鲸阻?”我有些...
    開封第一講書人閱讀 164,491評論 0 354
  • 文/不壞的土叔 我叫張陵跋涣,是天一觀的道長。 經(jīng)常有香客問我鸟悴,道長陈辱,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,636評論 1 293
  • 正文 為了忘掉前任细诸,我火速辦了婚禮沛贪,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘震贵。我一直安慰自己利赋,他們只是感情好,可當我...
    茶點故事閱讀 67,676評論 6 392
  • 文/花漫 我一把揭開白布猩系。 她就那樣靜靜地躺著媚送,像睡著了一般。 火紅的嫁衣襯著肌膚如雪寇甸。 梳的紋絲不亂的頭發(fā)上塘偎,一...
    開封第一講書人閱讀 51,541評論 1 305
  • 那天,我揣著相機與錄音拿霉,去河邊找鬼吟秩。 笑死,一個胖子當著我的面吹牛绽淘,可吹牛的內(nèi)容都是我干的涵防。 我是一名探鬼主播,決...
    沈念sama閱讀 40,292評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼沪铭,長吁一口氣:“原來是場噩夢啊……” “哼壮池!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起杀怠,我...
    開封第一講書人閱讀 39,211評論 0 276
  • 序言:老撾萬榮一對情侶失蹤火窒,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后驮肉,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體熏矿,經(jīng)...
    沈念sama閱讀 45,655評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,846評論 3 336
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了票编。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片褪储。...
    茶點故事閱讀 39,965評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡备埃,死狀恐怖讼呢,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情煌集,我是刑警寧澤昔榴,帶...
    沈念sama閱讀 35,684評論 5 347
  • 正文 年R本政府宣布辛藻,位于F島的核電站,受9級特大地震影響互订,放射性物質(zhì)發(fā)生泄漏吱肌。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,295評論 3 329
  • 文/蒙蒙 一仰禽、第九天 我趴在偏房一處隱蔽的房頂上張望氮墨。 院中可真熱鬧,春花似錦吐葵、人聲如沸规揪。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,894評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽猛铅。三九已至,卻和暖如春凤藏,著一層夾襖步出監(jiān)牢的瞬間奕坟,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,012評論 1 269
  • 我被黑心中介騙來泰國打工清笨, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人刃跛。 一個月前我還...
    沈念sama閱讀 48,126評論 3 370
  • 正文 我出身青樓抠艾,卻偏偏與公主長得像,于是被迫代替她去往敵國和親桨昙。 傳聞我的和親對象是個殘疾皇子检号,可洞房花燭夜當晚...
    茶點故事閱讀 44,914評論 2 355

推薦閱讀更多精彩內(nèi)容