/// <summary>
/// Writes a raw BSON array.
/// </summary>
/// <param name="slice">The byte buffer containing the raw BSON array.</param>
public virtual void WriteRawBsonArray(IByteBuffer slice)
{
// overridden in BsonBinaryWriter to write the raw bytes to the stream
// for all other streams, deserialize the raw bytes and serialize the resulting array instead
using (var chunkSource = new InputBufferChunkSource(BsonChunkPool.Default))
using (var buffer = new MultiChunkBuffer(chunkSource))
using (var stream = new ByteBufferStream(buffer))
{
// wrap the array in a fake document so we can deserialize it
var documentLength = slice.Length + 8;
buffer.EnsureCapacity(documentLength);
stream.WriteInt32(documentLength);
stream.WriteBsonType(BsonType.Array);
stream.WriteByte((byte)'x');
stream.WriteByte(0);
stream.WriteSlice(slice);
stream.WriteByte(0);
buffer.MakeReadOnly();
stream.Position = 0;
using (var reader = new BsonBinaryReader(stream, BsonBinaryReaderSettings.Defaults))
{
var deserializationContext = BsonDeserializationContext.CreateRoot(reader);
reader.ReadStartDocument();
reader.ReadName("x");
var array = BsonArraySerializer.Instance.Deserialize(deserializationContext);
reader.ReadEndDocument();
var serializationContext = BsonSerializationContext.CreateRoot(this);
BsonArraySerializer.Instance.Serialize(serializationContext, array);
}
}
}
/// <summary>
/// Writes a raw BSON array.
/// </summary>
/// <param name="slice">The byte buffer containing the raw BSON array.</param>
public virtual void WriteRawBsonArray(IByteBuffer slice)
{
// overridden in BsonBinaryWriter to write the raw bytes to the stream
// for all other streams, deserialize the raw bytes and serialize the resulting array instead
using (var chunkSource = new InputBufferChunkSource(BsonChunkPool.Default))
using (var buffer = new MultiChunkBuffer(chunkSource))
using (var stream = new ByteBufferStream(buffer))
{
// wrap the array in a fake document so we can deserialize it
var documentLength = slice.Length + 8;
buffer.EnsureCapacity(documentLength);
stream.WriteInt32(documentLength);
stream.WriteBsonType(BsonType.Array);
stream.WriteByte((byte)'x');
stream.WriteByte(0);
stream.WriteSlice(slice);
stream.WriteByte(0);
buffer.MakeReadOnly();
stream.Position = 0;
using (var reader = new BsonBinaryReader(stream, BsonBinaryReaderSettings.Defaults))
{
var deserializationContext = BsonDeserializationContext.CreateRoot(reader);
reader.ReadStartDocument();
reader.ReadName("x");
var array = BsonArraySerializer.Instance.Deserialize(deserializationContext);
reader.ReadEndDocument();
var serializationContext = BsonSerializationContext.CreateRoot(this);
BsonArraySerializer.Instance.Serialize(serializationContext, array);
}
}
}
public void EnsureCapacity_should_have_expected_effect(int minimumCapacity, int[] expectedChunkSizes)
{
var mockChunkSource = new Mock<IBsonChunkSource>();
var subject = new MultiChunkBuffer(mockChunkSource.Object);
var chunkSize = 1;
mockChunkSource.Setup(s => s.GetChunk(It.IsAny<int>())).Returns(() => new ByteArrayChunk(chunkSize++));
subject.EnsureCapacity(minimumCapacity);
var reflector = new Reflector(subject);
subject.Capacity.Should().BeGreaterOrEqualTo(minimumCapacity);
reflector._chunks.Select(c => c.Bytes.Count).Should().Equal(expectedChunkSizes);
}
public void ExpandCapacity_should_throw_when_expanded_capacity_exceeds_2GB()
{
RequireProcess.Is64Bit();
using (var subject = new MultiChunkBuffer(BsonChunkPool.Default))
{
subject.EnsureCapacity(int.MaxValue - 128 * 1024 * 1024);
Action action = () => subject.EnsureCapacity(int.MaxValue); // indirectly calls private ExpandCapacity method
action.ShouldThrow<InvalidOperationException>();
}
}
public void EnsureCapacity_should_have_expected_effect(int minimumCapacity, int[] expectedChunkSizes)
{
var chunkSource = Substitute.For<IBsonChunkSource>();
var subject = new MultiChunkBuffer(chunkSource);
var chunkSize = 1;
chunkSource.GetChunk(Arg.Any<int>()).Returns(x => new ByteArrayChunk(chunkSize++));
subject.EnsureCapacity(minimumCapacity);
var reflector = new Reflector(subject);
subject.Capacity.Should().BeGreaterOrEqualTo(minimumCapacity);
reflector._chunks.Select(c => c.Bytes.Count).Should().Equal(expectedChunkSizes);
}
