CassandraConnection AllocateConnection(IPAddress endPoint, HostDistance hostDistance, out Exception outExc)
{
CassandraConnection nconn = null;
outExc = null;
try
{
int no = 1;
if (!_allocatedConnections.TryAdd(endPoint, new AtomicValue<int>(1)))
{
AtomicValue<int> val;
_allocatedConnections.TryGetValue(endPoint, out val);
no = Interlocked.Increment(ref val.RawValue);
if (no > _poolingOptions.GetMaxConnectionPerHost(hostDistance))
{
Interlocked.Decrement(ref val.RawValue);
outExc = new ToManyConnectionsPerHost();
return null;
}
}
nconn = new CassandraConnection(this, endPoint, _protocolOptions, _socketOptions, _clientOptions, _authProvider);
var streamId = nconn.AllocateStreamId();
var options = ProcessExecuteOptions(nconn.ExecuteOptions(streamId));
if (!string.IsNullOrEmpty(_keyspace))
nconn.SetKeyspace(_keyspace);
}
catch (Exception ex)
{
if (nconn != null)
{
nconn.Dispose();
nconn = null;
}
AtomicValue<int> val;
_allocatedConnections.TryGetValue(endPoint, out val);
Interlocked.Decrement(ref val.RawValue);
if (CassandraConnection.IsStreamRelatedException(ex))
{
HostIsDown(endPoint);
outExc = ex;
return null;
}
else
throw ex;
}
_logger.Info("Allocated new connection");
return nconn;
}
CassandraConnection AllocateConnection(IPAddress endPoint, HostDistance hostDistance, out Exception outExc)
{
CassandraConnection nconn = null;
outExc = null;
try
{
int no = 1;
if (!_allocatedConnections.TryAdd(endPoint, new AtomicValue <int>(1)))
{
AtomicValue <int> val;
_allocatedConnections.TryGetValue(endPoint, out val);
no = Interlocked.Increment(ref val.RawValue);
if (no > _poolingOptions.GetMaxConnectionPerHost(hostDistance))
{
Interlocked.Decrement(ref val.RawValue);
outExc = new ToManyConnectionsPerHost();
return(null);
}
}
RETRY:
nconn = new CassandraConnection(this, endPoint, _protocolOptions, _socketOptions, _clientOptions, _authProvider, _authInfoProvider, _binaryProtocolVersion);
var streamId = nconn.AllocateStreamId();
try
{
var options = ProcessExecuteOptions(nconn.ExecuteOptions(streamId));
}
catch (CassandraConnectionBadProtocolVersionException)
{
if (_binaryProtocolVersion == 1)
{
throw;
}
else
{
_binaryProtocolVersion = 1;
goto RETRY;
}
}
if (!string.IsNullOrEmpty(_keyspace))
{
nconn.SetKeyspace(_keyspace);
}
}
catch (Exception ex)
{
if (nconn != null)
{
nconn.Dispose();
nconn = null;
}
AtomicValue <int> val;
_allocatedConnections.TryGetValue(endPoint, out val);
Interlocked.Decrement(ref val.RawValue);
if (CassandraConnection.IsStreamRelatedException(ex))
{
HostIsDown(endPoint);
outExc = ex;
return(null);
}
else
{
throw ex;
}
}
_logger.Info("Allocated new connection");
return(nconn);
}
