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);
    }
《L05 电商实战》
从零开发一个电商项目,功能包括电商后台、商品 & SKU 管理、购物车、订单管理、支付宝支付、微信支付、订单退款流程、优惠券等
《L03 构架 API 服务器》
你将学到如 RESTFul 设计风格、PostMan 的使用、OAuth 流程,JWT 概念及使用 和 API 开发相关的进阶知识。
最佳答案

把下述代码放到 \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

3年前 评论
hengbo_liu (楼主) 3年前
讨论数量: 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

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

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

8个月前 评论

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