2021-03-02 10:53:06 +00:00
|
|
|
#include <DataTypes/DataTypesNumber.h>
|
|
|
|
#include <Functions/FunctionFactory.h>
|
2021-05-15 17:33:15 +00:00
|
|
|
#include <Functions/IFunctionOld.h>
|
2021-03-02 10:53:06 +00:00
|
|
|
#include <Interpreters/Context.h>
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
2021-03-09 12:09:35 +00:00
|
|
|
/// Get the connection Id. It's used for MySQL handler only.
|
2021-04-10 23:33:54 +00:00
|
|
|
class FunctionConnectionId : public IFunction, WithContext
|
2021-03-02 15:17:02 +00:00
|
|
|
{
|
|
|
|
public:
|
2021-03-09 12:09:35 +00:00
|
|
|
static constexpr auto name = "connectionId";
|
2021-03-02 10:53:06 +00:00
|
|
|
|
2021-04-10 23:33:54 +00:00
|
|
|
explicit FunctionConnectionId(ContextPtr context_) : WithContext(context_) {}
|
2021-03-02 10:53:06 +00:00
|
|
|
|
2021-04-10 23:33:54 +00:00
|
|
|
static FunctionPtr create(ContextPtr context_) { return std::make_shared<FunctionConnectionId>(context_); }
|
2021-03-02 10:53:06 +00:00
|
|
|
|
2021-03-02 15:17:02 +00:00
|
|
|
String getName() const override { return name; }
|
2021-03-02 10:53:06 +00:00
|
|
|
|
2021-03-02 15:17:02 +00:00
|
|
|
size_t getNumberOfArguments() const override { return 0; }
|
2021-03-02 10:53:06 +00:00
|
|
|
|
2021-03-02 15:17:02 +00:00
|
|
|
DataTypePtr getReturnTypeImpl(const DataTypes & /*arguments*/) const override { return std::make_shared<DataTypeUInt64>(); }
|
2021-03-02 10:53:06 +00:00
|
|
|
|
2021-03-02 15:17:02 +00:00
|
|
|
ColumnPtr executeImpl(const ColumnsWithTypeAndName &, const DataTypePtr & result_type, size_t input_rows_count) const override
|
|
|
|
{
|
2021-04-10 23:33:54 +00:00
|
|
|
return result_type->createColumnConst(input_rows_count, getContext()->getClientInfo().connection_id);
|
2021-03-02 15:17:02 +00:00
|
|
|
}
|
|
|
|
};
|
2021-03-02 10:53:06 +00:00
|
|
|
|
2021-03-09 12:09:35 +00:00
|
|
|
void registerFunctionConnectionId(FunctionFactory & factory)
|
2021-03-02 10:53:06 +00:00
|
|
|
{
|
2021-03-29 21:43:00 +00:00
|
|
|
factory.registerFunction<FunctionConnectionId>(FunctionFactory::CaseInsensitive);
|
2021-03-25 23:38:49 +00:00
|
|
|
factory.registerAlias("connection_id", "connectionID", FunctionFactory::CaseInsensitive);
|
2021-03-02 10:53:06 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|