mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-13 01:41:59 +00:00
70d1adfe4b
* save format string for NetException * format exceptions * format exceptions 2 * format exceptions 3 * format exceptions 4 * format exceptions 5 * format exceptions 6 * fix * format exceptions 7 * format exceptions 8 * Update MergeTreeIndexGin.cpp * Update AggregateFunctionMap.cpp * Update AggregateFunctionMap.cpp * fix
54 lines
1.3 KiB
C++
54 lines
1.3 KiB
C++
#include "StorageMongoDBSocketFactory.h"
|
|
|
|
#include <Common/Exception.h>
|
|
|
|
#include "config.h"
|
|
|
|
#include <Poco/Net/IPAddress.h>
|
|
#include <Poco/Net/SocketAddress.h>
|
|
|
|
#if USE_SSL
|
|
# include <Poco/Net/SecureStreamSocket.h>
|
|
#endif
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int FEATURE_IS_NOT_ENABLED_AT_BUILD_TIME;
|
|
}
|
|
|
|
Poco::Net::StreamSocket StorageMongoDBSocketFactory::createSocket(const std::string & host, int port, Poco::Timespan connectTimeout, bool secure)
|
|
{
|
|
return secure ? createSecureSocket(host, port, connectTimeout) : createPlainSocket(host, port, connectTimeout);
|
|
}
|
|
|
|
Poco::Net::StreamSocket StorageMongoDBSocketFactory::createPlainSocket(const std::string & host, int port, Poco::Timespan connectTimeout)
|
|
{
|
|
Poco::Net::SocketAddress address(host, port);
|
|
Poco::Net::StreamSocket socket;
|
|
|
|
socket.connect(address, connectTimeout);
|
|
|
|
return socket;
|
|
}
|
|
|
|
|
|
Poco::Net::StreamSocket StorageMongoDBSocketFactory::createSecureSocket(const std::string & host [[maybe_unused]], int port [[maybe_unused]], Poco::Timespan connectTimeout [[maybe_unused]])
|
|
{
|
|
#if USE_SSL
|
|
Poco::Net::SocketAddress address(host, port);
|
|
Poco::Net::SecureStreamSocket socket;
|
|
|
|
socket.connect(address, connectTimeout);
|
|
|
|
return socket;
|
|
#else
|
|
throw Exception(ErrorCodes::FEATURE_IS_NOT_ENABLED_AT_BUILD_TIME, "SSL is not enabled at build time.");
|
|
#endif
|
|
}
|
|
|
|
}
|