基于Swoole的通用连接池 - 数据库连接池

21

连接池

open-smf/connection-pool 是一个基于Swoole的通用连接池,常被用作数据库连接池。

依赖

依赖 版本
PHP >=7.0.0
Swoole >=4.2.9 Recommend 4.2.13+

安装

通过Composer安装。
composer require "open-smf/connection-pool:~1.0"

使用

更多示例
  • 可用的连接器
连接器 说明
CoroutineMySQLConnector Swoole\Coroutine\MySQL的实例
CoroutinePostgreSQLConnector Swoole\Coroutine\PostgreSQL的实例,编译Swoole时需要添加参数--enable-coroutine-postgresql
CoroutineRedisConnector Swoole\Coroutine\Redis的实例
PhpRedisConnector Redis的实例,需要安装redis
YourConnector YourConnector必须实现接口ConnectorInterface,任何对象均可作为连接实例
  • 基本用法
use Smf\ConnectionPool\ConnectionPool;
use Smf\ConnectionPool\Connectors\CoroutineMySQLConnector;
use Swoole\Coroutine\MySQL;

go(function () {
    // MySQL连接数区间:[10, 30]
    $pool = new ConnectionPool(
        [
            'minActive'         => 10,
            'maxActive'         => 30,
            'maxWaitTime'       => 5,
            'maxIdleTime'       => 20,
            'idleCheckInterval' => 10,
        ],
        new CoroutineMySQLConnector, // 指明连接器实例,这里使用协程MySQL连接器,这样就可以创建一个协程MySQL的数据库连接池
        [
            'host'        => '127.0.0.1',
            'port'        => '3306',
            'user'        => 'root',
            'password'    => 'xy123456',
            'database'    => 'mysql',
            'timeout'     => 10,
            'charset'     => 'utf8mb4',
            'strict_type' => true,
            'fetch_mode'  => true,
        ]
    );
    echo "初始化连接池...\n";
    $pool->init();
    defer(function () use ($pool) {
        echo "关闭连接池...\n";
        $pool->close();
    });

    echo "从连接池中借出连接...\n";
    /**@var MySQL $connection */
    $connection = $pool->borrow();
    
    // 执行查询语句
    $status = $connection->query('SHOW STATUS LIKE "Threads_connected"');
    
    echo "用完连接后,尽快归还...\n";
    $pool->return($connection);
    
    var_dump($status);
});
  • 在Swoole Server中的用法
use Smf\ConnectionPool\ConnectionPool;
use Smf\ConnectionPool\ConnectionPoolTrait;
use Smf\ConnectionPool\Connectors\CoroutineMySQLConnector;
use Smf\ConnectionPool\Connectors\PhpRedisConnector;
use Swoole\Coroutine\MySQL;
use Swoole\Http\Request;
use Swoole\Http\Response;
use Swoole\Http\Server;

class HttpServer
{
    use ConnectionPoolTrait;

    protected $swoole;

    public function __construct(string $host, int $port)
    {
        $this->swoole = new Server($host, $port);

        $this->setDefault();
        $this->bindWorkerEvents();
        $this->bindHttpEvent();
    }

    protected function setDefault()
    {
        $this->swoole->set([
            'daemonize'             => false,
            'dispatch_mode'         => 1,
            'max_request'           => 8000,
            'open_tcp_nodelay'      => true,
            'reload_async'          => true,
            'max_wait_time'         => 60,
            'enable_reuse_port'     => true,
            'enable_coroutine'      => true,
            'http_compression'      => false,
            'enable_static_handler' => false,
            'buffer_output_size'    => 4 * 1024 * 1024,
            'worker_num'            => 4, // 每个Worker持有一个独立的连接池
        ]);
    }

    protected function bindHttpEvent()
    {
        $this->swoole->on('Request', function (Request $request, Response $response) {
            $pool1 = $this->getConnectionPool('mysql');
            /**@var MySQL $mysql */
            $mysql = $pool1->borrow();
            $status = $mysql->query('SHOW STATUS LIKE "Threads_connected"');
            // 用完连接后,尽快归还
            $pool1->return($mysql);


            $pool2 = $this->getConnectionPool('redis');
            /**@var Redis $redis */
            $redis = $pool2->borrow();
            $clients = $redis->info('Clients');
            // 用完连接后,尽快归还
           $pool2->return($redis);

            $json = [
                'status'  => $status,
                'clients' => $clients,
            ];
            // Other logic
            // ...
            $response->header('Content-Type', 'application/json');
            $response->end(json_encode($json));
        });
    }

    protected function bindWorkerEvents()
    {
        $createPools = function () {
            // 所有的MySQL连接数区间:[4 workers * 2 = 8, 4 workers * 10 = 40]
            $pool1 = new ConnectionPool(
                [
                    'minActive' => 2,
                    'maxActive' => 10,
                ],
                new CoroutineMySQLConnector,
                [
                    'host'        => '127.0.0.1',
                    'port'        => '3306',
                    'user'        => 'root',
                    'password'    => 'xy123456',
                    'database'    => 'mysql',
                    'timeout'     => 10,
                    'charset'     => 'utf8mb4',
                    'strict_type' => true,
                    'fetch_mode'  => true,
                ]);
            $pool1->init();
            $this->addConnectionPool('mysql', $pool1);

            // 所有Redis连接数区间:[4 workers * 5 = 20, 4 workers * 20 = 80]
            $pool2 = new ConnectionPool(
                [
                    'minActive' => 5,
                    'maxActive' => 20,
                ],
                new PhpRedisConnector,
                [
                    'host'     => '127.0.0.1',
                    'port'     => '6379',
                    'database' => 0,
                    'password' => null,
                ]);
            $pool2->init();
            $this->addConnectionPool('redis', $pool2);
        };
        $closePools = function () {
            $this->closeConnectionPools();
        };
        // Worker启动时创建MySQL和Redis连接池
        $this->swoole->on('WorkerStart', $createPools);
        
        // Worker正常退出或错误退出时,关闭连接池,释放连接
        $this->swoole->on('WorkerStop', $closePools);
        $this->swoole->on('WorkerError', $closePools);
    }

    public function start()
    {
        $this->swoole->start();
    }
}

// 启用协程Runtime来让PhpRedis扩展一键协程化
Swoole\Runtime::enableCoroutine(true);
$server = new HttpServer('0.0.0.0', 5200);
$server->start();
  • 本人已用于生产环境,表现稳定

监控图

贡献

Github,欢迎 Star & PR。

你可能感兴趣的

11 条评论
探索者 · 3月16日

对于使用的人来说,目的是执行一条sql语句,其他的都是无意义的,还增加了代码的复杂性。可以看看这个框架 https://github.com/lizhichao/one 对使用的人来说不需要关心是不是连接池,一切框架自动完成。

回复

0

换框架的成本更高

谢彪 作者 · 3月16日
0

在别人的宣传下搞自己的宣传,这不是招怼吗:)

netstu · 3月16日
huangzhhui · 3月16日

相当于每个协程占用一个连接,连接利用率不够高

回复

0

有何高见?

谢彪 作者 · 3月17日
0

其实也不是一直占用,最好用完了就归还

谢彪 作者 · 3月17日
0

只是例子中正好用下defer特性

谢彪 作者 · 3月17日
davidyanxw · 3月16日

支持开源,无论是框架还是composer包,最终选择权交给用户

回复

0

嗯,权衡一个度吧

谢彪 作者 · 3月17日
载入中...