使用宏(Macro)来扩展 Laravel 的数据库请求构建器

file

习惯了编写 SQL 语句的开发人员可能对 Laravel 的 ORM / Query Builder 并不是很了解。 当我们在需要用到查询构造器没有实现的方法时,通常会使用 select(DB::raw(...)) 来组装 SQL 表达式。

现在我们可以使用 Laravel 的 "Macros" 以更加灵活的方式来编写这些语句。 我将要执行的操作命名为 "insertOrUpdateMany" ,它将使用单个方法来构建和执行自定义 SQL 语句。

定义宏:

为了保证代码的清晰可读性,我将绑定到宏的查询生成器实例和回调的参数 "$rows" 传递到一个新的宏实例。

use Illuminate\Database\Query\Builder;
use App\Macros\InsertOrUpdateMany;
Builder::macro('insertOrUpdateMany', function(array $rows){
    return with(new InsertOrUpdateMany($this, $rows))->execute();
});

我们可以将这些定义放在 Laravel 的服务提供者里。

使用 Macro

我们的 Macro 功能类似于 Laravel 自带的 insert() 函数,返回结果是数据库影响的行数:

$affectedRowCount = DB::table('users')->insertOrUpdateMany(array(
    array(
        'id' => 1,
        'name' => 'Test 1',
        'email' => 'test1@test.local',
        'password' => 'XXX'
    ),
    array('id' => 2,
        'name' => 'Test 2',
        'email' => 'test2@test.local',
        'password' => 'XXX'
    ),
    array(
        'id' => 3,
        'name' => 'Test 3',
        'email' => 'test3@test.local',
        'password' => 'XXX'
    ),
));
dd($affectedRowCount);

数据库查询类

数据库查询类 InsertOrUpdateMany 里将处理各种的 SQL 拼接和查询处理。类的构建函数里我们将获取到 Builder 和 PDO 连接,剩下的一切就变得很容易了,PDO 的 affectingStatement 直接就可返回数据库影响的行数。很容易吧,下面是完整的类代码:

InsertOrUpdateMany.php

<?php namespace BayAreaWebPro\SimpleCsv\Macros;
use Illuminate\Database\Query\Builder;
class InsertOrUpdateMany
{
    /**
     * @param \Illuminate\Database\Query\Builder $builder
     * @var \PDO $pdo
     * @var array $entries
     * @var array $columnsArray
     * @var string $columnsString
     * @var string $valuesString
     * @var string $updateString
     * @var string $sql
     */
    protected $builder, $pdo, $entries;
    private $columnsArray, $columnsString, $valuesString, $updateString, $sql;
    /**
     * InsertOrUpdateMany constructor.
     * @param \Illuminate\Database\Query\Builder $builder
     * @param array $entries
     * @return void
     */
    public function __construct(Builder $builder, array $entries)
    {
        $this->builder = $builder;
        $this->entries = $entries;
        $this->pdo = $this->builder->getConnection()->getPdo();
        $this->prepareValues();
        $this->assembleStatement();
    }
    /**
     * Prepare the values.
     * @return void
     */
    private function prepareValues()
    {
        //获取 Columns Array
        $this->columnsArray = $this->getColumnsArray();
        //构建 Columns 字串
        $this->columnsString = $this->collapse($this->columnsArray);
        //构建数据
        $this->valuesString = $this->collapse(array_map(function ($row) {
            return $this->encase($this->collapse($this->escape($row)));
        }, $this->entries));
        //构建更新的字串
        $this->updateString = $this->collapse(array_map(function ($value) {
            return "$value = VALUES($value)";
        }, $this->columnsArray));
    }
    /**
     * 组合 SQL 语句
     * @return void
     */
    private function assembleStatement()
    {
        $this->sql = "INSERT INTO {$this->builder->from} ({$this->columnsString}) VALUES {$this->valuesString} ON DUPLICATE KEY UPDATE {$this->updateString}";
    }
    /**
     * 获取表结构数组
     * @return array
     */
    private function getColumnsArray()
    {
        return array_keys(reset($this->entries));
    }
    /**
     * 处理数组
     * @param array $values
     * @return array
     */
    private function escape(array $values)
    {
        return array_map(function ($value) {
            return $value ? $this->pdo->quote($value) : 'NULL';
        }, $values);
    }
    /**
     * 合并数组
     * @param array $values
     * @return string
     */
    private function collapse(array $values)
    {
        return implode(',', $values);
    }
    /**
     * 封装字串
     * @param string $value
     * @return string
     */
    private function encase(string $value)
    {
        return '(' . $value . ')';
    }
    /**
     * 执行语句
     * @return int
     */
    public function execute()
    {
        return $this->builder->getConnection()->affectingStatement($this->sql);
    }
}
本文中的所有译文仅用于学习和交流目的,转载请务必注明文章译者、出处、和本文链接
我们的翻译工作遵照 CC 协议,如果我们的工作有侵犯到您的权益,请及时联系我们。

原文地址:https://medium.com/@danielalvidrez/larav...

译文地址:https://learnku.com/laravel/t/15247/expa...

本帖已被设为精华帖!
本文为协同翻译文章,如您发现瑕疵请点击「改进」按钮提交优化建议
《L03 构架 API 服务器》
你将学到如 RESTFul 设计风格、PostMan 的使用、OAuth 流程,JWT 概念及使用 和 API 开发相关的进阶知识。
《L02 从零构建论坛系统》
以构建论坛项目 LaraBBS 为线索,展开对 Laravel 框架的全面学习。应用程序架构思路贴近 Laravel 框架的设计哲学。
讨论数量: 3
liuqing_hu

Macro 的功能还是很有必要研究和学习的,不知道为什么都没人评论,希望更多人能够阅读这篇优秀的文章!强推,感谢译者!!

5年前 评论

Macro 主要靠 Closure 的 bindTo方法实现的,laravel 作者确实牛逼

5年前 评论

有没有好一点的宏注册方式,使用服务提供者注册会存在每次请求都加载的情况

4年前 评论

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