Collection.h

Go to the documentation of this file.

/*
 * This file is part of KnowRob, please consult
 * https://github.com/knowrob/knowrob for license details.
 */
 
#ifndef KNOWROB_MONGO_COLLECTION_H
#define KNOWROB_MONGO_COLLECTION_H
 
#include <mongoc.h>
#include <string>
#include <vector>
#include "BulkOperation.h"
#include "Document.h"
#include "Connection.h"
#include "Index.h"
 
namespace knowrob::mongo {
    class Collection {
    public:
        Collection(
                const std::shared_ptr<Connection> &connection,
                std::string_view databaseName,
                std::string_view collectionName);
 
        Collection(const Collection &collection);
 
        ~Collection();
 
        void appendSession(bson_t *opts);
 
        mongoc_client_session_t *session();
 
        auto pool() { return connection_->pool_; }
 
        auto connection() { return connection_; }
 
        auto client() const { return client_; }
 
        auto coll() { return coll_; }
 
        const auto &name() const { return name_; }
 
        const auto &dbName() const { return dbName_; }
 
        const auto &dbURI() const { return connection_->uri_string_; }
 
        void drop();
 
        std::vector<std::string> distinctValues(std::string_view key);
 
        void removeOne(const Document &document);
 
        void removeOne(const bson_oid_t &oid);
 
        void removeAll(const Document &document);
 
        void storeOne(const Document &document);
 
        void update(const Document &query, const Document &update, bool upsert = false);
 
        void evalAggregation(const bson_t *pipeline);
 
        std::shared_ptr<BulkOperation> createBulkOperation();
 
        void createAscendingIndex(const std::vector<const char *> &keys);
 
        void createIndex(const std::vector<IndexKey> &keys);
 
        void createTripleIndex();
 
        bool empty();
 
    private:
        std::shared_ptr<Connection> connection_;
        mongoc_client_t *client_;
        mongoc_client_session_t *session_;
        mongoc_collection_t *coll_;
        mongoc_database_t *db_;
        const std::string name_;
        const std::string dbName_;
 
        void remove(const Document &document, mongoc_remove_flags_t flag);
 
        void createIndex_internal(const bson_t &keys);
    };
}
 
#endif //KNOWROB_MONGO_COLLECTION_H