aswi2020sebela-gitlab/website/src/OpenDataManager.php @ 3ae59f75

<?php

namespace App\OpenData;

use MongoDB\Driver\Query;
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) {
        $openData = $this->manager->executeQuery('open-data-db.'.$name.$date, new Query(['date' => $date.$hour], []));

        $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) {
            $openData = $this->manager->executeQuery('open-data-db.'.$value['name'].$date, new Query([], []));

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

            if (false == empty($openData->toArray())) {
                $availableInDate[$value['name']] = $index++;
            }
        }

        return $availableInDate;
    }

    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',
        ]);

        return $max->toArray()[0]['number'];
    }
}