/// <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 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 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);
}
private Cluster(IEnumerable<IPAddress> contactPoints, Configuration configuration)
{
_contactPoints = contactPoints;
_configuration = configuration;
_metadata = new Metadata(configuration.Policies.ReconnectionPolicy);
var controlpolicies = new Policies(
_configuration.Policies.LoadBalancingPolicy,
new ExponentialReconnectionPolicy(2*1000, 5*60*1000),
Policies.DefaultRetryPolicy);
foreach (IPAddress ep in _contactPoints)
Metadata.AddHost(ep);
//Use 1 connection per host
//The connection will be reused, it wont create a connection per host.
var controlPoolingOptions = new PoolingOptions()
.SetCoreConnectionsPerHost(HostDistance.Local, 1)
.SetMaxConnectionsPerHost(HostDistance.Local, 1)
.SetMinSimultaneousRequestsPerConnectionTreshold(HostDistance.Local, 0)
.SetMaxSimultaneousRequestsPerConnectionTreshold(HostDistance.Local, 127);
var controlConnection = new ControlConnection
(this,
new List<IPAddress>(),
controlpolicies,
new ProtocolOptions(_configuration.ProtocolOptions.Port, configuration.ProtocolOptions.SslOptions),
controlPoolingOptions,
_configuration.SocketOptions,
new ClientOptions(true, _configuration.ClientOptions.QueryAbortTimeout, null),
_configuration.AuthProvider,
_configuration.AuthInfoProvider
);
_metadata.SetupControllConnection(controlConnection);
_binaryProtocolVersion = controlConnection.ProtocolVersion;
_logger.Info("Binary protocol version: [" + _binaryProtocolVersion + "]");
}
/// <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();
}
private Cluster(IEnumerable<IPAddress> contactPoints, Configuration configuration)
{
this._contactPoints = contactPoints;
this._configuration = configuration;
this._metadata = new Metadata(configuration.Policies.ReconnectionPolicy);
var controlpolicies = new Cassandra.Policies(
_configuration.Policies.LoadBalancingPolicy,
new ExponentialReconnectionPolicy(2 * 1000, 5 * 60 * 1000),
Cassandra.Policies.DefaultRetryPolicy);
foreach (var ep in _contactPoints)
Metadata.AddHost(ep);
var poolingOptions = new PoolingOptions().SetCoreConnectionsPerHost(HostDistance.Local, 1);
var controlConnection = new ControlConnection(this, new List<IPAddress>(), controlpolicies,
new ProtocolOptions(_configuration.ProtocolOptions.Port),
poolingOptions, _configuration.SocketOptions,
new ClientOptions(
true,
_configuration.ClientOptions.QueryAbortTimeout, null,
_configuration.ClientOptions.AsyncCallAbortTimeout),
_configuration.AuthInfoProvider);
_metadata.SetupControllConnection(controlConnection);
}
/// <summary>
/// Sets the pooling options for the cluster.
/// </summary>
/// <returns>this instance</returns>
public new DseClusterBuilder WithPoolingOptions(PoolingOptions value)
{
base.WithPoolingOptions(value);
return this;
}
/// <summary>
/// Creates a new session on this cluster and sets a keyspace to use.
/// </summary>
/// <param name="keyspace"> The name of the keyspace to use for the created <code>Session</code>. </param>
/// <returns>a new session on this cluster set to keyspace:
/// <code>keyspaceName</code>. </returns>
public Session Connect(string keyspace)
{
bool controlConnectionCreated = false;
lock (_controlConnectionGuard)
{
if (_controlConnection == null)
{
controlConnectionCreated = true;
var controlpolicies = new Cassandra.Policies(
_configuration.Policies.LoadBalancingPolicy,
new ExponentialReconnectionPolicy(2 * 1000, 5 * 60 * 1000),
Cassandra.Policies.DefaultRetryPolicy);
_hosts = new Hosts(_configuration.Policies.ReconnectionPolicy);
foreach (var ep in _contactPoints)
_hosts.AddIfNotExistsOrBringUpIfDown(ep);
var poolingOptions = new PoolingOptions().SetCoreConnectionsPerHost(HostDistance.Local, 1);
_controlConnection = new ControlConnection(this, new List<IPAddress>(), controlpolicies,
_configuration.ProtocolOptions,
poolingOptions, _configuration.SocketOptions,
new ClientOptions(
_configuration.ClientOptions.WithoutRowSetBuffering,
_configuration.ClientOptions.QueryAbortTimeout, null,
_configuration.ClientOptions.AsyncCallAbortTimeout),
_configuration.AuthInfoProvider,
_configuration.MetricsEnabled);
_metadata = new Metadata(_hosts, _controlConnection);
_controlConnection.Init();
}
}
var scs = new Session(this, _contactPoints, _configuration.Policies,
_configuration.ProtocolOptions,
_configuration.PoolingOptions, _configuration.SocketOptions,
_configuration.ClientOptions,
_configuration.AuthInfoProvider, _configuration.MetricsEnabled, keyspace, _hosts);
scs.Init();
lock (_connectedSessions)
_connectedSessions.Add(scs);
_logger.Info("Session connected!");
if (controlConnectionCreated)
RefreshSchema();
return scs;
}
internal int?GetHeartBeatInterval()
{
return(PoolingOptions != null?PoolingOptions.GetHeartBeatInterval() : PoolingOptions.Create().GetHeartBeatInterval());
}
public Builder WithPoolingOptions(PoolingOptions value)
{
_poolingOptions = value;
return this;
}
public Builder WithPoolingOptions(PoolingOptions value)
{
_poolingOptions = value;
return(this);
}
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;
}
}
/// <summary>
/// Gets the pooling options. If not specified, creates a new instance with the default by protocol version.
/// This instance is not stored.
/// </summary>
internal PoolingOptions GetOrCreatePoolingOptions(ProtocolVersion protocolVersion)
{
return(PoolingOptions ?? PoolingOptions.Create(protocolVersion));
}
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>
/// Gets the pooling options. If not specified, gets the default by protocol version
/// </summary>
internal PoolingOptions GetPoolingOptions(byte protocolVersion)
{
if (this._poolingOptions != null)
{
return this._poolingOptions;
}
this._poolingOptions = PoolingOptions.GetDefault(protocolVersion);
return this._poolingOptions;
}
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;
}