如何让Yii2 ActiveRecord基于内存数据结构而非数据库运行？

嘿，这个需求其实Yii2本身就能通过定制ActiveQuery的方式优雅解决，完全不用大面积修改你现有的计算逻辑代码——核心就是把ActiveRecord的数据源从数据库切换到你内存中的模型对象数组，同时保留原有查询接口。下面给你一套标准的实现方案：

1. 定制内存版ActiveQuery类 #

首先我们需要写一个继承自ActiveQuery的子类，它不会去数据库查询，而是从你提供的内存模型数组里做筛选、排序等操作，模拟原生ActiveQuery的行为：

use yii\db\ActiveQuery;

class MemoryActiveQuery extends ActiveQuery
{
    private array $memoryData = [];

    // 设置内存中的模型数组
    public function setMemoryData(array $data): self
    {
        $this->memoryData = $data;
        return $this;
    }

    // 暴露内存数据（用于后续save/delete操作修改）
    public function getMemoryData(): array
    {
        return $this->memoryData;
    }

    // 重写all方法，从内存筛选数据
    public function all($db = null): array
    {
        $result = $this->memoryData;

        // 处理WHERE条件（这里先实现简单的相等匹配，复杂条件可以扩展）
        if (!empty($this->where)) {
            $result = array_filter($result, function ($model) {
                foreach ($this->where as $attribute => $value) {
                    if ($model->$attribute != $value) {
                        return false;
                    }
                }
                return true;
            });
        }

        // 处理ORDER BY排序
        if (!empty($this->orderBy)) {
            usort($result, function ($modelA, $modelB) {
                foreach ($this->orderBy as $attribute => $direction) {
                    $cmp = strcmp((string)$modelA->$attribute, (string)$modelB->$attribute);
                    if ($cmp !== 0) {
                        return $direction === SORT_DESC ? -$cmp : $cmp;
                    }
                }
                return 0;
            });
        }

        // 处理LIMIT和OFFSET分页
        if ($this->limit !== null || $this->offset !== null) {
            $offset = $this->offset ?? 0;
            $limit = $this->limit ?? count($result);
            $result = array_slice($result, $offset, $limit);
        }

        return array_values($result);
    }

    // 重写one方法，返回第一条匹配结果
    public function one($db = null): ?object
    {
        $results = $this->all($db);
        return $results[0] ?? null;
    }

    // 重写count方法，返回匹配结果数量
    public function count($q = '*', $db = null): int
    {
        return count($this->all($db));
    }
}

2. 在ActiveRecord模型中添加内存模式切换逻辑 #

接下来在你的模型类里添加静态方法，用来切换到内存模式，让模型的find()方法返回我们上面写的MemoryActiveQuery实例：

use yii\db\ActiveRecord;

class YourModel extends ActiveRecord
{
    private static ?MemoryActiveQuery $memoryQuery = null;

    // 切换到内存模式，传入内存中的模型数组
    public static function useMemoryData(array $models): void
    {
        $query = new MemoryActiveQuery(static::class);
        $query->setMemoryData($models);
        static::$memoryQuery = $query;
    }

    // 重置回数据库模式
    public static function useDatabase(): void
    {
        static::$memoryQuery = null;
    }

    // 重写find方法，优先返回内存查询实例
    public static function find(): ActiveQuery
    {
        if (static::$memoryQuery !== null) {
            return static::$memoryQuery;
        }
        return parent::find();
    }

    // 可选：重写save方法，让写操作只修改内存数据
    public function save($runValidation = true, $attributeNames = null): bool
    {
        if (static::$memoryQuery !== null) {
            $memoryData = static::$memoryQuery->getMemoryData();

            if ($this->getIsNewRecord()) {
                // 新增：添加到内存数组
                $memoryData[] = $this;
                $this->setIsNewRecord(false);
            } else {
                // 更新：替换内存中对应的模型
                foreach ($memoryData as $key => $model) {
                    if ($model->getPrimaryKey() == $this->getPrimaryKey()) {
                        $memoryData[$key] = $this;
                        break;
                    }
                }
            }

            static::$memoryQuery->setMemoryData($memoryData);
            return true;
        }

        return parent::save($runValidation, $attributeNames);
    }

    // 可选：重写delete方法，从内存数组移除模型
    public function delete(): int
    {
        if (static::$memoryQuery !== null) {
            $memoryData = static::$memoryQuery->getMemoryData();
            $primaryKey = $this->getPrimaryKey();

            $memoryData = array_filter($memoryData, function ($model) use ($primaryKey) {
                return $model->getPrimaryKey() != $primaryKey;
            });

            static::$memoryQuery->setMemoryData(array_values($memoryData));
            return 1;
        }

        return parent::delete();
    }
}

3. 使用方式（完全复用原有代码） #

现在你只需要在测试/校验场景开始前，初始化内存数据并切换到内存模式，之后所有原有基于ActiveRecord的查询、写操作都会直接操作内存数据，不会触碰数据库：

// 1. 准备内存中的模型数组（可以从文件读取、构造实例等）
$memoryModels = [
    new YourModel(['id' => 1, 'name' => '测试数据1', 'value' => 100]),
    new YourModel(['id' => 2, 'name' => '测试数据2', 'value' => 200]),
];

// 2. 切换模型到内存模式
YourModel::useMemoryData($memoryModels);

// 3. 原有计算逻辑代码完全不用改！
$targetModels = YourModel::find()
    ->where(['value' => 200])
    ->orderBy('id')
    ->all();

// 甚至写操作也会修改内存数据
$newModel = new YourModel(['name' => '新增数据', 'value' => 300]);
$newModel->save();

// 4. 测试完成后可以切回数据库模式
YourModel::useDatabase();

扩展说明 #

• 如果需要支持更复杂的查询条件（比如like、between、andWhere等），可以扩展MemoryActiveQuery中的WHERE条件处理逻辑，解析Yii的查询条件格式即可。
• 如果有多个模型需要切换内存模式，可以把内存模式的逻辑提取成Trait，避免重复代码。
• 这种方案完全遵循Yii2的ActiveRecord设计，不需要修改原有业务逻辑代码，非常适合测试和完整性校验场景。

内容的提问来源于stack exchange，提问作者Kostas Stamos