Balloon\Migration\Delta\CreateUniqueNodeIndex

Complexity

Total Complexity

3

Size/Duplication

Total Lines	64
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	0

Test Coverage

Coverage

0%

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	start()	0	43	2

<?php

declare(strict_types=1);

/**
 * balloon
 *
 * @copyright   Copryright (c) 2012-2018 gyselroth GmbH (https://gyselroth.com)
 * @license     GPL-3.0 https://opensource.org/licenses/GPL-3.0
 */

namespace Balloon\Migration\Delta;

use MongoDB\Database;

class CreateUniqueNodeIndex implements DeltaInterface
{
    /**
     * Database.
     *
     * @var Database
     */
    protected $db;

    /**
     * Construct.
     */
    public function __construct(Database $db)
    {
        $this->db = $db;
    }

    /**
     * Start.
     */
    public function start(): bool
    {
        $pipeline = [
            ['$group' => [
                '_id' => [
                    'name' => '$name',
                    'parent' => '$parent',
                    'owner' => '$owner',
                    'deleted' => '$deleted',
                ],
                'count' => [
                    '$sum' => 1,
                ],
                'nodes' => [
                    '$addToSet' => '$_id',
                ],
            ]],
            ['$match' => [
                '_id' => ['$ne' => null],
                'count' => ['$gt' => 1],
            ]],
        ];

        $cursor = $this->db->storage->aggregate($pipeline, [
            'allowDiskUse' => true,
        ]);

        foreach ($cursor as $object) {
            $id = array_pop($object['nodes']);
            $this->db->storage->deleteOne(['_id' => $id]);
        }

        $this->db->storage->createIndex([
            'name' => 1,
            'owner' => 1,
            'parent' => 1,
            'deleted' => 1,
        ], [
            'unique' => true,
        ]);

        return true;
    }
}