/// <summary>
/// Creates a new instance. This class is also used to shareable a context across all instance that are created below one Cluster instance.
/// One configuration instance per Cluster instance.
/// </summary>
internal Configuration(Policies policies,
ProtocolOptions protocolOptions,
PoolingOptions poolingOptions,
SocketOptions socketOptions,
ClientOptions clientOptions,
IAuthProvider authProvider,
IAuthInfoProvider authInfoProvider,
QueryOptions queryOptions,
IAddressTranslator addressTranslator)
{
if (addressTranslator == null)
{
throw new ArgumentNullException("addressTranslator");
}
if (queryOptions == null)
{
throw new ArgumentNullException("queryOptions");
}
_policies = policies;
_protocolOptions = protocolOptions;
_poolingOptions = poolingOptions;
_socketOptions = socketOptions;
_clientOptions = clientOptions;
_authProvider = authProvider;
_authInfoProvider = authInfoProvider;
_queryOptions = queryOptions;
_addressTranslator = addressTranslator;
}
internal ControlConnection(Cluster cluster,
IEnumerable<IPAddress> clusterEndpoints,
Policies policies,
ProtocolOptions protocolOptions,
PoolingOptions poolingOptions,
SocketOptions socketOptions,
ClientOptions clientOptions,
IAuthProvider authProvider,
IAuthInfoProvider authInfoProvider)
{
_cluster = cluster;
_reconnectionSchedule = _reconnectionPolicy.NewSchedule();
_reconnectionTimer = new Timer(ReconnectionClb, null, Timeout.Infinite, Timeout.Infinite);
var config = new Configuration
(
policies,
protocolOptions,
poolingOptions,
socketOptions,
clientOptions,
authProvider,
authInfoProvider,
new QueryOptions()
);
_session = new Session(cluster, config, "", ControlConnectionProtocolVersion);
}
internal CassandraConnection(Session owner, IPAddress serverAddress, ProtocolOptions protocolOptions,
SocketOptions socketOptions, ClientOptions clientOptions,
IAuthInfoProvider authInfoProvider)
{
this._owner = owner;
_bufferingMode = null;
switch (protocolOptions.Compression)
{
case CompressionType.Snappy:
_bufferingMode = new FrameBuffering();
break;
case CompressionType.NoCompression:
_bufferingMode = clientOptions.WithoutRowSetBuffering ? new NoBuffering() : new FrameBuffering();
break;
default:
throw new ArgumentException();
}
this._authInfoProvider = authInfoProvider;
if (protocolOptions.Compression == CompressionType.Snappy)
{
_startupOptions.Add("COMPRESSION", "snappy");
_compressor = new SnappyProtoBufCompressor();
}
this._serverAddress = serverAddress;
this._port = protocolOptions.Port;
this._queryAbortTimeout = clientOptions.QueryAbortTimeout;
this._asyncCallAbortTimeout = clientOptions.AsyncCallAbortTimeout;
this._socketOptions = socketOptions;
CreateConnection();
if (IsHealthy)
BeginReading();
}
internal ControlConnection(Cluster cluster,
IEnumerable<IPAddress> clusterEndpoints,
Policies policies,
ProtocolOptions protocolOptions,
PoolingOptions poolingOptions,
SocketOptions socketOptions,
ClientOptions clientOptions,
IAuthInfoProvider authProvider)
{
this._cluster = cluster;
this._reconnectionSchedule = _reconnectionPolicy.NewSchedule();
this._reconnectionTimer = new Timer(ReconnectionClb, null, Timeout.Infinite, Timeout.Infinite);
_session = new Session(cluster, policies, protocolOptions, poolingOptions, socketOptions,
clientOptions, authProvider, "", false);
}
internal ControlConnection(Cluster cluster,
IEnumerable<IPAddress> clusterEndpoints,
Policies policies,
ProtocolOptions protocolOptions,
PoolingOptions poolingOptions,
SocketOptions socketOptions,
ClientOptions clientOptions,
IAuthInfoProvider authProvider,
bool metricsEnabled)
{
this._cluster = cluster;
this._reconnectionTimer = new Timer(ReconnectionClb, null, Timeout.Infinite, Timeout.Infinite);
_session = new Session(cluster, clusterEndpoints, policies, protocolOptions, poolingOptions, socketOptions,
clientOptions, authProvider, metricsEnabled, "", _cluster._hosts);
}
internal CassandraConnection(Session owner, IPAddress serverAddress, ProtocolOptions protocolOptions,
SocketOptions socketOptions, ClientOptions clientOptions,
IAuthInfoProvider authInfoProvider)
{
this.Guid = Guid.NewGuid();
this._owner = owner;
_bufferingMode = null;
switch (protocolOptions.Compression)
{
case CompressionType.Snappy:
_bufferingMode = new FrameBuffering();
break;
case CompressionType.NoCompression:
_bufferingMode = clientOptions.WithoutRowSetBuffering ? new NoBuffering() : new FrameBuffering();
break;
default:
throw new ArgumentException();
}
this._authInfoProvider = authInfoProvider;
if (protocolOptions.Compression == CompressionType.Snappy)
{
_startupOptions.Add("COMPRESSION", "snappy");
_compressor = new SnappyProtoBufCompressor();
}
this._serverAddress = serverAddress;
this._port = protocolOptions.Port;
this._queryAbortTimeout = clientOptions.QueryAbortTimeout;
this._socketOptions = socketOptions;
for (int i = 0; i <= sbyte.MaxValue; i++)
_freeStreamIDs.Push(i);
_protocolErrorHandlerAction = new Action<ErrorActionParam>((param) =>
{
if (param.AbstractResponse is ErrorResponse)
JobFinished(
param.Jar,
(param.AbstractResponse as ErrorResponse).Output);
});
_frameEventCallback.Value = new Action<ResponseFrame>(EventOccured);
_buffer = new byte[][] {
new byte[_bufferingMode.PreferedBufferSize()],
new byte[_bufferingMode.PreferedBufferSize()] };
var newSock = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
if (_socketOptions.KeepAlive != null)
newSock.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.KeepAlive,
_socketOptions.KeepAlive.Value);
newSock.SendTimeout = _socketOptions.ConnectTimeoutMillis;
if (_socketOptions.SoLinger != null)
newSock.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.Linger,
new LingerOption(true, _socketOptions.SoLinger.Value));
if (_socketOptions.ReceiveBufferSize != null)
newSock.ReceiveBufferSize = _socketOptions.ReceiveBufferSize.Value;
if (_socketOptions.SendBufferSize != null)
newSock.ReceiveBufferSize = _socketOptions.SendBufferSize.Value;
if (_socketOptions.TcpNoDelay != null)
newSock.NoDelay = _socketOptions.TcpNoDelay.Value;
newSock.Connect(new IPEndPoint(_serverAddress, _port));
_socket = newSock;
_bufferingMode.Reset();
if (protocolOptions.SslOptions == null)
_socketStream = new NetworkStream(_socket);
else
{
string targetHost;
try
{
targetHost = Dns.GetHostEntry(_serverAddress).HostName;
}
catch (SocketException ex)
{
targetHost = serverAddress.ToString();
_logger.Error(string.Format("SSL connection: Can not resolve {0} address. Using IP address instead of hostname. This may cause RemoteCertificateNameMismatch error during Cassandra host authentication. Note that Cassandra node SSL certificate's CN(Common Name) must match the Cassandra node hostname.", _serverAddress.ToString()), ex);
}
_socketStream = new SslStream(new NetworkStream(_socket), false, new RemoteCertificateValidationCallback(protocolOptions.SslOptions.RemoteCertValidationCallback), null);
(_socketStream as SslStream).AuthenticateAsClient(targetHost, new X509CertificateCollection(), protocolOptions.SslOptions.SslProtocol, false);
}
if (IsHealthy)
BeginReading();
}
internal Session(Cluster cluster,
Policies policies,
ProtocolOptions protocolOptions,
PoolingOptions poolingOptions,
SocketOptions socketOptions,
ClientOptions clientOptions,
IAuthInfoProvider authProvider,
string keyspace, bool init)
{
try
{
this._cluster = cluster;
this._protocolOptions = protocolOptions;
this._poolingOptions = poolingOptions;
this._socketOptions = socketOptions;
this._clientOptions = clientOptions;
this._authProvider = authProvider;
this._policies = policies ?? Policies.DefaultPolicies;
this._policies.LoadBalancingPolicy.Initialize(_cluster);
_keyspace = keyspace ?? clientOptions.DefaultKeyspace;
Guid = Guid.NewGuid();
_trashcanCleaner = new Timer(TranscanCleanup, null, Timeout.Infinite, Timeout.Infinite);
if (init)
{
var ci = this._policies.LoadBalancingPolicy.NewQueryPlan(null).GetEnumerator();
if (!ci.MoveNext())
{
var ex = new NoHostAvailableException(new Dictionary<IPAddress, List<Exception>>());
_logger.Error(ex.Message);
throw ex;
}
var triedHosts = new List<IPAddress>();
var innerExceptions = new Dictionary<IPAddress, List<Exception>>();
int streamId;
var con = Connect(ci, triedHosts, innerExceptions, out streamId);
con.FreeStreamId(streamId);
}
}
catch
{
InternalDispose();
throw;
}
}
internal Session(Cluster cluster,
Policies policies,
ProtocolOptions protocolOptions,
PoolingOptions poolingOptions,
SocketOptions socketOptions,
ClientOptions clientOptions,
IAuthInfoProvider authProvider,
string keyspace)
{
this._cluster = cluster;
this._protocolOptions = protocolOptions;
this._poolingOptions = poolingOptions;
this._socketOptions = socketOptions;
this._clientOptions = clientOptions;
this._authProvider = authProvider;
this._policies = policies ?? Policies.DefaultPolicies;
this._policies.LoadBalancingPolicy.Initialize(_cluster);
_keyspace = keyspace ?? clientOptions.DefaultKeyspace;
}
internal Session(Cluster cluster,
IEnumerable<IPAddress> clusterEndpoints,
Policies policies,
ProtocolOptions protocolOptions,
PoolingOptions poolingOptions,
SocketOptions socketOptions,
ClientOptions clientOptions,
IAuthInfoProvider authProvider,
bool metricsEnabled,
string keyspace,
Hosts hosts,ControlConnection controlConnection=null)
{
this._cluster = cluster;
this._protocolOptions = protocolOptions;
this._poolingOptions = poolingOptions;
this._socketOptions = socketOptions;
this._clientOptions = clientOptions;
this._authProvider = authProvider;
this._metricsEnabled = metricsEnabled;
this._policies = policies ?? Policies.DefaultPolicies;
_hosts = hosts ;
this._controlConnection = controlConnection;
foreach (var ep in clusterEndpoints)
_hosts.AddIfNotExistsOrBringUpIfDown(ep, this._policies.ReconnectionPolicy);
this._policies.LoadBalancingPolicy.Initialize(_cluster);
_keyspace = keyspace ?? clientOptions.DefaultKeyspace;
}
/// <summary>
/// Creates a new instance. This class is also used to shareable a context across all instance that are created below one Cluster instance.
/// One configuration instance per Cluster instance.
/// </summary>
internal Configuration(Policies policies,
ProtocolOptions protocolOptions,
PoolingOptions poolingOptions,
SocketOptions socketOptions,
ClientOptions clientOptions,
IAuthProvider authProvider,
IAuthInfoProvider authInfoProvider,
QueryOptions queryOptions,
IAddressTranslator addressTranslator,
IReadOnlyDictionary <string, IExecutionProfile> executionProfiles,
MetadataSyncOptions metadataSyncOptions,
IEndPointResolver endPointResolver,
IDriverMetricsProvider driverMetricsProvider,
DriverMetricsOptions metricsOptions,
string sessionName,
GraphOptions graphOptions,
Guid?clusterId,
string appVersion,
string appName,
MonitorReportingOptions monitorReportingOptions,
TypeSerializerDefinitions typeSerializerDefinitions,
ISessionFactory sessionFactory = null,
IRequestOptionsMapper requestOptionsMapper = null,
IStartupOptionsFactory startupOptionsFactory = null,
IInsightsSupportVerifier insightsSupportVerifier = null,
IRequestHandlerFactory requestHandlerFactory = null,
IHostConnectionPoolFactory hostConnectionPoolFactory = null,
IRequestExecutionFactory requestExecutionFactory = null,
IConnectionFactory connectionFactory = null,
IControlConnectionFactory controlConnectionFactory = null,
IPrepareHandlerFactory prepareHandlerFactory = null,
ITimerFactory timerFactory = null,
IObserverFactoryBuilder observerFactoryBuilder = null,
IInsightsClientFactory insightsClientFactory = null)
{
AddressTranslator = addressTranslator ?? throw new ArgumentNullException(nameof(addressTranslator));
QueryOptions = queryOptions ?? throw new ArgumentNullException(nameof(queryOptions));
GraphOptions = graphOptions ?? new GraphOptions();
ClusterId = clusterId ?? Guid.NewGuid();
ApplicationVersion = appVersion ?? Configuration.DefaultApplicationVersion;
ApplicationName = appName ?? Configuration.FallbackApplicationName;
ApplicationNameWasGenerated = appName == null;
Policies = policies;
ProtocolOptions = protocolOptions;
PoolingOptions = poolingOptions;
SocketOptions = socketOptions;
ClientOptions = clientOptions;
AuthProvider = authProvider;
AuthInfoProvider = authInfoProvider;
StartupOptionsFactory = startupOptionsFactory ?? new StartupOptionsFactory(ClusterId, ApplicationVersion, ApplicationName);
SessionFactory = sessionFactory ?? new SessionFactory();
RequestOptionsMapper = requestOptionsMapper ?? new RequestOptionsMapper();
MetadataSyncOptions = metadataSyncOptions?.Clone() ?? new MetadataSyncOptions();
DnsResolver = new DnsResolver();
EndPointResolver = endPointResolver ?? new EndPointResolver(DnsResolver, protocolOptions);
MetricsOptions = metricsOptions ?? new DriverMetricsOptions();
MetricsProvider = driverMetricsProvider ?? new NullDriverMetricsProvider();
SessionName = sessionName;
MetricsEnabled = driverMetricsProvider != null;
TypeSerializers = typeSerializerDefinitions?.Definitions;
ObserverFactoryBuilder = observerFactoryBuilder ?? (MetricsEnabled ? (IObserverFactoryBuilder) new MetricsObserverFactoryBuilder() : new NullObserverFactoryBuilder());
RequestHandlerFactory = requestHandlerFactory ?? new RequestHandlerFactory();
HostConnectionPoolFactory = hostConnectionPoolFactory ?? new HostConnectionPoolFactory();
RequestExecutionFactory = requestExecutionFactory ?? new RequestExecutionFactory();
ConnectionFactory = connectionFactory ?? new ConnectionFactory();
ControlConnectionFactory = controlConnectionFactory ?? new ControlConnectionFactory();
PrepareHandlerFactory = prepareHandlerFactory ?? new PrepareHandlerFactory();
TimerFactory = timerFactory ?? new TaskBasedTimerFactory();
RequestOptions = RequestOptionsMapper.BuildRequestOptionsDictionary(executionProfiles, policies, socketOptions, clientOptions, queryOptions, GraphOptions);
ExecutionProfiles = BuildExecutionProfilesDictionary(executionProfiles, RequestOptions);
MonitorReportingOptions = monitorReportingOptions ?? new MonitorReportingOptions();
InsightsSupportVerifier = insightsSupportVerifier ?? Configuration.DefaultInsightsSupportVerifier;
InsightsClientFactory = insightsClientFactory ?? Configuration.DefaultInsightsClientFactory;
// Create the buffer pool with 16KB for small buffers and 256Kb for large buffers.
// The pool does not eagerly reserve the buffers, so it doesn't take unnecessary memory
// to create the instance.
BufferPool = new RecyclableMemoryStreamManager(16 * 1024, 256 * 1024, ProtocolOptions.MaximumFrameLength);
Timer = new HashedWheelTimer();
}
internal CassandraConnection(Session owner, IPAddress serverAddress, ProtocolOptions protocolOptions,
SocketOptions socketOptions, ClientOptions clientOptions,
IAuthProvider authProvider,
IAuthInfoProvider authInfoProvider)
{
this.Guid = Guid.NewGuid();
this._owner = owner;
_bufferingMode = null;
switch (protocolOptions.Compression)
{
case CompressionType.Snappy:
_bufferingMode = new FrameBuffering();
break;
case CompressionType.NoCompression:
_bufferingMode = clientOptions.WithoutRowSetBuffering ? new NoBuffering() : new FrameBuffering();
break;
default:
throw new ArgumentException();
}
this._authProvider = authProvider;
this._authInfoProvider = authInfoProvider;
if (protocolOptions.Compression == CompressionType.Snappy)
{
_startupOptions.Add("COMPRESSION", "snappy");
_compressor = new SnappyProtoBufCompressor();
}
this._serverAddress = serverAddress;
this._port = protocolOptions.Port;
this._queryAbortTimeout = clientOptions.QueryAbortTimeout;
this._socketOptions = socketOptions;
for (int i = 0; i <= sbyte.MaxValue; i++)
{
_freeStreamIDs.Push(i);
}
_protocolErrorHandlerAction = new Action <ErrorActionParam>((param) =>
{
if (param.AbstractResponse is ErrorResponse)
{
JobFinished(
param.Jar,
(param.AbstractResponse as ErrorResponse).Output);
}
});
_frameEventCallback.Value = new Action <ResponseFrame>(EventOccured);
_buffer = new byte[][] {
new byte[_bufferingMode.PreferedBufferSize()],
new byte[_bufferingMode.PreferedBufferSize()]
};
var newSock = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
if (_socketOptions.KeepAlive != null)
{
newSock.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.KeepAlive,
_socketOptions.KeepAlive.Value);
}
newSock.SendTimeout = _socketOptions.ConnectTimeoutMillis;
if (_socketOptions.SoLinger != null)
{
newSock.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.Linger,
new LingerOption(true, _socketOptions.SoLinger.Value));
}
if (_socketOptions.ReceiveBufferSize != null)
{
newSock.ReceiveBufferSize = _socketOptions.ReceiveBufferSize.Value;
}
if (_socketOptions.SendBufferSize != null)
{
newSock.ReceiveBufferSize = _socketOptions.SendBufferSize.Value;
}
if (_socketOptions.TcpNoDelay != null)
{
newSock.NoDelay = _socketOptions.TcpNoDelay.Value;
}
//Avoid waiting more time that expected
var connectionResult = newSock.BeginConnect(_serverAddress, _port, null, null);
connectionResult.AsyncWaitHandle.WaitOne(_socketOptions.ConnectTimeoutMillis);
if (!newSock.Connected)
{
newSock.Close();
throw new SocketException((int)SocketError.TimedOut);
}
_socket = newSock;
_bufferingMode.Reset();
if (protocolOptions.SslOptions == null)
{
_socketStream = new NetworkStream(_socket);
}
else
{
string targetHost;
try
{
targetHost = Dns.GetHostEntry(_serverAddress).HostName;
}
catch (SocketException ex)
{
targetHost = serverAddress.ToString();
_logger.Error(string.Format("SSL connection: Can not resolve {0} address. Using IP address instead of hostname. This may cause RemoteCertificateNameMismatch error during Cassandra host authentication. Note that Cassandra node SSL certificate's CN(Common Name) must match the Cassandra node hostname.", _serverAddress.ToString()), ex);
}
_socketStream = new SslStream(new NetworkStream(_socket), false, new RemoteCertificateValidationCallback(protocolOptions.SslOptions.RemoteCertValidationCallback), null);
(_socketStream as SslStream).AuthenticateAsClient(targetHost, new X509CertificateCollection(), protocolOptions.SslOptions.SslProtocol, false);
}
if (IsHealthy)
{
BeginReading();
}
}