/// <summary>
/// Read any BsonValue. Use 1 byte for data type, 1 byte for length (optional), 0-255 bytes to value.
/// For document or array, use BufferReader
/// </summary>
public static BsonValue ReadIndexKey(this BufferSlice buffer, int offset)
{
ExtendedLengthHelper.ReadLength(buffer[offset++], buffer[offset], out var type, out var len);
switch (type)
{
case BsonType.Null: return(BsonValue.Null);
case BsonType.Int32: return(buffer.ReadInt32(offset));
case BsonType.Int64: return(buffer.ReadInt64(offset));
case BsonType.Double: return(buffer.ReadDouble(offset));
case BsonType.Decimal: return(buffer.ReadDecimal(offset));
case BsonType.String:
offset++; // for byte length
return(buffer.ReadString(offset, len));
case BsonType.Document:
using (var r = new BufferReader(buffer))
{
r.Skip(offset); // skip first byte for value.Type
return(r.ReadDocument());
}
case BsonType.Array:
using (var r = new BufferReader(buffer))
{
r.Skip(offset); // skip first byte for value.Type
return(r.ReadArray());
}
case BsonType.Binary:
offset++; // for byte length
return(buffer.ReadBytes(offset, len));
case BsonType.ObjectId: return(buffer.ReadObjectId(offset));
case BsonType.Guid: return(buffer.ReadGuid(offset));
case BsonType.Boolean: return(buffer[offset] != 0);
case BsonType.DateTime: return(buffer.ReadDateTime(offset));
case BsonType.MinValue: return(BsonValue.MinValue);
case BsonType.MaxValue: return(BsonValue.MaxValue);
default: throw new NotImplementedException();
}
}
/// <summary>
/// Wrtie any BsonValue. Use 1 byte for data type, 1 byte for length (optional), 0-255 bytes to value.
/// For document or array, use BufferWriter
/// </summary>
public static void WriteIndexKey(this BufferSlice buffer, BsonValue value, int offset)
{
ENSURE(value.GetBytesCount(false) <= MAX_INDEX_KEY_LENGTH, "index key must have less than 1023 bytes");
if (value.IsString)
{
var str = value.AsString;
var strLength = (ushort)Encoding.UTF8.GetByteCount(str);
ExtendedLengthHelper.WriteLength(BsonType.String, strLength, out var typeByte, out var lengthByte);
buffer[offset++] = typeByte;
buffer[offset++] = lengthByte;
buffer.Write(str, offset);
}
else if (value.IsBinary)
{
var arr = value.AsBinary;
ExtendedLengthHelper.WriteLength(BsonType.Binary, (ushort)arr.Length, out var typeByte, out var lengthByte);
buffer[offset++] = typeByte;
buffer[offset++] = lengthByte;
buffer.Write(arr, offset);
}
else
{
buffer[offset++] = (byte)value.Type;
switch (value.Type)
{
case BsonType.Null:
case BsonType.MinValue:
case BsonType.MaxValue:
break;
case BsonType.Int32: buffer.Write(value.AsInt32, offset); break;
case BsonType.Int64: buffer.Write(value.AsInt64, offset); break;
case BsonType.Double: buffer.Write(value.AsDouble, offset); break;
case BsonType.Decimal: buffer.Write(value.AsDecimal, offset); break;
case BsonType.Document:
using (var w = new BufferWriter(buffer))
{
w.Skip(offset); // skip offset from buffer
w.WriteDocument(value.AsDocument, true);
}
break;
case BsonType.Array:
using (var w = new BufferWriter(buffer))
{
w.Skip(offset); // skip offset from buffer
w.WriteArray(value.AsArray, true);
}
break;
case BsonType.ObjectId: buffer.Write(value.AsObjectId, offset); break;
case BsonType.Guid: buffer.Write(value.AsGuid, offset); break;
case BsonType.Boolean: buffer[offset] = (value.AsBoolean) ? (byte)1 : (byte)0; break;
case BsonType.DateTime: buffer.Write(value.AsDateTime, offset); break;
default: throw new NotImplementedException();
}
}
}
