HasManyFromStr.php 4.3 KB
<?php
/**
+-----------------------------------------------------------------------------------------------------------------------
 * 模型层基类:业务模型的基类,提供对数据库基本的增删改查类操作,不包含业务逻辑
+-----------------------------------------------------------------------------------------------------------------------
 *
 * PHP version 7
 *
 * @category  App\Restructure\Relations
 * @package   App\Restructure\Relations
 * @author    Richer <yangzi1028@163.com>
 * @date      2022年7月6日16:17:44
 * @copyright 2022-2022 Richer (http://www.Richer.com/)
 * @license   http://www.Richer.com/ License
 * @link      http://www.Richer.com/
 */
namespace App\Restructure\Relations;

use Illuminate\Database\Eloquent\Relations\HasMany;
use Illuminate\Database\Eloquent\Builder;
use Illuminate\Database\Eloquent\Model;
use Illuminate\Database\Eloquent\Collection;

/**
 * Class HasManyFromStr
 *
 * @category  App\Restructure\Relations
 * @package   App\Restructure\Relations
 * @author    Richer <yangzi1028@163.com>
 * @date      2022年7月6日16:17:44
 * @copyright 2022-2022 Richer (http://www.Richer.com/)
 * @license   http://www.Richer.com/ License
 * @link      http://www.Richer.com/
 */
class HasManyFromStr extends HasMany
{
    protected $separator = ','; //分隔符

    protected $strict;

    public function __construct(Builder $query, Model $parent, string $foreignKey, string $localKey, $separator, $strict = false)
    {
        $this->separator = $separator;
        $this->strict = $strict; //元素执行函数 true 代表转为int
        parent::__construct($query, $parent, $foreignKey, $localKey);
    }

    /**
     * 重写获取单个模型建方法
     */
    public function addConstraints()
    {
        if (static::$constraints) {
            $this->query->whereIn($this->foreignKey, $this->handleIn($this->separator, $this->getParentKey()));

            $this->query->whereNotNull($this->foreignKey);
        }
    }

    /**
     * 重写获取所有主键方法.
     *
     * @param  array   $models
     * @param  string  $key
     * @return array
     */
    protected function getKeys(array $models, $key = null)
    {
        $keysArr = [];
        collect($models)->map(function ($value) use ($key, &$keysArr) {
            $result = $key ? $value->getAttribute($key) : $value->getKey();
            $keysArr = array_merge($keysArr, $this->handleIn($this->separator, $result));
        });
        return collect($keysArr)->values()->unique()->sort()->all();
    }

    /**
     * 重写匹配方法
     * @param array      $models
     * @param Collection $results
     * @param string     $relation
     * @return array
     */
    public function match(array $models, Collection $results, $relation)
    {
        $dictionary = $this->buildDictionary($results);

        // Once we have the dictionary we can simply spin through the parent models to
        // link them up with their children using the keyed dictionary to make the
        // matching very convenient and easy work. Then we'll just return them.
        foreach ($models as $model) {
            $keys = $model->getAttribute($this->localKey);
            $keys = $this->handleIn($this->separator, $keys);
            $keys = array_unique(array_filter($keys));
            $type = 'one';
            $relationResults = [];
            foreach ($keys as $key) {
                if (isset($dictionary[$key])) {
                    $temp = $this->getRelationValue($dictionary, $key, $type);
                    $relationResults[] = $temp;
                }
            }
            $model->setRelation(
                $relation,
                collect($relationResults)
            );
        }

        return $models;
    }


    /**
     * 自定义转换方法
     * @param $separator
     * @param $keyString
     * @return array
     */
    private function handleIn($separator, $keyString)
    {
        $keys = is_array($keyString)?$keyString : explode($separator, $keyString);
        $keys = array_unique($keys);
        if ($this->strict === false) {
            return $keys;
        }
        array_walk($keys, function (&$value) {
            $fun = $this->strict === true ? 'intval' : $this->strict;

            $value = $fun($value);
        });
        return $keys;
    }
}