rsload/values.go

package main

import (
	"fmt"
	"strconv"
	"strings"
)

var (
	start_string     byte = '+'
	start_error      byte = '-'
	start_integer    byte = ':'
	start_bulkstring byte = '$'
	start_array      byte = '*'
)

type value interface {
}

func readValue(b []byte) (value, error) {
	if len(b) < 2 {
		return nil, fmt.Errorf("unable to read redis protocol value: input is too small")
	}
	switch b[0] {
	case start_string:
		return readString(b[1:])
	case start_error:
		return readError(b[1:])
	case start_integer:
		return readInteger(b[1:])
	default:
		return nil, fmt.Errorf("unable to read redis protocol value: illegal start character: %c", b[0])
	}
}

// ------------------------------------------------------------------------------

type String string

func readString(b []byte) (value, error) {
	return String(strings.Trim(string(b), "\r\n")), nil
}

// ------------------------------------------------------------------------------

type Error string

func readError(b []byte) (value, error) {
	return Error(strings.Trim(string(b), "\r\n")), nil
}

// ------------------------------------------------------------------------------

type Integer int64

func readInteger(b []byte) (value, error) {
	i, err := strconv.ParseInt(strings.Trim(string(b), "\r\n"), 10, 64)
	if err != nil {
		return nil, fmt.Errorf("unable to read integer in redis protocol format: %v", err)
	}
	return Integer(i), nil
}