Laravel或hyperf使用Db时如何给所有的select语句增加“/*FORCE_MASTER*/”

1. Laravel5.6或hyperf1.1

2. 业务场景

目前主从复制是使用了阿里云的rds,默认select语句查询的都是从库,但有一些场景需要查询主库,阿里云查询主库的方式是在select语句前面加 “/FORCE_MASTER/”,但是项目使用的大部分是模型或db生成器查询,想实现一个可以允许链式调用查询主库的方式

3. 期望得到的结果?(有啥想法的也可以提一下,说不定可以实现)

Db::connection("xpx")
            ->table('users')
            ->forceMaster()  // 加上此调用代表查询主库
            ->first();

期望生成的SQL为:

/*FORCE_MASTER*/ select * from `ecs_users` where `user_id` = ?

forceMaster() 这个方法的作用是期望在当前select语句中开头增加 “/FORCE_MASTER/” 字符串

4. 当前进度

目前只找到了在vendor中如何增加,但不知道怎么通过模型调用增加前缀字符串,找到的vendor中的代码及需要的改动如下:

hyperf中的文件路径为(Laravel中基本差不多):namespace Hyperf\Database\Query\Grammars;

 /**
     * Compile the "select *" portion of the query.
     *
     * @param array $columns
     * @return null|string
     */
    protected function compileColumns(Builder $query, $columns)
    {
        // If the query is actually performing an aggregating select, we will let that
        // compiler handle the building of the select clauses, as it will need some
        // more syntax that is best handled by that function to keep things neat.
        if (! is_null($query->aggregate)) {
            return;
        }

        $select = $query->distinct ? 'select distinct ' : 'select ';

        // 在vendor中改的话是在此处增加
        // $select = "/*FORCE_MASTER*/ {$select}";


        return $select . $this->columnize($columns);
    }
《L04 微信小程序从零到发布》
从小程序个人账户申请开始,带你一步步进行开发一个微信小程序,直到提交微信控制台上线发布。
《G01 Go 实战入门》
从零开始带你一步步开发一个 Go 博客项目,让你在最短的时间内学会使用 Go 进行编码。项目结构很大程度上参考了 Laravel。
最佳答案

把下述代码放到 \App\Providers\AppServiceProvider::register 里面(适用于 Laravel 5.6)

use Illuminate\Database\MySqlConnection;
use Illuminate\Database\Query\Builder;
use Illuminate\Database\Query\Grammars\Grammar;

Builder::macro('forceMaster', function () {
    $this->force_master = true;
    return $this;
});

Builder::macro('isForceMaster', function () {
    return $this->force_master ?? false;
});

\Illuminate\Database\Connection::resolverFor('mysql', function ($connection, $database, $prefix, $config) {
    $connection = new MySqlConnection($connection, $database, $prefix, $config);
    $connection->setQueryGrammar(new class extends Grammar {
        protected function compileColumns(Builder $query, $columns)
        {
            $sql = parent::compileColumns($query, $columns);
            if ($query->isForceMaster()) {
                $sql = '/*FORCE_MASTER*/ ' . $sql;
            }
            return $sql;
        }
    });
    return $connection;
});

最终成果:

file

2年前 评论
hengbo_liu (楼主) 2年前
讨论数量: 3

把下述代码放到 \App\Providers\AppServiceProvider::register 里面(适用于 Laravel 5.6)

use Illuminate\Database\MySqlConnection;
use Illuminate\Database\Query\Builder;
use Illuminate\Database\Query\Grammars\Grammar;

Builder::macro('forceMaster', function () {
    $this->force_master = true;
    return $this;
});

Builder::macro('isForceMaster', function () {
    return $this->force_master ?? false;
});

\Illuminate\Database\Connection::resolverFor('mysql', function ($connection, $database, $prefix, $config) {
    $connection = new MySqlConnection($connection, $database, $prefix, $config);
    $connection->setQueryGrammar(new class extends Grammar {
        protected function compileColumns(Builder $query, $columns)
        {
            $sql = parent::compileColumns($query, $columns);
            if ($query->isForceMaster()) {
                $sql = '/*FORCE_MASTER*/ ' . $sql;
            }
            return $sql;
        }
    });
    return $connection;
});

最终成果:

file

2年前 评论
hengbo_liu (楼主) 2年前

Laravel 5.7是如何处理呢?在 Laravel 5.7 中,不能使用 Connection::resolverFor 来直接修改连接解析器,因为该方法在 Laravel 5.5 之后已被移除。

7个月前 评论

讨论应以学习和精进为目的。请勿发布不友善或者负能量的内容,与人为善,比聪明更重要!