ClickHouse/src/Storages/StorageMongoDB.h

59 lines
1.6 KiB
C++
Raw Normal View History

2020-05-13 23:20:45 +00:00
#pragma once
#include <Poco/MongoDB/Connection.h>
2020-05-13 23:20:45 +00:00
#include <Storages/IStorage.h>
2021-09-03 11:16:32 +00:00
#include <Storages/ExternalDataSourceConfiguration.h>
2020-06-26 14:28:00 +00:00
2020-05-13 23:20:45 +00:00
namespace DB
{
/* Implements storage in the MongoDB database.
* Use ENGINE = mysql(host_port, database_name, table_name, user_name, password)
* Read only.
*/
class StorageMongoDB final : public IStorage
2020-05-13 23:20:45 +00:00
{
public:
StorageMongoDB(
const StorageID & table_id_,
const std::string & host_,
uint16_t port_,
2020-05-13 23:20:45 +00:00
const std::string & database_name_,
const std::string & collection_name_,
const std::string & username_,
const std::string & password_,
2021-03-23 15:01:13 +00:00
const std::string & options_,
2020-05-13 23:20:45 +00:00
const ColumnsDescription & columns_,
2021-04-23 12:18:23 +00:00
const ConstraintsDescription & constraints_,
const String & comment);
2020-05-13 23:20:45 +00:00
std::string getName() const override { return "MongoDB"; }
2020-08-03 13:54:14 +00:00
Pipe read(
2020-05-13 23:20:45 +00:00
const Names & column_names,
const StorageSnapshotPtr & storage_snapshot,
SelectQueryInfo & query_info,
ContextPtr context,
2020-05-13 23:20:45 +00:00
QueryProcessingStage::Enum processed_stage,
size_t max_block_size,
unsigned num_streams) override;
2021-09-03 11:16:32 +00:00
static StorageMongoDBConfiguration getConfiguration(ASTs engine_args, ContextPtr context);
2020-05-13 23:20:45 +00:00
private:
void connectIfNotConnected();
const std::string database_name;
const std::string collection_name;
const std::string username;
const std::string password;
2021-03-23 15:01:13 +00:00
const std::string uri;
2020-05-13 23:20:45 +00:00
std::shared_ptr<Poco::MongoDB::Connection> connection;
2021-06-28 17:02:22 +00:00
bool authenticated = false;
std::mutex connection_mutex; /// Protects the variables `connection` and `authenticated`.
2020-05-13 23:20:45 +00:00
};
}