forked from mirror/redis
156 lines
2.8 KiB
Go
156 lines
2.8 KiB
Go
package proto
|
|
|
|
import (
|
|
"encoding"
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/go-redis/redis/v8/internal/util"
|
|
)
|
|
|
|
type writer interface {
|
|
io.Writer
|
|
io.ByteWriter
|
|
// WriteString implement io.StringWriter.
|
|
WriteString(s string) (n int, err error)
|
|
}
|
|
|
|
type Writer struct {
|
|
writer
|
|
|
|
lenBuf []byte
|
|
numBuf []byte
|
|
}
|
|
|
|
func NewWriter(wr writer) *Writer {
|
|
return &Writer{
|
|
writer: wr,
|
|
|
|
lenBuf: make([]byte, 64),
|
|
numBuf: make([]byte, 64),
|
|
}
|
|
}
|
|
|
|
func (w *Writer) WriteArgs(args []interface{}) error {
|
|
if err := w.WriteByte(RespArray); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := w.writeLen(len(args)); err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, arg := range args {
|
|
if err := w.WriteArg(arg); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (w *Writer) writeLen(n int) error {
|
|
w.lenBuf = strconv.AppendUint(w.lenBuf[:0], uint64(n), 10)
|
|
w.lenBuf = append(w.lenBuf, '\r', '\n')
|
|
_, err := w.Write(w.lenBuf)
|
|
return err
|
|
}
|
|
|
|
func (w *Writer) WriteArg(v interface{}) error {
|
|
switch v := v.(type) {
|
|
case nil:
|
|
return w.string("")
|
|
case string:
|
|
return w.string(v)
|
|
case []byte:
|
|
return w.bytes(v)
|
|
case int:
|
|
return w.int(int64(v))
|
|
case int8:
|
|
return w.int(int64(v))
|
|
case int16:
|
|
return w.int(int64(v))
|
|
case int32:
|
|
return w.int(int64(v))
|
|
case int64:
|
|
return w.int(v)
|
|
case uint:
|
|
return w.uint(uint64(v))
|
|
case uint8:
|
|
return w.uint(uint64(v))
|
|
case uint16:
|
|
return w.uint(uint64(v))
|
|
case uint32:
|
|
return w.uint(uint64(v))
|
|
case uint64:
|
|
return w.uint(v)
|
|
case float32:
|
|
return w.float(float64(v))
|
|
case float64:
|
|
return w.float(v)
|
|
case bool:
|
|
if v {
|
|
return w.int(1)
|
|
}
|
|
return w.int(0)
|
|
case time.Time:
|
|
w.numBuf = v.AppendFormat(w.numBuf[:0], time.RFC3339Nano)
|
|
return w.bytes(w.numBuf)
|
|
case time.Duration:
|
|
return w.int(v.Nanoseconds())
|
|
case encoding.BinaryMarshaler:
|
|
b, err := v.MarshalBinary()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return w.bytes(b)
|
|
default:
|
|
return fmt.Errorf(
|
|
"redis: can't marshal %T (implement encoding.BinaryMarshaler)", v)
|
|
}
|
|
}
|
|
|
|
func (w *Writer) bytes(b []byte) error {
|
|
if err := w.WriteByte(RespString); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := w.writeLen(len(b)); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := w.Write(b); err != nil {
|
|
return err
|
|
}
|
|
|
|
return w.crlf()
|
|
}
|
|
|
|
func (w *Writer) string(s string) error {
|
|
return w.bytes(util.StringToBytes(s))
|
|
}
|
|
|
|
func (w *Writer) uint(n uint64) error {
|
|
w.numBuf = strconv.AppendUint(w.numBuf[:0], n, 10)
|
|
return w.bytes(w.numBuf)
|
|
}
|
|
|
|
func (w *Writer) int(n int64) error {
|
|
w.numBuf = strconv.AppendInt(w.numBuf[:0], n, 10)
|
|
return w.bytes(w.numBuf)
|
|
}
|
|
|
|
func (w *Writer) float(f float64) error {
|
|
w.numBuf = strconv.AppendFloat(w.numBuf[:0], f, 'f', -1, 64)
|
|
return w.bytes(w.numBuf)
|
|
}
|
|
|
|
func (w *Writer) crlf() error {
|
|
if err := w.WriteByte('\r'); err != nil {
|
|
return err
|
|
}
|
|
return w.WriteByte('\n')
|
|
}
|