mirror of
https://github.com/hanzo-ds/native.git
synced 2026-10-30 20:00:34 +00:00
zeekay
765c3d0aa1
Module path, package names, and identifiers de-branded to native/proto/compress/ pool/tracing. Server wire-protocol strings preserved. Vendored codec surface used by hanzo-ds/go (proto + compress).
53 lines
1.0 KiB
Go
53 lines
1.0 KiB
Go
//go:build !(amd64 || arm64 || riscv64) || purego
|
|
|
|
package proto
|
|
|
|
import "github.com/go-faster/errors"
|
|
|
|
// EncodeColumn encodes Bool rows to *Buffer.
|
|
func (c ColBool) EncodeColumn(b *Buffer) {
|
|
start := len(b.Buf)
|
|
b.Buf = append(b.Buf, make([]byte, len(c))...)
|
|
dst := b.Buf[start:]
|
|
for i, v := range c {
|
|
dst[i] = boolToByte(v)
|
|
}
|
|
}
|
|
|
|
func boolToByte(b bool) byte {
|
|
if b {
|
|
return boolTrue
|
|
}
|
|
return boolFalse
|
|
}
|
|
|
|
// DecodeColumn decodes Bool rows from *Reader.
|
|
func (c *ColBool) DecodeColumn(r *Reader, rows int) error {
|
|
data, err := r.ReadRaw(rows)
|
|
if err != nil {
|
|
return errors.Wrap(err, "read")
|
|
}
|
|
v := *c
|
|
v = append(v, make([]bool, rows)...)
|
|
for i := range data {
|
|
switch data[i] {
|
|
case boolTrue:
|
|
v[i] = true
|
|
case boolFalse:
|
|
v[i] = false
|
|
default:
|
|
return errors.Errorf("[%d]: bad value %d for Bool", i, data[i])
|
|
}
|
|
}
|
|
*c = v
|
|
return nil
|
|
}
|
|
|
|
// WriteColumn encodes ColBool rows to *Writer.
|
|
func (c ColBool) WriteColumn(w *Writer) {
|
|
if len(c) == 0 {
|
|
return
|
|
}
|
|
w.ChainBuffer(c.EncodeColumn)
|
|
}
|