aswi2020sebela-gitlab/website/src/OpenDataManager.php @ 2dd5d57f

<?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']] = $value['name'];
            }
        }

        return $availableInDate;
    }

    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 getDatesWithAvailableCollection() {
        $command = new Command(['listCollections' => 1]);
        $result = $this->manager->executeCommand('open-data-db', $command)->toArray();

        $dates = [];
        foreach ($result as $value) {
            if (preg_match('/\\d{4}-\\d{2}-\\d{2}/', $value->name)) {
                array_push($dates, substr($value->name, -10));
            }
        }

        return array_values(array_unique($dates));
    }

    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'];
    }
}