mongodb:跨日期范围查询

时间:2017-02-09 23:00:22

标签: c++ mongodb mongo-cxx-driver

使用mongocxx驱动程序,我需要查询mongodb以查找属于某个日期范围的文档(库存数据)。

考虑以下文档格式:

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

假设每个库存有一个集合,每个集合有数百个这样的文档,每个文档都有不同的日期。

如果用户提供两个格式为字符串的日期(yyyy-mm-dd):

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

如何查询mongo以获取日期介于" start_date"之间的所有文件。和" end_date",(包括在内)?

注意:我使用的是mongodb 3.2.12,mongocxx驱动程序版本3.0.2

谢谢,

1 个答案:

答案 0 :(得分:0)

不幸的是,似乎没有办法从具有任意时区的字符串中解析日期;所有日期解析都假定在用户的语言环境中,这意味着您需要提供一个偏移量才能正确查询存储在数据库中的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;
    }
}