队列应用场景，自己实现队列（二）

原文详见：http://www.ucai.cn/blogdetail/7026?mid=1&f=12

可以在线运行查看效果哦!    

 

 

 

五、队列具体实现二：定义调用接口

 

 

<?php
/**
 * 优才网公开课示例代码
 *
 * 任务队列实现
 *
 * @author 优才网全栈工程师教研组  张友林
 * @see http://www.ucai.cn
 */

include_once('db.php');
class Queue
{

    /**
     * 把任务扔到队列
     *
     * @param string $taskphp   执行任务的程序
     * @param string $param     执行任务程序所用的参数

     * 例如，群发消息加入队列：
     * $arr = array(
     *      "uid" => 4,//发信息的人的UID
     *      "uids" => array(6,234,34,67,7888,2355), //接收信息的人的UID
     *      "content" => 'xxxxx',//信息内容
     *  );
     * $cqueue = new Queue();
     * $cqueue->add("/app/send_msg.php", serialize($arr));
     *
     */
    public function add($taskphp,$param)
    {
        $taskphp = mysql_real_escape_string($taskphp);
        //$param = mysql_real_escape_string($param);
        $param = $param;
        $sql = "insert into queue (taskphp, param) values('".$taskphp."', '".$param."')";
        $re = execute($sql);
        if ($re)
        {
            $pid = mysql_insert_id();
            return $pid;
        }
        else
        {
            return false;
        }
    }


    /**
     * 读取任务队列
     *
     * @param string $limit 一次取多少条
     */
     public function getQueueTask($limit = 1000)
     {
        $limit = (int)$limit;
        $sql = "select id, taskphp, param from queue  where status = 0 order by id asc";
        $re = query($sql);
        return $re;
     }

    /**
     * 更新任务状态
     *
     * @param string $limit 一次取多少条
     */
     public function updateTaskByID($id)
     {
        $id = (int)$id;
        $mtime = time();
        $sql = "update queue  set status =1, mtime = ".$mtime." where id = ".$id;
        $re = execute($sql);
        return $re;
     }


     public static function a2s($arr)
    {
        $str = "";
        foreach ($arr as $key => $value)
        {
            if (is_array($value))
            {
                foreach ($value as $value2)
                {
                    $str .= urlencode($key) . "[]=" . urlencode($value2) . "&";
                }
            }
            else
            {
                $str .= urlencode($key) . "=" . urlencode($value) . "&";
            }
        }
        return $str;
    }

    public static function s2a($str)
    {
        $arr = array();
        parse_str($str, $arr);
        return $arr;
    }

}

?>

$cqueue = new Queue();

 

 

1：加入队列接口

  //$param1 为执行任务的程序，$param2 为程序参数，可以为序列化的数据

  $cqueue->add($param1,$param2);

 

2:  读取队列接口

  $tasks = $cqueue->getQueueTask($limit = 1000);

 

3：更新任务状态

  $cqueue->updateTaskStatus($id);

 

4：a2s是自定义的一个数组转换字符串方法，这里不要使用json_encode，容易出现问题，同样，从数据库中取出转换为数组的时候，使用s2a方法

  $re = $cqueue->add("sendMsg.php", Queue::a2s($arr));

 

 

六、队列具体实现三：写执行队列的程序

 

根据设计，执行队列的程序文件是 do_queue.php ， 它的主要功能是把任务从队列表里取出来，并且在后台执行。

do_queue.php部分代码：

$phpcmd = exec("which php");    //查找到php安装位置
$cqueue = new Queue();
$tasks = $cqueue->getQueueTask(200);
foreach ($tasks as $t)
{
    $taskphp = $t['taskphp'];
    $param = $t['param'];
    $job = $phpcmd . " " . escapeshellarg($taskphp) . " " . escapeshellarg($param);
    system($job);
}

 

七、具体任务的业务实现

还是拿群发消息来做例子，我们需要写好一个群发消息的程序，这个程序接收事先定义好的参数，然后根据参数调用发消息的接口把消息发送出去。

这个一般由做业务功能的工程师实现。但是架构师事先得写文档例子，教会别人使用。

 

send_msg.php：

$para = $argv[1];
$arr = unserialize($para);
$cmessage = new Message();
foreach($arr['uids'] as $touid)
{
    $cmessage->send($arr['uid'], $touid, $arr['content']);
}

八 、服务器部署一：配置crontab
        咱们执行队列的程序都写好了， 这个程序怎么触发呢，当然就要用到linux的定时任务，每隔一定的时间，执行do_queue.php一次。但是呢，这里不是直do_queue.php，咱们再提高一层，加个调度程序cron_mission.php， 在cron_mission.php里面调用do_queue.php。


