基于swoole扩展实现真正的PHP数据库链接池

转自：  http://rango.swoole.com/archives/265

PHP的数据库链接池一直以来都是一个难题，不少从PHP语言转向Java的项目，大多数缘由都是由于Java有更好的链接池实现。PHP的MySQL扩展提供了长链接的API，但在PHP机器数量较多，规模较大的状况下，mysql_pconnect非但不能节约MySQL资源，反而会加重数据库的负荷。

假设有100台PHP的应用服务器，每一个机器须要启动100个apache或fpm工做进程，那每一个进程都会产生一个长链接到MySQL。这一共会产生1万个My SQL链接。你们都知道MySQL是每一个链接会占用1个线程。那MYSQL就须要建立1万个线程，这样大量的系统资源被浪费在线程间上下文切换上。而你的业务代码中并非全部地方都在作数据库操做，因此这个就是浪费的。

链接池就不一样了，100个worker进程，公用10个数据库链接便可，当操做完数据库后，当即释放资源给其余worker进程。这样就算有100台PHP的服务器，那也只会建立1000个MySQL的链接，彻底能够接受的。

之前确实没有好的办法来解决此问题的，如今有了swoole扩展，利用swoole提供的task功能能够很方便作出一个链接池来。

代码以下：

$serv = new swoole_server("127.0.0.1", 9508);
$serv->set(array(
    'worker_num' => 100,
    'task_worker_num' => 10, //MySQL链接的数量
));

function my_onReceive($serv, $fd, $from_id, $data)
{
    //taskwait就是投递一条任务，这里直接传递SQL语句了
    //而后阻塞等待SQL完成
    $result = $serv->taskwait("show tables");
    if ($result !== false) {
        list($status, $db_res) = explode(':', $result, 2);
        if ($status == 'OK') {
            //数据库操做成功了，执行业务逻辑代码，这里就自动释放掉MySQL链接的占用
            $serv->send($fd, var_export(unserialize($db_res), true) . "\n");
        } else {
            $serv->send($fd, $db_res);
        }
        return;
    } else {
        $serv->send($fd, "Error. Task timeout\n");
    }
}

function my_onTask($serv, $task_id, $from_id, $sql)
{
    static $link = null;
    if ($link == null) {
        $link = mysqli_connect("127.0.0.1", "root", "root", "test");
        if (!$link) {
            $link = null;
            $serv->finish("ER:" . mysqli_error($link));
            return;
        }
    }
    $result = $link->query($sql);
    if (!$result) {
        $serv->finish("ER:" . mysqli_error($link));
        return;
    }
    $data = $result->fetch_all(MYSQLI_ASSOC);
    $serv->finish("OK:" . serialize($data));
}

function my_onFinish($serv, $data)
{
    echo "AsyncTask Finish:Connect.PID=" . posix_getpid() . PHP_EOL;
}

$serv->on('Receive', 'my_onReceive');
$serv->on('Task', 'my_onTask');
$serv->on('Finish', 'my_onFinish');
$serv->start();

 

这里task_worker_num就是要启用的数据库链接池数量，worker进程为100时，链接池数量为10就能够设置为worker_num = 100, task_worker_num = 10。