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

add user parsing in HDFS URI (fixes after review, style fixes)

Browse Source
This commit is contained in:
akonyaev 2019-07-09 16:16:04 +03:00
parent 74b2440f66
commit c0dc8fc9d3
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2,7 +2,9 @@
#if USE_HDFS
#include <Common/Exception.h>
#include <string.h>
#include <boost/algorithm/string/split.hpp>
#include <boost/algorithm/string/classification.hpp>
namespace DB
{
namespace ErrorCodes
@ -27,9 +29,13 @@ HDFSBuilderPtr createHDFSBuilder(const Poco::URI & uri)
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) {
if (!userInfo.empty() && userInfo.front() != ':')
{
std::vector<std::string> splitRes;
boost::split(splitRes, userInfo, boost::algorithm::is_any_of(":"), boost::algorithm::token_compress_on);
const char *user = splitRes.front().c_str();
if (user)
{
hdfsBuilderSetUserName(builder.get(), user);
}
}