swoole channel之mysql连接池实现

2018年12月3日 323点热度 0人点赞 0条评论

上一篇文章介绍了利用channel来实现waitgroup和进程间通信的功能，本篇文章将继续利用channel来实现一个简单的mysql连接池，并且介绍利用新的的特性：defer来实现资源的回收

为什么要实现mysql连接池？
有以下几个原因：

保持长连接可以节省连接相关的开销(不过由于swoole本身常驻进程，所以只要不手工close，也还是长连接了)
mysql本身对连接有限制，所以每个请求（协程）都建立一个连接很容易导致mysql连接被打满

下面就来看一段代码，看swoole里实现一个连接池是怎么样的简单

<?php

use Swoole\Coroutine\MySQL;
class MysqlPool
{
    private static $instance;
    private $pool;  //连接池容器，一个channel
    private $config;

    /**
     * @param null $config
     * @return MysqlPool
     * @desc 获取连接池实例
     */
    public static function getInstance($config = null)
    {
        if (empty(self::$instance)) {
            if (empty($config)) {
                throw new RuntimeException("mysql config empty");
            }
            self::$instance = new static($config);
        }

        return self::$instance;
    }

    /**
     * MysqlPool constructor.
     * @param $config
     * @desc 初始化，自动创建实例,需要放在workerstart中执行
     */
    public function __construct($config)
    {
        if (empty($this->pool)) {
            $this->config = $config;
            $this->pool = new chan($config['pool_size']);
            for ($i = 0; $i < $config['pool_size']; $i++) {
                $mysql = new MySQL();
                $res = $mysql->connect($config);
                if ($res == false) {
                    //连接失败，抛弃常
                    throw new RuntimeException("failed to connect mysql server.");
                } else {
                    //mysql连接存入channel
                    $this->put($mysql);
                }
            }
        }
    }

    /**
     * @param $mysql
     * @desc 放入一个mysql连接入池
     */
    public function put($mysql)
    {
        $this->pool->push($mysql);
    }

    /**
     * @return mixed
     * @desc 获取一个连接，当超时，返回一个异常
     */
    public function get()
    {
        $mysql = $this->pool->pop($this->config['pool_get_timeout']);
        if (false === $mysql) {
            throw new RuntimeException("get mysql timeout, all mysql connection is used");
        }
        return $mysql;
    }

    /**
     * @return mixed
     * @desc 获取当时连接池可用对象
     */
    public function getLength()
    {
        return $this->pool->length();
    }

}

那如何使用呢？继续看一段代码：

<?php
require_once "mysql_pool.php";
$config = [
    'host' => '127.0.0.1',   //数据库ip
    'port' => 3306,          //数据库端口
    'user' => 'root',        //数据库用户名
    'password' => '123456', //数据库密码
    'database' => 'test',   //默认数据库名
    'timeout' => 0.5,       //数据库连接超时时间
    'charset' => 'utf8mb4', //默认字符集
    'strict_type' => true,  //ture，会自动表数字转为int类型
    'pool_size' => '3',     //连接池大小
    'pool_get_timeout' => 0.5, //当在此时间内未获得到一个连接，会立即返回。（表示所以的连接都已在使用中）
];

//创建http server
$http = new Swoole\Http\Server("0.0.0.0", 9501);
$http->set([
    //"daemonize" => true,
    "worker_num" => 1,
    "log_level" => SWOOLE_LOG_ERROR,
]);

$http->on('WorkerStart', function ($serv, $worker_id) use ($config) {
    //worker启动时,每个进程都初始化连接池，在onRequest中可以直接使用
    try {
        MysqlPool::getInstance($config);
    } catch (\Exception $e) {
        //初始化异常，关闭服务
        echo $e->getMessage() . PHP_EOL;
        $serv->shutdown();
    } catch (\Throwable $throwable) {
        //初始化异常，关闭服务
        echo $throwable->getMessage() . PHP_EOL;
        $serv->shutdown();
    }
});

$http->on('request', function ($request, $response) {

    //浏览器会自动发起这个请求，这也是很多人碰到的一个问题：
    //为什么我浏览器打开网站，收到了两个请求?
    if ($request->server['path_info'] == '/favicon.ico') {
        $response->end('');
        return;
    }

    //获取数据库
    if ($request->server['path_info'] == '/list') {

        go(function () use ($request, $response) {
            //从池子中获取一个实例
            try {
                $pool = MysqlPool::getInstance();
                $mysql = $pool->get();
                defer(function () use ($mysql) {
                    //利用defer特性，可以达到协程执行完成，归还$mysql到连接池
                    //好处是 可能因为业务代码很长，导致乱用或者忘记把资源归还
                    MysqlPool::getInstance()->put($mysql);
                    echo "当前可用连接数：" . MysqlPool::getInstance()->getLength() . PHP_EOL;
                });
                $result = $mysql->query("select * from test");
                $response->end(json_encode($result));
            } catch (\Exception $e) {
                $response->end($e->getMessage());
            }
        });
        return;
    }

    //模拟timeout, 浏览器打开4个tab，都请求 http://127.0.0.1:9501/timeout，前三个应该是等10秒出结果，第四个500ms后出超时结果
    //ps: chrome浏览器，需要加一个随机数，http://127.0.0.1:9501/timeout?t=0, http://127.0.0.1:9501/timeout?t=1, 因为chrome会对完全一样的url做并发请求限制
    echo "get request:".time().PHP_EOL;
    if ($request->server['path_info'] == '/timeout') {
        go(function () use ($request, $response) {
            //从池子中获取一个实例
            try {
                $pool = MysqlPool::getInstance();
                echo "当前可用连接数：" . $pool->getLength() . PHP_EOL;
                $mysql = $pool->get();
                echo "当前可用连接数：" . $pool->getLength() . PHP_EOL;
                defer(function () use ($mysql) {
                    //协程执行完成，归还$mysql到连接池
                    MysqlPool::getInstance()->put($mysql);
                    echo "当前可用连接数：" . MysqlPool::getInstance()->getLength() . PHP_EOL;
                });
                $result = $mysql->query("select * from test");
                \Swoole\Coroutine::sleep(10); //sleep 10秒,模拟耗时操作
                $response->end(json_encode($result));
            } catch (\Exception $e) {
                $response->end($e->getMessage());
            }
        });
        return;
    }

});

$http->start();