mongodb: запрос в диапазоне дат

Question

Используя драйвер mongocxx, мне нужно запросить у mongodb документы (данные о запасах), которые попадают в определенный диапазон дат.

Рассмотрим следующий формат документа:

{
  date : ISODate("2010-01-01T00:00:00Z"),
  open : 12.00,
  high : 13.00,
  low : 11.00,
  close : 12.50,
  volume : 100000
}

Скажем, у меня есть одна коллекция на одну акцию и сотни этих документов на каждую коллекцию, каждая с разной датой.

Если пользователь указывает две даты, отформатированные в виде строк (гггг-мм-дд):

std::string start_date = "2010-01-01";
std::string end_date = "2010-02-05";

Как я могу запросить монго, чтобы получить все файлы с датами между "start_date" и "end_date", (включительно)?

Примечание: я использую mongodb 3.2.12, версия драйвера mongocxx 3.0.2

Спасибо,

Answer 1

К сожалению, кажется, что нет способа проанализировать даты из строк с произвольными часовыми поясами; предполагается, что все синтаксические разборы находятся в локали пользователя, что означает, что вам нужно будет указать смещение, чтобы можно было правильно запрашивать даты UTC, хранящиеся в базе данных. В идеале они могут быть сгенерированы, когда пользователь предоставляет строку, но это, очевидно, будет зависеть от характера вашего приложения.

Как только у вас есть смещение и строка даты, std::get_time дадут вам большую часть пути туда. После этого вам просто нужно конвертировать std::tm к типу, который вы можете построить bsoncxx::types::b_date от, а затем запросите, как обычно. Вот пример кода, который делает эту работу:

#include <chrono>
#include <cstdint>
#include <ctime>
#include <iomanip>
#include <iostream>
#include <ostream>
#include <sstream>
#include <string>

#include <bsoncxx/builder/basic/document.hpp>
#include <bsoncxx/builder/basic/kvp.hpp>
#include <bsoncxx/builder/basic/sub_document.hpp>
#include <bsoncxx/json.hpp>
#include <bsoncxx/types.hpp>
#include <mongocxx/client.hpp>
#include <mongocxx/uri.hpp>

bsoncxx::types::b_date read_date(const std::string& date,
                                 std::int32_t offset_from_utc) {
    std::tm utc_tm{};
    std::istringstream ss{date};

    // Read time into std::tm.
    ss >> std::get_time(&utc_tm, "%Y-%m-%d");

    // Convert std::tm to std::time_t.
    std::time_t utc_time = std::mktime(&utc_tm);

    // Convert std::time_t std::chrono::systemclock::time_point.
    std::chrono::system_clock::time_point time_point =
        std::chrono::system_clock::from_time_t(utc_time);

    return bsoncxx::types::b_date{time_point +
                                  std::chrono::hours{offset_from_utc}};
}

int main() {
    // User inputs

    std::string start_date = "2010-01-01";
    std::string end_date = "2010-02-05";

    std::int32_t offset_from_utc = -5;

    // Code to execute query

    mongocxx::client client{mongocxx::uri{}};
    mongocxx::collection coll = client["db_name"]["coll_name"];

    bsoncxx::builder::basic::document filter;
    filter.append(bsoncxx::builder::basic::kvp(
        "date", [start_date, end_date,
                 offset_from_utc](bsoncxx::builder::basic::sub_document sd) {
            sd.append(bsoncxx::builder::basic::kvp(
                "$gte", read_date(start_date, offset_from_utc)));
            sd.append(bsoncxx::builder::basic::kvp(
                "$lte", read_date(end_date, offset_from_utc)));
        }));

    for (auto&& result : coll.find(filter.view())) {
        std::cout << bsoncxx::to_json(result) << std::endl;
    }
}