Swoole4.x之协程变量访问安全与协程链接池实现

访问安全问题

为何说有访问安全问题呢？传统地，在php的的环境中，不多有Phper遇到所谓变量安全访问问题。举个例子，代码大约以下:

class db
{
    protected static $instance;
    protected $dbCon;
    
    function __construct()
    {
        /*
         * 咱们这里用stdclass来模拟一个数据库链接
         */
        $this->dbCon = new \stdClass();
    }

    public static function getInstance()
    {
        if(!isset(self::$instance)){
            self::$instance = new db();
        }
        return self::$instance;
    }

    function dbCon()
    {
        return $this->dbCon;
    }
}

$con = db::getInstance()->dbCon();
$con->key = 'new';
var_dump($con->key);

这个是在fpm模式下，很常见的数据库链接单例模式的使用。乍一看没有问题，但实际上，在协程环境下，会出现链接跨协程使用问题,举例以下

go(function (){
    go(function (){
        db::getInstance()->dbCon()->key = 'one';
        //假设这sql执行了1s
        \co::sleep(1);
        var_dump(db::getInstance()->dbCon()->key);
    });
    go(function (){
        db::getInstance()->dbCon()->key = 'two';
        //假设这sql执行了0.1s
        \co::sleep(0.1);
        var_dump(db::getInstance()->dbCon()->key);
    });
});

咱们会发现，以上代码当中，协程2的数据污染到了协程1的数据，那么所以这样确定是不行的。

上下文管理器

为了解决这个问题，咱们引入协程上下文管理这样的概念，由此来实现每一个协程环境内的数据隔离。

class dbContext
{
    private $container = [];

    private static $instance;

    public static function getInstance()
    {
        if(!isset(self::$instance)){
            self::$instance = new dbContext();
        }
        return self::$instance;
    }

    function dbCon()
    {
        $cid = \co::getCid();
        if(!isset($this->container[$cid])){
            $this->container[$cid] = new stdClass();
            defer(function (){
                $this->destroy();
            });
        }
        return $this->container[$cid];
    }

    function destroy()
    {
        $cid = \co::getCid();
        if(!isset($this->container[$cid])){
            unset($this->container[$cid]);
        }
    }
}

go(function (){
    go(function (){
        dbContext::getInstance()->dbCon()->key = 'one';
        //假设这sql执行了1s
        \co::sleep(1);
        var_dump(dbContext::getInstance()->dbCon()->key);
    });
    go(function (){
        dbContext::getInstance()->dbCon()->key = 'two';
        //假设这sql执行了0.1s
        \co::sleep(0.1);
        var_dump(dbContext::getInstance()->dbCon()->key);
    });
});

以上代码中，咱们用每一个协程的id,来做为每一个协程栈的数据token,用了defer方法，实现了每一个协程退出的时候的数据自动清理，从而避免了内存泄露。

通用版本的链接池与协程上下文管理

咱们不难发现，以上代码中，实际上依旧是短链接的管理方式，没办法对连接进行复用，因为本文章仅作基础原理讲解之用，具体有兴趣的同窗，能够查看下Easyswoole这个框架的链接池和协程上下文管理器，项目主页在 www.easyswoole.com ，若以为喜欢，有帮助，能够给easyswoole的github仓库点个赞。