eleven26

class A extends \Illuminate\Database\Eloquent\Model
{
    // 可以让我们关联 MongoDB 的一个 trait
    use \Jenssegers\Mongodb\Eloquent\HybridRelations;

    public function b()
    {
        return $this->hasOne(B::class, '_id', 'b_id');
    }
}

class B extends \Jenssegers\Mongodb\Eloquent\Model
{
    
}

$a = A::query()->with('b')->whereKey(10)->first();
// $a 可以查询出来
dump($a);

$a = A::query()->whereHas('b')->whereKey(10)->first();
// $a 为 null
dump($a);

/**
 * Returns key we are constraining this parent model's query with
 * @param $relation
 * @return string
 * @throws \Exception
 */
protected function getRelatedConstraintKey($relation)
{
    if ($relation instanceof HasOneOrMany) {
        // 这里返回了关联模型的主键（id），但是实际上应该返回的是关联里面指定的那个键（b_id）
        return $this->model->getKeyName();
    }

    if ($relation instanceof BelongsTo) {
        return $relation->getForeignKey();
    }

    throw new \Exception(class_basename($relation) . ' Is Not supported for hybrid query constraints!');
}

/**
 * Returns key we are constraining this parent model's query with
 * @param Relation $relation
 * @return string
 * @throws Exception
 */
protected function getRelatedConstraintKey(Relation $relation)
{
    if ($relation instanceof HasOneOrMany) {
        // 返回了正确的关联键
        return $relation->getLocalKeyName();
    }

    if ($relation instanceof BelongsTo) {
        return $relation->getForeignKeyName();
    }

    if ($relation instanceof BelongsToMany && !$this->isAcrossConnections($relation)) {
        return $this->model->getKeyName();
    }

    throw new Exception(class_basename($relation) . ' is not supported for hybrid query constraints.');
}

$patterns = [
    'product.xx_'
];

function format($bytes): string
{
    if ($bytes < 1024) {
        return "{$bytes} bytes";
    }

    if ($bytes < 1024 * 1024) {
        $bytes = round($bytes / 1024, 2);

        return "{$bytes} Kb";
    }

    if ($bytes < 1024 * 1024 * 1024) {
        $bytes = round($bytes / 1024 / 1024, 2);

        return "{$bytes} Mb";
    }

    $bytes = round($bytes / 1024 / 1024 / 1024, 2);

    return "{$bytes} Gb";
}

class CollectionStats
{
    private $stats;

    public function __construct(array $stats)
    {
        $this->stats = $stats;
    }

    public function toArray(): array
    {
        return [
            'collection'     => $this->stats['ns'],
            'size'           => format($this->stats['size']),
            'count'          => $this->stats['count'],
            'avgObjSize'     => format($this->stats['avgObjSize'] ?? 0),
            'storageSize'    => format($this->stats['storageSize']),
            'totalIndexSize' => format($this->stats['totalIndexSize']),

            'raw' => [
                'count'          => $this->stats['count'],
                'size'           => $this->stats['size'],
                'avgObjSize'     => $this->stats['avgObjSize'] ?? 0,
                'storageSize'    => $this->stats['storageSize'],
                'totalIndexSize' => $this->stats['totalIndexSize'],
                'bytes_in_cache' => $this->stats['wiredTiger']['cache']['bytes currently in the cache'],
            ],
        ];
    }
}

foreach ($patterns as $pattern) {
    // $connectionName 连接名
    // $collectionPrefix Collection 前缀
    [$connectionName, $collectionPrefix] = explode('.', $pattern);

    dump(compact('connectionName', 'collectionPrefix'));

    $collections = get_collection_names($connectionName);
    $collections = collect($collections)
        ->filter(function ($collection) use ($collectionPrefix) {
            return Str::startsWith($collection, $collectionPrefix);
        })
        ->values()
        ->toArray();

    // 查看 Collection 的状态：
    // https://docs.mongodb.com/manual/reference/command/
    // https://docs.mongodb.com/manual/reference/command/collStats/#mongodb-dbcommand-dbcmd.collStats

    $result = [];

    foreach ($collections as $collection) {
        $cursor = DB::connection($connectionName)->getMongoDB()->command(['collStats' => $collection]);
        $res     = json_decode(json_encode($cursor->toArray()[0]), true);
        $reply   = json_encode($res, JSON_UNESCAPED_UNICODE | JSON_PRETTY_PRINT);

        $result[] = (new CollectionStats($res))->toArray();
    }

    $results = collect($result);

    dump('collection_count: ' . $results->count());
    dump('total_document_count: ' . $results->sum('raw.count'));
    dump('total_size: ' . format($results->sum('raw.size')));
    dump('avg_obj_size: ' . format($results->avg('raw.avgObjSize')));
    dump('total_storage_size: ' . format($results->sum('raw.storageSize')));
    dump('total_index_size: ' . format($results->sum('raw.totalIndexSize')));
    dump('total_bytes_in_cache: ' . format($results->sum('raw.bytes_in_cache')));
    echo PHP_EOL;
}