/// <summary>Add the given <c>msg</c> and returns <see cref="Task" /> for completion of processing <c>msg</c>.</summary>
public void Add(object msg, IPromise promise)
{
Debug.Assert(_ctx.Executor.InEventLoop);
if (msg is null)
{
ThrowHelper.ThrowArgumentNullException(ExceptionArgument.msg);
}
int messageSize = _estimatorHandle.Size(msg);
if (messageSize < 0)
{
// Size may be unknow so just use 0
messageSize = 0;
}
PendingWrite currentTail = _tail;
if (currentTail is object)
{
bool canBundle = CanBatch(msg, messageSize, currentTail.Size);
if (canBundle)
{
currentTail.Add(msg, messageSize);
if (!promise.IsVoid)
{
currentTail.Promise.Task.LinkOutcome(promise);
}
return;
}
}
PendingWrite write;
if (promise.IsVoid || promise is SimplePromiseAggregator)
{
var headPromise = _ctx.NewPromise();
headPromise.Task.LinkOutcome(promise);
write = PendingWrite.NewInstance(msg, messageSize, headPromise);
}
else
{
write = PendingWrite.NewInstance(msg, messageSize, promise);
}
if (currentTail is null)
{
_tail = _head = write;
}
else
{
currentTail.Next = write;
_tail = write;
}
_size++;
// We need to guard against null as channel.Unsafe.OutboundBuffer may returned null
// if the channel was already closed when constructing the PendingWriteQueue.
// See https://github.com/netty/netty/issues/3967
_buffer?.IncrementPendingOutboundBytes(messageSize);
}
private int GetSize(object msg)
{
// It is possible for writes to be triggered from removeAndFailAll(). To preserve ordering,
// we should add them to the queue and let removeAndFailAll() fail them later.
int messageSize = _estimatorHandle.Size(msg);
if (messageSize < 0)
{
// Size may be unknown so just use 0
messageSize = 0;
}
return(messageSize + PendingWriteQueue.PendingWriteOverhead);
}
public virtual int Size(object msg) => _estimatorHandle.Size(msg);