forked from mirror/redis
134 lines
2.4 KiB
Go
134 lines
2.4 KiB
Go
package proto
|
|
|
|
import (
|
|
"encoding"
|
|
"fmt"
|
|
"reflect"
|
|
|
|
"github.com/go-redis/redis/internal"
|
|
)
|
|
|
|
func Scan(b []byte, v interface{}) error {
|
|
switch v := v.(type) {
|
|
case nil:
|
|
return fmt.Errorf("redis: Scan(nil)")
|
|
case *string:
|
|
*v = internal.BytesToString(b)
|
|
return nil
|
|
case *[]byte:
|
|
*v = b
|
|
return nil
|
|
case *int:
|
|
var err error
|
|
*v, err = atoi(b)
|
|
return err
|
|
case *int8:
|
|
n, err := parseInt(b, 10, 8)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = int8(n)
|
|
return nil
|
|
case *int16:
|
|
n, err := parseInt(b, 10, 16)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = int16(n)
|
|
return nil
|
|
case *int32:
|
|
n, err := parseInt(b, 10, 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = int32(n)
|
|
return nil
|
|
case *int64:
|
|
n, err := parseInt(b, 10, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = n
|
|
return nil
|
|
case *uint:
|
|
n, err := parseUint(b, 10, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = uint(n)
|
|
return nil
|
|
case *uint8:
|
|
n, err := parseUint(b, 10, 8)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = uint8(n)
|
|
return nil
|
|
case *uint16:
|
|
n, err := parseUint(b, 10, 16)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = uint16(n)
|
|
return nil
|
|
case *uint32:
|
|
n, err := parseUint(b, 10, 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = uint32(n)
|
|
return nil
|
|
case *uint64:
|
|
n, err := parseUint(b, 10, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = n
|
|
return nil
|
|
case *float32:
|
|
n, err := parseFloat(b, 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = float32(n)
|
|
return err
|
|
case *float64:
|
|
var err error
|
|
*v, err = parseFloat(b, 64)
|
|
return err
|
|
case *bool:
|
|
*v = len(b) == 1 && b[0] == '1'
|
|
return nil
|
|
case encoding.BinaryUnmarshaler:
|
|
return v.UnmarshalBinary(b)
|
|
default:
|
|
return fmt.Errorf(
|
|
"redis: can't unmarshal %T (consider implementing BinaryUnmarshaler)", v)
|
|
}
|
|
}
|
|
|
|
func ScanSlice(data []string, slice interface{}) error {
|
|
v := reflect.ValueOf(slice)
|
|
if !v.IsValid() {
|
|
return fmt.Errorf("redis: ScanSlice(nil)")
|
|
}
|
|
if v.Kind() != reflect.Ptr {
|
|
return fmt.Errorf("redis: ScanSlice(non-pointer %T)", slice)
|
|
}
|
|
v = v.Elem()
|
|
if v.Kind() != reflect.Slice {
|
|
return fmt.Errorf("redis: ScanSlice(non-slice %T)", slice)
|
|
}
|
|
|
|
next := internal.MakeSliceNextElemFunc(v)
|
|
for i, s := range data {
|
|
elem := next()
|
|
if err := Scan([]byte(s), elem.Addr().Interface()); err != nil {
|
|
err = fmt.Errorf("redis: ScanSlice index=%d value=%q failed: %s", i, s, err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|