internal TcpConnection(IntPtr socket, long connectionId, WorkBundle wb, RIO rio)
{
_socket = socket;
_connectionId = connectionId;
_rio = rio;
_wb = wb;
_requestQueue = _rio.CreateRequestQueue(_socket, MaxPendingReceives, 1, MaxPendingSends, 1, wb.completionQueue, wb.completionQueue, connectionId);
if (_requestQueue == IntPtr.Zero)
{
var error = RIOImports.WSAGetLastError();
RIOImports.WSACleanup();
throw new Exception(String.Format("ERROR: CreateRequestQueue returned {0}", error));
}
_receiveTasks = new ReceiveTask[MaxPendingReceives];
_receiveRequestBuffers = new ArraySegment <byte> [MaxPendingReceives];
for (var i = 0; i < _receiveTasks.Length; i++)
{
_receiveTasks[i] = new ReceiveTask(this, wb.bufferPool.GetBuffer());
}
_sendSegments = new PooledSegment[MaxPendingSends];
for (var i = 0; i < _sendSegments.Length; i++)
{
_sendSegments[i] = wb.bufferPool.GetBuffer();
}
wb.connections.TryAdd(connectionId, this);
for (var i = 0; i < _receiveTasks.Length; i++)
{
PostReceive(i);
}
}
internal TcpConnection(IntPtr socket, long connectionId, WorkBundle wb, RIO rio)
{
_socket = socket;
_connectionId = connectionId;
_rio = rio;
_wb = wb;
_requestQueue = _rio.CreateRequestQueue(_socket, MaxPendingReceives, 1, MaxPendingSends, 1, wb.completionQueue, wb.completionQueue, connectionId);
if (_requestQueue == IntPtr.Zero)
{
var error = RIOImports.WSAGetLastError();
RIOImports.WSACleanup();
throw new Exception(String.Format("ERROR: CreateRequestQueue returned {0}", error));
}
_receiveTasks = new ReceiveTask[MaxPendingReceives];
_receiveRequestBuffers = new ArraySegment<byte>[MaxPendingReceives];
for (var i = 0; i < _receiveTasks.Length; i++)
{
_receiveTasks[i] = new ReceiveTask(this, wb.bufferPool.GetBuffer());
}
_sendSegments = new PooledSegment[MaxPendingSends];
for (var i = 0; i < _sendSegments.Length; i++)
{
_sendSegments[i] = wb.bufferPool.GetBuffer();
}
wb.connections.TryAdd(connectionId, this);
for (var i = 0; i < _receiveTasks.Length; i++)
{
PostReceive(i);
}
}
public unsafe RIOThreadPool(RIO rio, IntPtr socket, CancellationToken token)
{
_socket = socket;
_rio = rio;
_token = token;
_maxThreads = Environment.ProcessorCount;
_workers = new WorkBundle[_maxThreads];
for (var i = 0; i < _workers.Length; i++)
{
var worker = new WorkBundle()
{
id = i,
bufferPool = new RIOBufferPool(_rio)
};
worker.completionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, IntPtr.Zero, 0, 0);
if (worker.completionPort == IntPtr.Zero)
{
var error = GetLastError();
RIOImports.WSACleanup();
throw new Exception(String.Format("ERROR: CreateIoCompletionPort returned {0}", error));
}
var completionMethod = new RIO_NOTIFICATION_COMPLETION()
{
Type = RIO_NOTIFICATION_COMPLETION_TYPE.IOCP_COMPLETION,
Iocp = new RIO_NOTIFICATION_COMPLETION_IOCP()
{
IocpHandle = worker.completionPort,
QueueCorrelation = (ulong)i,
Overlapped = (NativeOverlapped*)(-1)// nativeOverlapped
}
};
worker.completionQueue = _rio.CreateCompletionQueue(MaxOutsandingCompletions, completionMethod);
if (worker.completionQueue == IntPtr.Zero)
{
var error = RIOImports.WSAGetLastError();
RIOImports.WSACleanup();
throw new Exception(String.Format("ERROR: CreateCompletionQueue returned {0}", error));
}
worker.connections = new ConcurrentDictionary<long, TcpConnection>();
worker.thread = new Thread(GetThreadStart(i));
worker.thread.IsBackground = true;
_workers[i] = worker;
}
// gc
//GC.Collect(2, GCCollectionMode.Forced, true, true);
//GC.WaitForPendingFinalizers();
//GC.Collect(2, GCCollectionMode.Forced, true, true);
GC.Collect(2, GCCollectionMode.Forced, true);
GC.WaitForPendingFinalizers();
GC.Collect(2, GCCollectionMode.Forced, true);
for (var i = 0; i < _workers.Length; i++)
{
// pin buffers
_workers[i].bufferPool.Initalize();
}
for (var i = 0; i < _workers.Length; i++)
{
_workers[i].thread.Start();
}
}
public unsafe RIOThreadPool(RIO rio, IntPtr socket, CancellationToken token)
{
_socket = socket;
_rio = rio;
_token = token;
_maxThreads = Environment.ProcessorCount;
_workers = new WorkBundle[_maxThreads];
for (var i = 0; i < _workers.Length; i++)
{
var worker = new WorkBundle()
{
id = i,
bufferPool = new RIOBufferPool(_rio)
};
worker.completionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, IntPtr.Zero, 0, 0);
if (worker.completionPort == IntPtr.Zero)
{
var error = GetLastError();
RIOImports.WSACleanup();
throw new Exception(String.Format("ERROR: CreateIoCompletionPort returned {0}", error));
}
var completionMethod = new RIO_NOTIFICATION_COMPLETION()
{
Type = RIO_NOTIFICATION_COMPLETION_TYPE.IOCP_COMPLETION,
Iocp = new RIO_NOTIFICATION_COMPLETION_IOCP()
{
IocpHandle = worker.completionPort,
QueueCorrelation = (ulong)i,
Overlapped = (NativeOverlapped *)(-1)// nativeOverlapped
}
};
worker.completionQueue = _rio.CreateCompletionQueue(MaxOutsandingCompletions, completionMethod);
if (worker.completionQueue == IntPtr.Zero)
{
var error = RIOImports.WSAGetLastError();
RIOImports.WSACleanup();
throw new Exception(String.Format("ERROR: CreateCompletionQueue returned {0}", error));
}
worker.connections = new ConcurrentDictionary <long, TcpConnection>();
worker.thread = new Thread(GetThreadStart(i));
worker.thread.IsBackground = true;
_workers[i] = worker;
}
// gc
//GC.Collect(2, GCCollectionMode.Forced, true, true);
//GC.WaitForPendingFinalizers();
//GC.Collect(2, GCCollectionMode.Forced, true, true);
GC.Collect(2, GCCollectionMode.Forced, true);
GC.WaitForPendingFinalizers();
GC.Collect(2, GCCollectionMode.Forced, true);
for (var i = 0; i < _workers.Length; i++)
{
// pin buffers
_workers[i].bufferPool.Initalize();
}
for (var i = 0; i < _workers.Length; i++)
{
_workers[i].thread.Start();
}
}
