106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
package proto
|
|
|
|
import (
|
|
"encoding"
|
|
"fmt"
|
|
"strconv"
|
|
)
|
|
|
|
const bufferSize = 4096
|
|
|
|
type WriteBuffer struct {
|
|
b []byte
|
|
}
|
|
|
|
func NewWriteBuffer() *WriteBuffer {
|
|
return &WriteBuffer{
|
|
b: make([]byte, 0, 4096),
|
|
}
|
|
}
|
|
|
|
func (w *WriteBuffer) Len() int { return len(w.b) }
|
|
func (w *WriteBuffer) Bytes() []byte { return w.b }
|
|
func (w *WriteBuffer) Reset() { w.b = w.b[:0] }
|
|
|
|
func (w *WriteBuffer) Append(args []interface{}) error {
|
|
w.b = append(w.b, ArrayReply)
|
|
w.b = strconv.AppendUint(w.b, uint64(len(args)), 10)
|
|
w.b = append(w.b, '\r', '\n')
|
|
|
|
for _, arg := range args {
|
|
if err := w.append(arg); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (w *WriteBuffer) append(val interface{}) error {
|
|
switch v := val.(type) {
|
|
case nil:
|
|
w.AppendString("")
|
|
case string:
|
|
w.AppendString(v)
|
|
case []byte:
|
|
w.AppendBytes(v)
|
|
case int:
|
|
w.AppendString(formatInt(int64(v)))
|
|
case int8:
|
|
w.AppendString(formatInt(int64(v)))
|
|
case int16:
|
|
w.AppendString(formatInt(int64(v)))
|
|
case int32:
|
|
w.AppendString(formatInt(int64(v)))
|
|
case int64:
|
|
w.AppendString(formatInt(v))
|
|
case uint:
|
|
w.AppendString(formatUint(uint64(v)))
|
|
case uint8:
|
|
w.AppendString(formatUint(uint64(v)))
|
|
case uint16:
|
|
w.AppendString(formatUint(uint64(v)))
|
|
case uint32:
|
|
w.AppendString(formatUint(uint64(v)))
|
|
case uint64:
|
|
w.AppendString(formatUint(v))
|
|
case float32:
|
|
w.AppendString(formatFloat(float64(v)))
|
|
case float64:
|
|
w.AppendString(formatFloat(v))
|
|
case bool:
|
|
if v {
|
|
w.AppendString("1")
|
|
} else {
|
|
w.AppendString("0")
|
|
}
|
|
default:
|
|
if bm, ok := val.(encoding.BinaryMarshaler); ok {
|
|
bb, err := bm.MarshalBinary()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
w.AppendBytes(bb)
|
|
} else {
|
|
return fmt.Errorf(
|
|
"redis: can't marshal %T (consider implementing encoding.BinaryMarshaler)", val)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (w *WriteBuffer) AppendString(s string) {
|
|
w.b = append(w.b, StringReply)
|
|
w.b = strconv.AppendUint(w.b, uint64(len(s)), 10)
|
|
w.b = append(w.b, '\r', '\n')
|
|
w.b = append(w.b, s...)
|
|
w.b = append(w.b, '\r', '\n')
|
|
}
|
|
|
|
func (w *WriteBuffer) AppendBytes(p []byte) {
|
|
w.b = append(w.b, StringReply)
|
|
w.b = strconv.AppendUint(w.b, uint64(len(p)), 10)
|
|
w.b = append(w.b, '\r', '\n')
|
|
w.b = append(w.b, p...)
|
|
w.b = append(w.b, '\r', '\n')
|
|
}
|