public static byte[] BuildPrimaryKeyWithHeader(PrimaryKey primaryKey)
{
int size = ComputePrimaryKeyWithHeader(primaryKey);
PlainBufferOutputStream output = new PlainBufferOutputStream(size);
output.WriteRawLittleEndian32(PlainBufferConsts.HEADER);
output.WriteRawByte(PlainBufferConsts.TAG_ROW_PK);
byte rowChecksum = (byte)0x0, cellChecksum;
foreach (var key in primaryKey)
{
var column = new Column(key.Key, key.Value);
cellChecksum = PlainBufferCrc8.crc8((byte)0x0, column.GetNameRawData());
cellChecksum = column.Value.GetChecksum(cellChecksum);
WritePrimaryKeyColumn(column, output, cellChecksum);
rowChecksum = PlainBufferCrc8.crc8(rowChecksum, cellChecksum);
}
// 没有deleteMarker, 要与0x0做crc.
rowChecksum = PlainBufferCrc8.crc8(rowChecksum, (byte)0x0);
output.WriteRawByte(PlainBufferConsts.TAG_ROW_CHECKSUM);
output.WriteRawByte(rowChecksum);
if (!output.IsFull())
{
throw new IOException("Bug: serialize primary key failed.");
}
return(output.GetBuffer());
}
public static void WritePrimaryKeyColumn(Column column, PlainBufferOutputStream output, byte checksum)
{
output.WriteRawByte(PlainBufferConsts.TAG_CELL);
output.WriteRawByte(PlainBufferConsts.TAG_CELL_NAME);
byte[] rawData = column.GetNameRawData();
output.WriteRawLittleEndian32(rawData.Length);
output.WriteBytes(rawData);
output.WriteRawByte(PlainBufferConsts.TAG_CELL_VALUE);
WritePrimaryKeyValue(column.Value, output);
output.WriteRawByte(PlainBufferConsts.TAG_CELL_CHECKSUM);
output.WriteRawByte(checksum);
}
public static void WritePrimaryKeyValue(ColumnValue value, PlainBufferOutputStream output)
{
if (value.IsInfMin())
{
output.WriteRawLittleEndian32(1);
output.WriteRawByte(PlainBufferConsts.VT_INF_MIN);
return;
}
if (value.IsInfMax())
{
output.WriteRawLittleEndian32(1);
output.WriteRawByte(PlainBufferConsts.VT_INF_MAX);
return;
}
if (value.IsPlaceHolderForAutoIncr())
{
output.WriteRawLittleEndian32(1);
output.WriteRawByte(PlainBufferConsts.VT_AUTO_INCREMENT);
return;
}
switch (value.Type)
{
case ColumnValueType.String:
{
byte[] rawData = value.AsStringInBytes();
int prefixLength = PlainBufferOutputStream.LITTLE_ENDIAN_32_SIZE + 1; // length + type + length
output.WriteRawLittleEndian32(prefixLength + rawData.Length); // length + type + value
output.WriteRawByte(PlainBufferConsts.VT_STRING);
output.WriteRawLittleEndian32(rawData.Length);
output.WriteBytes(rawData);
break;
}
case ColumnValueType.Integer:
{
output.WriteRawLittleEndian32(1 + PlainBufferOutputStream.LITTLE_ENDIAN_64_SIZE);
output.WriteRawByte(PlainBufferConsts.VT_INTEGER);
output.WriteRawLittleEndian64(value.IntegerValue);
break;
}
case ColumnValueType.Binary:
{
byte[] rawData = value.BinaryValue;
int prefixLength = PlainBufferOutputStream.LITTLE_ENDIAN_32_SIZE + 1; // length + type + length
output.WriteRawLittleEndian32(prefixLength + rawData.Length); // length + type + value
output.WriteRawByte(PlainBufferConsts.VT_BLOB);
output.WriteRawLittleEndian32(rawData.Length);
output.WriteBytes(rawData);
break;
}
default:
throw new IOException("Bug: unsupported primary key type: " + value.GetType());
}
}
public void WriteTag(byte tag)
{
output.WriteRawByte(tag);
}