mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 02:41:59 +00:00
495c6e03aa
* Replace all Context references with std::weak_ptr * Fix shared context captured by value * Fix build * Fix Context with named sessions * Fix copy context * Fix gcc build * Merge with master and fix build * Fix gcc-9 build
39 lines
1.3 KiB
C++
39 lines
1.3 KiB
C++
#include <DataTypes/DataTypesNumber.h>
|
|
#include <Functions/FunctionFactory.h>
|
|
#include <Functions/IFunctionImpl.h>
|
|
#include <Interpreters/Context.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/// Get the connection Id. It's used for MySQL handler only.
|
|
class FunctionConnectionId : public IFunction, WithContext
|
|
{
|
|
public:
|
|
static constexpr auto name = "connectionId";
|
|
|
|
explicit FunctionConnectionId(ContextPtr context_) : WithContext(context_) {}
|
|
|
|
static FunctionPtr create(ContextPtr context_) { return std::make_shared<FunctionConnectionId>(context_); }
|
|
|
|
String getName() const override { return name; }
|
|
|
|
size_t getNumberOfArguments() const override { return 0; }
|
|
|
|
DataTypePtr getReturnTypeImpl(const DataTypes & /*arguments*/) const override { return std::make_shared<DataTypeUInt64>(); }
|
|
|
|
ColumnPtr executeImpl(const ColumnsWithTypeAndName &, const DataTypePtr & result_type, size_t input_rows_count) const override
|
|
{
|
|
return result_type->createColumnConst(input_rows_count, getContext()->getClientInfo().connection_id);
|
|
}
|
|
};
|
|
|
|
void registerFunctionConnectionId(FunctionFactory & factory)
|
|
{
|
|
factory.registerFunction<FunctionConnectionId>(FunctionFactory::CaseInsensitive);
|
|
factory.registerAlias("connection_id", "connectionID", FunctionFactory::CaseInsensitive);
|
|
}
|
|
|
|
}
|