write pre-encoded key and value lengths

Author: apacker

protocol/encode.go

@@ -340,6 +340,16 @@ func (e *encoder) writeVarNullBytesFrom(b Bytes) error {
 	}
 }
 
+func (e *encoder) writeRawVarNullBytesFrom(b Bytes) error {
+	if b == nil {
+		e.writeVarInt(-1)
+		return nil
+	} else {
+		_, err := io.Copy(e, b)
+		return err
+	}
+}
+
 func (e *encoder) writeVarInt(i int64) {
 	e.writeUnsignedVarInt(uint64((i << 1) ^ (i >> 63)))
 }

protocol/record_v2.go

@@ -370,11 +370,13 @@ func (rs *RecordSet) writePreCompressed(buffer *pageBuffer, bufferOffset int64)
 		e.writeVarInt(timestampDelta)
 		e.writeVarInt(offsetDelta)
 
-		if err := e.writeVarNullBytesFrom(r.Key); err != nil {
+		e.writeUnsignedVarInt(uint64(r.PreEncodedKeylen))
+		if err := e.writeRawVarNullBytesFrom(r.Key); err != nil {
 			return err
 		}
 
-		if err := e.writeVarNullBytesFrom(r.Value); err != nil {
+		e.writeUnsignedVarInt(uint64(r.PreEncodedValueLen))
+		if err := e.writeRawVarNullBytesFrom(r.Value); err != nil {
 			return err
 		}