mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-17 11:52:27 +00:00
45 lines
1022 B
C++
45 lines
1022 B
C++
#pragma once
|
|
|
|
#include <Dictionaries/CassandraHelpers.h>
|
|
|
|
#if USE_CASSANDRA
|
|
#include <Core/Block.h>
|
|
#include <Processors/ISource.h>
|
|
#include <Core/ExternalResultDescription.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class CassandraSource final : public ISource
|
|
{
|
|
public:
|
|
CassandraSource(
|
|
const CassSessionShared & session_,
|
|
const String & query_str,
|
|
const Block & sample_block,
|
|
size_t max_block_size);
|
|
|
|
String getName() const override { return "Cassandra"; }
|
|
|
|
private:
|
|
using ValueType = ExternalResultDescription::ValueType;
|
|
|
|
Chunk generate() override;
|
|
static void insertValue(IColumn & column, ValueType type, const CassValue * cass_value);
|
|
void assertTypes(const CassResultPtr & result);
|
|
|
|
CassSessionShared session;
|
|
CassStatementPtr statement;
|
|
CassFuturePtr result_future;
|
|
const size_t max_block_size;
|
|
ExternalResultDescription description;
|
|
cass_bool_t has_more_pages;
|
|
bool assert_types = true;
|
|
bool is_initialized = false;
|
|
};
|
|
|
|
}
|
|
|
|
#endif
|