redis/internal/proto/proto.go

111 lines
1.9 KiB
Go
Raw Normal View History

2016-07-02 15:52:10 +03:00
package proto
import (
"encoding"
"fmt"
"strconv"
"gopkg.in/redis.v5/internal"
2016-07-02 15:52:10 +03:00
)
2016-11-20 10:50:49 +03:00
func Scan(s string, val interface{}) error {
2016-07-02 15:52:10 +03:00
switch v := val.(type) {
case nil:
2016-11-20 10:50:49 +03:00
return internal.RedisError("redis: Scan(nil)")
2016-07-02 15:52:10 +03:00
case *string:
2016-11-20 10:50:49 +03:00
*v = s
2016-07-02 15:52:10 +03:00
return nil
case *[]byte:
2016-11-20 10:50:49 +03:00
*v = []byte(s)
2016-07-02 15:52:10 +03:00
return nil
case *int:
var err error
2016-11-20 10:50:49 +03:00
*v, err = strconv.Atoi(s)
2016-07-02 15:52:10 +03:00
return err
case *int8:
2016-11-20 10:50:49 +03:00
n, err := strconv.ParseInt(s, 10, 8)
2016-07-02 15:52:10 +03:00
if err != nil {
return err
}
*v = int8(n)
return nil
case *int16:
2016-11-20 10:50:49 +03:00
n, err := strconv.ParseInt(s, 10, 16)
2016-07-02 15:52:10 +03:00
if err != nil {
return err
}
*v = int16(n)
return nil
case *int32:
2016-11-20 10:50:49 +03:00
n, err := strconv.ParseInt(s, 10, 32)
2016-07-02 15:52:10 +03:00
if err != nil {
return err
}
*v = int32(n)
return nil
case *int64:
2016-11-20 10:50:49 +03:00
n, err := strconv.ParseInt(s, 10, 64)
2016-07-02 15:52:10 +03:00
if err != nil {
return err
}
*v = n
return nil
case *uint:
2016-11-20 10:50:49 +03:00
n, err := strconv.ParseUint(s, 10, 64)
2016-07-02 15:52:10 +03:00
if err != nil {
return err
}
*v = uint(n)
return nil
case *uint8:
2016-11-20 10:50:49 +03:00
n, err := strconv.ParseUint(s, 10, 8)
2016-07-02 15:52:10 +03:00
if err != nil {
return err
}
*v = uint8(n)
return nil
case *uint16:
2016-11-20 10:50:49 +03:00
n, err := strconv.ParseUint(s, 10, 16)
2016-07-02 15:52:10 +03:00
if err != nil {
return err
}
*v = uint16(n)
return nil
case *uint32:
2016-11-20 10:50:49 +03:00
n, err := strconv.ParseUint(s, 10, 32)
2016-07-02 15:52:10 +03:00
if err != nil {
return err
}
*v = uint32(n)
return nil
case *uint64:
2016-11-20 10:50:49 +03:00
n, err := strconv.ParseUint(s, 10, 64)
2016-07-02 15:52:10 +03:00
if err != nil {
return err
}
*v = n
return nil
case *float32:
2016-11-20 10:50:49 +03:00
n, err := strconv.ParseFloat(s, 32)
2016-07-02 15:52:10 +03:00
if err != nil {
return err
}
*v = float32(n)
return err
case *float64:
var err error
2016-11-20 10:50:49 +03:00
*v, err = strconv.ParseFloat(s, 64)
2016-07-02 15:52:10 +03:00
return err
case *bool:
2016-11-20 10:50:49 +03:00
*v = len(s) == 1 && s[0] == '1'
2016-07-02 15:52:10 +03:00
return nil
default:
if bu, ok := val.(encoding.BinaryUnmarshaler); ok {
2016-11-20 10:50:49 +03:00
return bu.UnmarshalBinary([]byte(s))
2016-07-02 15:52:10 +03:00
}
err := fmt.Errorf(
"redis: can't unmarshal %T (consider implementing BinaryUnmarshaler)", val)
return err
}
}