154d055f9f
for convenience. Signed-off-by: Izuru Yakumo <yakumo.izuru@chaotic.ninja> git-svn-id: https://svn.yakumo.dev/yakumo.izuru/suika/trunk@822 f0ae65fe-ee39-954e-97ec-027ff2717ef4
117 lines
2.5 KiB
Go
117 lines
2.5 KiB
Go
package bare
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"fmt"
|
|
"io"
|
|
"math"
|
|
)
|
|
|
|
// A Writer for BARE primitive types.
|
|
type Writer struct {
|
|
base io.Writer
|
|
scratch [binary.MaxVarintLen64]byte
|
|
}
|
|
|
|
// Returns a new BARE primitive writer wrapping the given io.Writer.
|
|
func NewWriter(base io.Writer) *Writer {
|
|
return &Writer{base: base}
|
|
}
|
|
|
|
func (w *Writer) WriteUint(i uint64) error {
|
|
n := binary.PutUvarint(w.scratch[:], i)
|
|
_, err := w.base.Write(w.scratch[:n])
|
|
return err
|
|
}
|
|
|
|
func (w *Writer) WriteU8(i uint8) error {
|
|
return binary.Write(w.base, binary.LittleEndian, i)
|
|
}
|
|
|
|
func (w *Writer) WriteU16(i uint16) error {
|
|
return binary.Write(w.base, binary.LittleEndian, i)
|
|
}
|
|
|
|
func (w *Writer) WriteU32(i uint32) error {
|
|
return binary.Write(w.base, binary.LittleEndian, i)
|
|
}
|
|
|
|
func (w *Writer) WriteU64(i uint64) error {
|
|
return binary.Write(w.base, binary.LittleEndian, i)
|
|
}
|
|
|
|
func (w *Writer) WriteInt(i int64) error {
|
|
var buf [binary.MaxVarintLen64]byte
|
|
n := binary.PutVarint(buf[:], i)
|
|
_, err := w.base.Write(buf[:n])
|
|
return err
|
|
}
|
|
|
|
func (w *Writer) WriteI8(i int8) error {
|
|
return binary.Write(w.base, binary.LittleEndian, i)
|
|
}
|
|
|
|
func (w *Writer) WriteI16(i int16) error {
|
|
return binary.Write(w.base, binary.LittleEndian, i)
|
|
}
|
|
|
|
func (w *Writer) WriteI32(i int32) error {
|
|
return binary.Write(w.base, binary.LittleEndian, i)
|
|
}
|
|
|
|
func (w *Writer) WriteI64(i int64) error {
|
|
return binary.Write(w.base, binary.LittleEndian, i)
|
|
}
|
|
|
|
func (w *Writer) WriteF32(f float32) error {
|
|
if math.IsNaN(float64(f)) {
|
|
return fmt.Errorf("NaN is not permitted in BARE floats")
|
|
}
|
|
return binary.Write(w.base, binary.LittleEndian, f)
|
|
}
|
|
|
|
func (w *Writer) WriteF64(f float64) error {
|
|
if math.IsNaN(f) {
|
|
return fmt.Errorf("NaN is not permitted in BARE floats")
|
|
}
|
|
return binary.Write(w.base, binary.LittleEndian, f)
|
|
}
|
|
|
|
func (w *Writer) WriteBool(b bool) error {
|
|
return binary.Write(w.base, binary.LittleEndian, b)
|
|
}
|
|
|
|
func (w *Writer) WriteString(str string) error {
|
|
return w.WriteData([]byte(str))
|
|
}
|
|
|
|
// Writes a fixed amount of arbitrary data, defined by the length of the slice.
|
|
func (w *Writer) WriteDataFixed(data []byte) error {
|
|
var amt int = 0
|
|
for amt < len(data) {
|
|
n, err := w.base.Write(data[amt:])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
amt += n
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Writes arbitrary data whose length is encoded into the message.
|
|
func (w *Writer) WriteData(data []byte) error {
|
|
err := w.WriteUint(uint64(len(data)))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
var amt int = 0
|
|
for amt < len(data) {
|
|
n, err := w.base.Write(data[amt:])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
amt += n
|
|
}
|
|
return nil
|
|
}
|