配置定时任务 crontab：

crontab –e

* * * * *  cd /ucai/schedule; php cron_mission.php >> cron_mission.log


#可以先使用crontab -l查看本机已经使用的定时任务

 

 

九、服务器部署二：写定时任务调度程序

思路：将定时任务写入到任务调度程序cron_mission.php中，这样可以在cron_mission.php中灵活控制队列任务。相比较直接通过crontab控制doQueue.php而言，避免了频繁修改服务器上的crontab，从安全，便于维护等角度来说，都是上策。

 

cron_mission.php 示例：

if ($minute % 5 == 0)
{
    if(chdir($site_dir."app/")) {
        $cmd = "$phpcmd do_queue.php > do_queue.log &";
        echo '[' , $ymd , ' ' , $hour , ':' , $minute , '] ' , $cmd , "\n";
        system($cmd);
    }
}

 

十、开启多进程并发执行队列

 

思路：对任务序列进行编号，数据库中执行的时候

where条件加上id%每个队列要执行任务总数=队列编号

这样可以避免重复处理

 

例如：每个进程执行10条任务，修改如下

 

1：定时任务的修改

修改前：

if ($minute % 5 == 0)
{
    if(chdir($site_dir."app/")) {
        $cmd = "$phpcmd do_queue.php > do_queue.log &";
        echo '[' , $ymd , ' ' , $hour , ':' , $minute , '] ' , $cmd , "\n";
        system($cmd);
    }
}

修改后：

if ($minute % 5 == 0)
{
    for ($i=0; $i < 10; $i++) {
        $cmd = "$phpcmd doQueue.php 10 $i>> doQueueMission".date('Y-m-d').".log  ";
        echo  date("Y-m-d H:i:s") . "\t : " .$cmd."\n";
        system($cmd);
    }
}

//每次进行10个进程，$i来区分是当前的进程标示

 

 

2：队列执行程序的修改

修改前：

$phpcmd = 'D:\work\wamp\bin\php\php5.3.10\php ';
$cqueue = new Queue();
$tasks = $cqueue->getQueueTask(200);

修改后：

 

$phpcmd = 'D:\work\wamp\bin\php\php5.3.10\php ';
$total=$argv[1];
$i=$argb[2];
$cqueue = new Queue();
$tasks = $cqueue->getQueueTask($total,$i,200);

 

 

3：取队列接口的修改

修改前：

 

public function getQueueTask($limit = 1000)
     {
        $limit = (int)$limit;
        $sql = "select id, taskphp, param from queue  where status = 0 order by id asc";
        $re = query($sql);
        return $re;
     }

修改后：

 

 

public function getQueueTask($total,$i,$limit = 1000)
     {
        $limit = (int)$limit;
        $sql = "select id, taskphp, param from queue  where status = 0 and id%$total=$i order by id asc";
        $re = query($sql);
        return $re;
     }

 

 

4：需要关注服务器压力

进程数定为多少，取决于服务器压力

 

十一、实现任务优先级 

1：任务存储表加优先级字段

       在数据表里，加一个优先级字段，按字段值的数值大小来区分优先级。

2：修改取队列任务接口，按优先级取

       同样是在sql语句中增加order by。

十二、记录队列日志

1：关键地方加echo

2：shell脚本的>>和>的各自作用 

总结：

        我们这里的队列实现借助了服务器的计划任务来实现，例如linux中的crontab，这本身是linux系统中的一个程序，平时我们还可以使用他来进行定时执行.sh脚本，例如将数据库备份打包并ftp传送到指定服务器上，这个功能不需要借助php脚本，直接用.sh脚本就可以实现。在这里我们巧妙的将crontab和php脚本结合，并且使用crontab来不断调用一个队列调度接口cronMission.php，再通过cronMission.php直接来控制具体什么时候或者是满足什么条件来执行什么队列任务。

这里面几个需要注意的地方

       1：往数据库中存取数据时，不要直接使用json_encode或者json_decode，容易造成一些意外问题，在代码中，我们定义了a2s和s2a两个方法，分别是处理数组转为字符串，和从数据库中读取字符串后转为数组。

       2：当任务量比较大，同时服务器负载又没有充分利用的时候，可以使用多进程并发处理，在并发处理的时候需要考虑一个问题，就是如何避免重复，在这里我们使用了，对队列任务进行标记，每次从数据库中读取一个进程需要处理的一批任务，使用数据库中id与批次标示取余等于0的方法来区分，避免不同批次的队列，重复处理相同任务。（上面步骤10中有具体实现）