thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-21 09:10:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

add user parsing in HDFS URI

Browse Source
This commit is contained in:
akonyaev 2019-07-09 13:26:06 +03:00
parent ef3c06b3d0
commit 74b2440f66
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
dbms/src/IO/HDFSCommon.cpp
View File

@ -2,6 +2,7 @@
#if USE_HDFS
#include <Common/Exception.h>
#include <string.h>
namespace DB
{
namespace ErrorCodes
@ -25,6 +26,13 @@ HDFSBuilderPtr createHDFSBuilder(const Poco::URI & uri)
hdfsBuilderConfSetStr(builder.get(), "input.write.timeout", "60000"); // 1 min
hdfsBuilderConfSetStr(builder.get(), "input.connect.timeout", "60000"); // 1 min
std::string userInfo = uri.getUserInfo();
if (!userInfo.empty() && userInfo.front() != ':') {
char *user = strtok(const_cast<char *> (userInfo.c_str()), ":");
if (user != NULL) {
hdfsBuilderSetUserName(builder.get(), user);
}
}
hdfsBuilderSetNameNode(builder.get(), host.c_str());
hdfsBuilderSetNameNodePort(builder.get(), port);
return builder;