mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 02:12:21 +00:00
130 lines
3.9 KiB
C++
130 lines
3.9 KiB
C++
#pragma once
|
|
|
|
|
|
#include <Core/Names.h>
|
|
#include <DataStreams/IBlockStream_fwd.h>
|
|
#include <Interpreters/IExternalLoadable.h>
|
|
#include <Poco/Util/XMLConfiguration.h>
|
|
#include <Common/PODArray.h>
|
|
#include <common/StringRef.h>
|
|
#include "IDictionarySource.h"
|
|
|
|
#include <chrono>
|
|
#include <memory>
|
|
|
|
namespace DB
|
|
{
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int NOT_IMPLEMENTED;
|
|
}
|
|
|
|
struct IDictionaryBase;
|
|
using DictionaryPtr = std::unique_ptr<IDictionaryBase>;
|
|
|
|
struct DictionaryStructure;
|
|
class ColumnString;
|
|
|
|
struct IDictionaryBase : public IExternalLoadable
|
|
{
|
|
using Key = UInt64;
|
|
|
|
virtual const std::string & getDatabase() const = 0;
|
|
virtual const std::string & getName() const = 0;
|
|
virtual const std::string & getFullName() const = 0;
|
|
|
|
const std::string & getLoadableName() const override { return getFullName(); }
|
|
|
|
/// Specifies that no database is used.
|
|
/// Sometimes we cannot simply use an empty string for that because an empty string is
|
|
/// usually replaced with the current database.
|
|
static constexpr char NO_DATABASE_TAG[] = "<no_database>";
|
|
|
|
std::string_view getDatabaseOrNoDatabaseTag() const
|
|
{
|
|
const std::string & database = getDatabase();
|
|
if (!database.empty())
|
|
return database;
|
|
return NO_DATABASE_TAG;
|
|
}
|
|
|
|
virtual std::string getTypeName() const = 0;
|
|
|
|
virtual size_t getBytesAllocated() const = 0;
|
|
|
|
virtual size_t getQueryCount() const = 0;
|
|
|
|
virtual double getHitRate() const = 0;
|
|
|
|
virtual size_t getElementCount() const = 0;
|
|
|
|
virtual double getLoadFactor() const = 0;
|
|
|
|
virtual const IDictionarySource * getSource() const = 0;
|
|
|
|
virtual const DictionaryStructure & getStructure() const = 0;
|
|
|
|
virtual bool isInjective(const std::string & attribute_name) const = 0;
|
|
|
|
virtual BlockInputStreamPtr getBlockInputStream(const Names & column_names, size_t max_block_size) const = 0;
|
|
|
|
bool supportUpdates() const override { return true; }
|
|
|
|
bool isModified() const override
|
|
{
|
|
auto source = getSource();
|
|
return source && source->isModified();
|
|
}
|
|
|
|
virtual std::exception_ptr getLastException() const { return {}; }
|
|
|
|
std::shared_ptr<IDictionaryBase> shared_from_this()
|
|
{
|
|
return std::static_pointer_cast<IDictionaryBase>(IExternalLoadable::shared_from_this());
|
|
}
|
|
|
|
std::shared_ptr<const IDictionaryBase> shared_from_this() const
|
|
{
|
|
return std::static_pointer_cast<const IDictionaryBase>(IExternalLoadable::shared_from_this());
|
|
}
|
|
};
|
|
|
|
|
|
struct IDictionary : IDictionaryBase
|
|
{
|
|
virtual bool hasHierarchy() const = 0;
|
|
|
|
virtual void toParent(const PaddedPODArray<Key> & ids, PaddedPODArray<Key> & out) const = 0;
|
|
|
|
virtual void has(const PaddedPODArray<Key> & ids, PaddedPODArray<UInt8> & out) const = 0;
|
|
|
|
/// Methods for hierarchy.
|
|
|
|
virtual void isInVectorVector(
|
|
const PaddedPODArray<Key> & /*child_ids*/, const PaddedPODArray<Key> & /*ancestor_ids*/, PaddedPODArray<UInt8> & /*out*/) const
|
|
{
|
|
throw Exception("Hierarchy is not supported for " + getName() + " dictionary.", ErrorCodes::NOT_IMPLEMENTED);
|
|
}
|
|
|
|
virtual void
|
|
isInVectorConstant(const PaddedPODArray<Key> & /*child_ids*/, const Key /*ancestor_id*/, PaddedPODArray<UInt8> & /*out*/) const
|
|
{
|
|
throw Exception("Hierarchy is not supported for " + getName() + " dictionary.", ErrorCodes::NOT_IMPLEMENTED);
|
|
}
|
|
|
|
virtual void
|
|
isInConstantVector(const Key /*child_id*/, const PaddedPODArray<Key> & /*ancestor_ids*/, PaddedPODArray<UInt8> & /*out*/) const
|
|
{
|
|
throw Exception("Hierarchy is not supported for " + getName() + " dictionary.", ErrorCodes::NOT_IMPLEMENTED);
|
|
}
|
|
|
|
void isInConstantConstant(const Key child_id, const Key ancestor_id, UInt8 & out) const
|
|
{
|
|
PaddedPODArray<UInt8> out_arr(1);
|
|
isInVectorConstant(PaddedPODArray<Key>(1, child_id), ancestor_id, out_arr);
|
|
out = out_arr[0];
|
|
}
|
|
};
|
|
|
|
}
|