thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-26 01:22:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support MySQL 'SELECT DATABASE()' query replacement

Browse Source
This commit is contained in:
BohuTANG 2020-07-09 11:07:20 +08:00
parent b4c2c4906e
commit 5ca3aef919
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/Server/MySQLHandler.cpp
View File

@ -48,6 +48,7 @@ namespace ErrorCodes
static String selectEmptyReplacementQuery(const String & query);
static String showTableStatusReplacementQuery(const String & query);
static String killConnectionIdReplacementQuery(const String & query);
static String selectDatabaseReplacementQuery(const String & query);
MySQLHandler::MySQLHandler(IServer & server_, const Poco::Net::StreamSocket & socket_,
bool ssl_enabled, size_t connection_id_)
@ -65,6 +66,7 @@ MySQLHandler::MySQLHandler(IServer & server_, const Poco::Net::StreamSocket & so
replacements.emplace("KILL QUERY", killConnectionIdReplacementQuery);
replacements.emplace("SHOW TABLE STATUS LIKE", showTableStatusReplacementQuery);
replacements.emplace("SHOW VARIABLES", selectEmptyReplacementQuery);
replacements.emplace("SELECT DATABASE()", selectDatabaseReplacementQuery);
}
void MySQLHandler::run()
@ -437,5 +439,14 @@ static String killConnectionIdReplacementQuery(const String & query)
return query;
}
/// Replace "SELECT DATABASE()" into "SELECT currentDatabase() AS `DATABASE()`".
static String selectDatabaseReplacementQuery(const String & query)
{
const String prefix = "SELECT DATABASE()";
if (query.size() > prefix.size())
return query;
else
return ("SELECT currentDatabase() AS `DATABASE()`");
}
}