如何自定义 Laravel Database\Query\Builder（制作更好的子查询）

Question

我正在开发 Laravel 4。 据我所知，我可以做子查询：

Project::whereIn('project_id', function($q) {
    $q->select('project_id')
        ->from('company')
        ->whereNull('deleted_at');
});

我发现了一些复杂情况，即我不能在子查询中使用范围并禁用 soft_delete 使我对源代码进行了如此多的更改。

我希望是：

Project::whereIn('project_id', function(&$q) {
    $q = Company::select('project_id')->getQuery();
});

现在，我可以轻松添加范围，禁用 soft_delete。

我尝试过，找到了一个解决方案，我必须更改 Laravel 的 Database\Query\Builder 代码，函数 whereInSub，第 786 行。

call_user_func($callback, $query = $this->newQuery());

到：

$query = $this->newQuery();
call_user_func_array($callback, array(&$query));

修改 Laravel 框架的供应商是有害的。所以我想问一下如何安全地做到这一点。

对不起，我的英语不好。

感谢您的阅读。

Answer 1

哦！这是一个相当棘手的问题，因为您的模型将扩展 Eloquent，然后 Eloquent 使用 Illuminate\Database\Query\Builder。

但我注意到Eloquent 实际上是app/config/app.php 文件中的别名。因此，您可以按照以下步骤操作。

1. 使用您的自定义whereInSub() 将Illuminate\Database\Query\Builder 扩展到MyQueryBuilder。
2. 将Illuminate\Database\Eloquent\Model 扩展为MyModel，并将use 设为您的MyQueryBuilder。
3. 将app/config/app.php 中的Eloquent 别名设置为新的MyModel 类。

类似这样的：

MyQueryBuilder.php：

use Closure;
use Illuminate\Support\Collection;
use Illuminate\Database\ConnectionInterface;
use Illuminate\Database\Query\Grammars\Grammar;
use Illuminate\Database\Query\Processors\Processor;

class MyQueryBuilder extends Illuminate\Database\Query\Builder
{
    protected function whereInSub($column, Closure $callback, $boolean, $not)
    {
        $type = $not ? 'NotInSub' : 'InSub';

        $query = $this->newQuery(); // Your changes
        call_user_func_array($callback, array(&$query)); // Your changes

        $this->wheres[] = compact('type', 'column', 'query', 'boolean');

        $this->mergeBindings($query);

        return $this;
    }
}

MyModel.php：

use DateTime;
use ArrayAccess;
use Carbon\Carbon;
use LogicException;
use Illuminate\Events\Dispatcher;
use Illuminate\Database\Eloquent\Relations\Pivot;
use Illuminate\Database\Eloquent\Relations\HasOne;
use Illuminate\Database\Eloquent\Relations\HasMany;
use Illuminate\Database\Eloquent\Relations\MorphTo;
use Illuminate\Support\Contracts\JsonableInterface;
use Illuminate\Support\Contracts\ArrayableInterface;
use Illuminate\Database\Eloquent\Relations\Relation;
use Illuminate\Database\Eloquent\Relations\MorphOne;
use Illuminate\Database\Eloquent\Relations\MorphMany;
use Illuminate\Database\Eloquent\Relations\BelongsTo;
// use Illuminate\Database\Query\Builder as QueryBuilder;
use Illuminate\Database\Eloquent\Relations\MorphToMany;
use Illuminate\Database\Eloquent\Relations\BelongsToMany;
use Illuminate\Database\Eloquent\Relations\HasManyThrough;
use Illuminate\Database\ConnectionResolverInterface as Resolver;
use MyQueryBuilder as QueryBuilder; // MyModel should now use your MyQueryBuilder instead of the default which I commented out above

abstract class MyModel extends Illuminate\Database\Eloquent\Model
{

}

app/config/app.php：

'aliases' => array(
    ...
    'Eloquent'        => 'MyModel',
    ...
);

请注意，我在上面放了很长的use 列表，因为"use" keyword does not get inherited。另外，为了简单起见，我没有将 MyQueryBuilder 和 MyModel 放在命名空间中。我的use 列表也可能与你的不同，这取决于我们使用的 Laravel 版本，所以请也检查一下用途。

Answer 2

您可以创建一个自定义查询构建器并像这样使用它。

我的自定义查询生成器：

use Illuminate\Database\Query\Builder as QueryBuilder;

class MyCustomBuilder extends QueryBuilder{

protected function whereInSub($column, Closure $callback, $boolean, $not)
{
       $type = $not ? 'NotInSub' : 'InSub';

       $query = $this->newQuery(); // Your changes
       call_user_func_array($callback, array(&$query)); // Your changes

       $this->wheres[] = compact('type', 'column', 'query', 'boolean');

       $this->mergeBindings($query);

       return $this;
    }
}

在任何模型中覆盖方法 newBaseQueryBuilder() 并返回您自己的查询构建器的实例

class MyModel extends Model{
   protected function newBaseQueryBuilder()
   {
      $connection = $this->getConnection();

      return new MyQueryBuilder(
           $connection, $connection->getQueryGrammar(), 
              $connection->getPostProcessor()
    );
}
}