public void HandleInboundMultiPartChunk(MultiPartChunkDto chunk)
{
bool isAdded = false;
ChunkReassemblyContext addedChunkReassemblyContext = null;
var chunkReassemblyContext = chunkReassemblerContextsByMessageId.GetOrAdd(
chunk.MultiPartMessageId,
add => {
// logger.Info(Thread.CurrentThread.ManagedThreadId + ": " + "NEW " + chunk.MultiPartMessageId + " " + this.GetHashCode());
isAdded = true;
return(addedChunkReassemblyContext = new ChunkReassemblyContext(chunk.ChunkCount));
});
// if (isAdded) {
// logger.Info(Thread.CurrentThread.ManagedThreadId + ": " + chunkReassemblyContext.GetHashCode() + " " + new ChunkReassemblyContext(0).GetHashCode() + "");
// }
if (chunkReassemblyContext == addedChunkReassemblyContext)
{
Go(async() => {
await Task.Delay(kSomethingExpiration).ConfigureAwait(false);
RemoveAssemblerFromCache(chunk.MultiPartMessageId);
});
}
var completedChunks = chunkReassemblyContext.AddChunk(chunk);
if (completedChunks != null)
{
ReassembleChunksAndDispatch(completedChunks);
}
}
public MultiPartChunkDto[] AddChunk(MultiPartChunkDto chunk)
{
if (Interlocked.CompareExchange(ref x[chunk.ChunkIndex], chunk, null) == null)
{
var newChunksRemaining = Interlocked.Decrement(ref chunksRemaining);
if (newChunksRemaining < 100)
{
// Console.WriteLine(chunk.MultiPartMessageId.ToString("n").Substring(0, 6) + " MPP REMAINING " + newChunksRemaining);
}
if (newChunksRemaining == 0)
{
return(x);
}
}
return(null);
}
