do not parse input, just read all
This commit is contained in:
65
src/adapt.go
65
src/adapt.go
@@ -65,90 +65,85 @@ func adapt(ctx context.Context, config Config, conn net.Conn) error {
|
|||||||
return io.EOF
|
return io.EOF
|
||||||
}
|
}
|
||||||
|
|
||||||
func readMessage(reader *bufio.Reader) ([]byte, []any, error) {
|
func readMessage(reader *bufio.Reader) ([]byte, error) {
|
||||||
w := bytes.NewBuffer(nil)
|
w := bytes.NewBuffer(nil)
|
||||||
arr, err := readMessageTo(w, reader)
|
err := readMessageTo(w, reader)
|
||||||
return w.Bytes(), arr, err
|
return w.Bytes(), err
|
||||||
}
|
}
|
||||||
|
|
||||||
func readMessageTo(w io.Writer, reader *bufio.Reader) ([]any, error) {
|
func readMessageTo(w io.Writer, reader *bufio.Reader) error {
|
||||||
w2 := bufio.NewWriter(w)
|
w2 := bufio.NewWriter(w)
|
||||||
defer w2.Flush()
|
defer w2.Flush()
|
||||||
arr, err := _readMessageTo(w2, reader)
|
err := _readMessageTo(w2, reader)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return err
|
||||||
}
|
}
|
||||||
return arr, w2.Flush()
|
return w2.Flush()
|
||||||
}
|
}
|
||||||
|
|
||||||
func _readMessageTo(w io.Writer, reader *bufio.Reader) ([]any, error) {
|
func _readMessageTo(w io.Writer, reader *bufio.Reader) error {
|
||||||
firstLine, _, err := reader.ReadLine()
|
firstLine, _, err := reader.ReadLine()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return err
|
||||||
}
|
}
|
||||||
firstLine = bytes.TrimSuffix(firstLine, []byte("\r\n"))
|
firstLine = bytes.TrimSuffix(firstLine, []byte("\r\n"))
|
||||||
if len(firstLine) == 0 {
|
if len(firstLine) == 0 {
|
||||||
return nil, nil
|
return nil
|
||||||
}
|
}
|
||||||
fmt.Fprintf(w, "%s\r\n", firstLine)
|
fmt.Fprintf(w, "%s\r\n", firstLine)
|
||||||
|
|
||||||
switch firstLine[0] {
|
switch firstLine[0] {
|
||||||
case '+': // simple string, like +OK
|
case '+': // simple string, like +OK
|
||||||
return []any{string(firstLine[1:])}, nil
|
return nil
|
||||||
case '-': // simple error, like -message
|
case '-': // simple error, like -message
|
||||||
return []any{string(firstLine)}, nil
|
return nil
|
||||||
case ':': // number, like /[+-][0-9]+/
|
case ':': // number, like /[+-][0-9]+/
|
||||||
firstLine = bytes.TrimPrefix(firstLine[1:], []byte("+"))
|
return err
|
||||||
n, err := strconv.Atoi(string(firstLine))
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("num not a num: %q: %w", firstLine, err)
|
|
||||||
}
|
|
||||||
return []any{n}, err
|
|
||||||
case '$': // long string, like $-1 for nil, like $LEN\r\nSTRING\r\n
|
case '$': // long string, like $-1 for nil, like $LEN\r\nSTRING\r\n
|
||||||
if firstLine[1] == '-' {
|
if firstLine[1] == '-' {
|
||||||
return []any{nil}, nil
|
return nil
|
||||||
}
|
}
|
||||||
firstLine = bytes.TrimPrefix(firstLine[1:], []byte("+"))
|
firstLine = bytes.TrimPrefix(firstLine[1:], []byte("+"))
|
||||||
n, err := strconv.Atoi(string(firstLine))
|
n, err := strconv.Atoi(string(firstLine))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("num not a num: %q: %w", firstLine, err)
|
return fmt.Errorf("num not a num: %q: %w", firstLine, err)
|
||||||
}
|
}
|
||||||
//log.Printf("reading %v+2 bytes for bulk string", n)
|
|
||||||
nextLine := make([]byte, n+2)
|
nextLine := make([]byte, n+2)
|
||||||
nAt := 0
|
nAt := 0
|
||||||
for nAt < n+2 {
|
for nAt < n+2 {
|
||||||
nMore, err := reader.Read(nextLine[nAt:])
|
nMore, err := reader.Read(nextLine[nAt:])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("couldnt read %v more/%v bytes for long string: %w", n+2-nAt, n+2, err)
|
return fmt.Errorf("couldnt read %v more/%v bytes for long string: %w", n+2-nAt, n+2, err)
|
||||||
}
|
}
|
||||||
nAt += nMore
|
nAt += nMore
|
||||||
}
|
}
|
||||||
fmt.Fprintf(w, "%s", nextLine)
|
fmt.Fprintf(w, "%s", nextLine)
|
||||||
return []any{string(nextLine)}, err
|
return err
|
||||||
case '*': // array, like *-1 for nil, like *4 for [1,2,3,4]
|
case '*', '%': // *=array %=map, like *-1 for nil, like *4 for [1,2,3,4]
|
||||||
n, err := strconv.Atoi(string(firstLine[1:]))
|
n, err := strconv.Atoi(string(firstLine[1:]))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("arr not a num: %q: %w", firstLine, err)
|
return fmt.Errorf("arr not a num: %q: %w", firstLine, err)
|
||||||
} else if n == -1 {
|
} else if n == -1 {
|
||||||
return nil, nil
|
return nil
|
||||||
|
}
|
||||||
|
if firstLine[0] == '%' {
|
||||||
|
n *= 2
|
||||||
}
|
}
|
||||||
var result []any
|
|
||||||
for i := 0; i < n; i++ {
|
for i := 0; i < n; i++ {
|
||||||
more, err := _readMessageTo(w, reader)
|
err := _readMessageTo(w, reader)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return err
|
||||||
}
|
}
|
||||||
result = append(result, more...)
|
|
||||||
}
|
}
|
||||||
return result, nil
|
return nil
|
||||||
case '_': // nil
|
case '_': // nil
|
||||||
return nil, nil
|
return nil
|
||||||
case '#': // boolean, like #t or #f
|
case '#': // boolean, like #t or #f
|
||||||
return []any{firstLine[1] == 't'}, nil
|
return nil
|
||||||
case ',': // double
|
case ',': // double
|
||||||
log.Fatal("not impl")
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
log.Fatalf("not impl: %q", firstLine)
|
log.Fatalf("not impl: %q", firstLine)
|
||||||
return nil, nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -73,7 +73,7 @@ func NewConfig(ctx context.Context) (Config, error) {
|
|||||||
v.Close()
|
v.Close()
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
if raw, _, err := readMessage(bufio.NewReader(v)); err != nil {
|
if raw, err := readMessage(bufio.NewReader(v)); err != nil {
|
||||||
log.Printf("! failed read hello %q: %v", forward, err)
|
log.Printf("! failed read hello %q: %v", forward, err)
|
||||||
v.Close()
|
v.Close()
|
||||||
return nil
|
return nil
|
||||||
|
|||||||
Reference in New Issue
Block a user