view-components / eloquent-data-processing

src/EloquentProcessingService.php

<?php

namespace ViewComponents\Eloquent;

use ArrayIterator;
use Illuminate\Database\Eloquent\Builder as EloquentBuilder;
use Illuminate\Database\Query\Builder;
use RuntimeException;
use Traversable;
use ViewComponents\ViewComponents\Data\ProcessingService\AbstractProcessingService;

/**
 * Class DbTableProcessingManager.
 */
class EloquentProcessingService extends AbstractProcessingService
{
    /**
     * @param Builder|EloquentBuilder $data
     * @return Traversable
     */
    protected function afterOperations($data)
    {
        if ($data instanceof EloquentBuilder) {

The class Illuminate\Database\Eloquent\Builder does not exist. Did you forget a USE statement, or did you not list all dependencies?

            return $data->get();
        } elseif ($data instanceof Builder) {

The class Illuminate\Database\Query\Builder does not exist. Did you forget a USE statement, or did you not list all dependencies?

            return new ArrayIterator($data->get());
        }
        throw new RuntimeException('Unsupported type of data source.');
    }

    /**
     * @param Builder|EloquentBuilder $data
     * @return Builder|EloquentBuilder
     */
    protected function beforeOperations($data)
    {
        return clone $data;
    }

    /**
     * @return int
     */
    public function count()
    {
        return $this->applyOperations(
            $this->beforeOperations($this->dataSource)
        )->count();
    }
}