aswi2020sebela-gitlab/website/src/OpenDataManager.php @ 91c8fbfb

<?php

namespace App\OpenData;

use MongoDB\Driver\Query;
use MongoDB\Driver\Command;
use MongoDB\Driver\Manager;

class OpenDataManager implements IOpenDataManager {
    private $manager;

    public function __construct() {
        $this->manager = new Manager(
            $_ENV['DATABASE_CONNECTION_STRING']
        );
    }

    public function getCollectionDataByName($name, $date, $hour) {
        $valh = $hour < 10 ? '0'.$hour : $hour;
        $openData = $this->manager->executeQuery('open-data-db.'.$name.$date, new Query(['date' => $date.$valh], []));

        $openData->setTypeMap([
            'array' => 'array',
            'document' => 'array',
            'root' => 'array',
        ]);

        return $openData->toArray();
    }

    public function getAvailableCollections() {
        $openData = $this->manager->executeQuery('open-data-db.DATASETS', new Query([], ['projection' => ['name' => 1, '_id' => 0]]));

        $openData->setTypeMap([
            'array' => 'array',
            'document' => 'array',
            'root' => 'array',
        ]);

        return $openData->toArray();
    }

    public function getAvailableCollectionsByDay($date) {
        $availableInDate = [];
        $available = $this->getAvailableCollections();
        $index = 0;
        foreach ($available as $key => $value) {
            if ($this->isCollectionAvailable($value['name'], $date) && false == array_key_exists($value['name'], $availableInDate)) {
                $availableInDate[$value['name']] = $index++;
            }
        }

        return $availableInDate;
    }

    public function getXthAvailableCollectionByDay($index, $date) {
        $availableInDate = [];
        $available = $this->getAvailableCollections();
        $currentIndex = 0;

        foreach ($available as $key => $value) {
            if ($this->isCollectionAvailable($value['name'], $date) && false == array_key_exists($value['name'], $availableInDate)) {
                if ($currentIndex == $index) {
                    return $value['name'];
                } else {
                    ++$currentIndex;
                }
            }
        }

        return '';
    }

    public function isCollectionAvailable($name, $date) {
        $command = new Command(['listCollections' => 1, 'filter' => ['name' => $name.$date]]);
        $result = $this->manager->executeCommand('open-data-db', $command)->toArray();

        return !empty($result);
    }

    public function getMaxCollectionNumberAtDay($name, $date) {
        $max = $this->manager->executeQuery('open-data-db.'.$name.$date, new Query([], ['sort' => ['number' => -1], 'limit' => 1]));

        $max->setTypeMap([
            'array' => 'array',
            'document' => 'array',
            'root' => 'array',
        ]);

        $result = $max->toArray();

        return empty($result) ? 1 : $result[0]['number'];
    }
}