MongoQueryBuilder.hpp

Go to the documentation of this file.

#pragma once
#include <string>
#include <memory>
#include <vector>
#include <orbit/http/json.hpp>
#include <orbit/database/MongoClient.hpp>
 
namespace orm {
 
template <typename ModelType>
struct MongoGetAwaiter {
    database::MongoClient::QueryAwaiter native_awaiter;
 
    bool await_ready() const { return native_awaiter.await_ready(); }
    void await_suspend(std::coroutine_handle<> h) { native_awaiter.await_suspend(h); }
    
    std::vector<ModelType> await_resume() {
        database::MongoClient::QueryResult result = native_awaiter.await_resume();
        std::vector<ModelType> models;
        for (const auto& doc_str : result.documents) {
            try {
                auto j = nlohmann::json::parse(doc_str);
                models.push_back(j.get<ModelType>());
            } catch (...) {
                // Ignore parse errors for now
            }
        }
        return models;
    }
};
 
template <typename ModelType>
struct MongoInsertAwaiter {
    database::MongoClient::InsertAwaiter native_awaiter;
 
    bool await_ready() const { return native_awaiter.await_ready(); }
    void await_suspend(std::coroutine_handle<> h) { native_awaiter.await_suspend(h); }
    
    bool await_resume() {
        return native_awaiter.await_resume();
    }
};
 
template <typename ModelType>
class MongoQueryBuilder {
public:
    MongoQueryBuilder(std::shared_ptr<database::MongoClient> db) 
        : db_(std::move(db)) {}
        
    MongoQueryBuilder& where(const nlohmann::json& filter) {
        // Merge filters
        for (auto& el : filter.items()) {
            filter_[el.key()] = el.value();
        }
        return *this;
    }
    
    MongoGetAwaiter<ModelType> get_async() {
        return MongoGetAwaiter<ModelType>{ db_->find_async(filter_.dump()) };
    }
 
    MongoInsertAwaiter<ModelType> insert_async(const ModelType& model) {
        nlohmann::json j = model;
        return MongoInsertAwaiter<ModelType>{ db_->insert_async(j.dump()) };
    }
 
private:
    std::shared_ptr<database::MongoClient> db_;
    nlohmann::json filter_ = nlohmann::json::object();
};
 
} // namespace orm