mirror of
https://github.com/go-gitea/gitea
synced 2024-10-31 15:24:25 +00:00
ac97ea573c
* Update go-redis to v8.4.0 * github.com/go-redis/redis/v8 v8.4.0 -> v8.5.0 * Apply suggestions from code review Co-authored-by: zeripath <art27@cantab.net> * TODO * Use the Queue termination channel as the default context for pushes Signed-off-by: Andrew Thornton <art27@cantab.net> * missed one Signed-off-by: Andrew Thornton <art27@cantab.net> Co-authored-by: zeripath <art27@cantab.net>
174 lines
3.3 KiB
Go
Vendored
174 lines
3.3 KiB
Go
Vendored
package proto
|
|
|
|
import (
|
|
"encoding"
|
|
"fmt"
|
|
"reflect"
|
|
"time"
|
|
|
|
"github.com/go-redis/redis/v8/internal/util"
|
|
)
|
|
|
|
// Scan parses bytes `b` to `v` with appropriate type.
|
|
// nolint: gocyclo
|
|
func Scan(b []byte, v interface{}) error {
|
|
switch v := v.(type) {
|
|
case nil:
|
|
return fmt.Errorf("redis: Scan(nil)")
|
|
case *string:
|
|
*v = util.BytesToString(b)
|
|
return nil
|
|
case *[]byte:
|
|
*v = b
|
|
return nil
|
|
case *int:
|
|
var err error
|
|
*v, err = util.Atoi(b)
|
|
return err
|
|
case *int8:
|
|
n, err := util.ParseInt(b, 10, 8)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = int8(n)
|
|
return nil
|
|
case *int16:
|
|
n, err := util.ParseInt(b, 10, 16)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = int16(n)
|
|
return nil
|
|
case *int32:
|
|
n, err := util.ParseInt(b, 10, 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = int32(n)
|
|
return nil
|
|
case *int64:
|
|
n, err := util.ParseInt(b, 10, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = n
|
|
return nil
|
|
case *uint:
|
|
n, err := util.ParseUint(b, 10, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = uint(n)
|
|
return nil
|
|
case *uint8:
|
|
n, err := util.ParseUint(b, 10, 8)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = uint8(n)
|
|
return nil
|
|
case *uint16:
|
|
n, err := util.ParseUint(b, 10, 16)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = uint16(n)
|
|
return nil
|
|
case *uint32:
|
|
n, err := util.ParseUint(b, 10, 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = uint32(n)
|
|
return nil
|
|
case *uint64:
|
|
n, err := util.ParseUint(b, 10, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = n
|
|
return nil
|
|
case *float32:
|
|
n, err := util.ParseFloat(b, 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = float32(n)
|
|
return err
|
|
case *float64:
|
|
var err error
|
|
*v, err = util.ParseFloat(b, 64)
|
|
return err
|
|
case *bool:
|
|
*v = len(b) == 1 && b[0] == '1'
|
|
return nil
|
|
case *time.Time:
|
|
var err error
|
|
*v, err = time.Parse(time.RFC3339Nano, util.BytesToString(b))
|
|
return err
|
|
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 := 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: %w", i, s, err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func makeSliceNextElemFunc(v reflect.Value) func() reflect.Value {
|
|
elemType := v.Type().Elem()
|
|
|
|
if elemType.Kind() == reflect.Ptr {
|
|
elemType = elemType.Elem()
|
|
return func() reflect.Value {
|
|
if v.Len() < v.Cap() {
|
|
v.Set(v.Slice(0, v.Len()+1))
|
|
elem := v.Index(v.Len() - 1)
|
|
if elem.IsNil() {
|
|
elem.Set(reflect.New(elemType))
|
|
}
|
|
return elem.Elem()
|
|
}
|
|
|
|
elem := reflect.New(elemType)
|
|
v.Set(reflect.Append(v, elem))
|
|
return elem.Elem()
|
|
}
|
|
}
|
|
|
|
zero := reflect.Zero(elemType)
|
|
return func() reflect.Value {
|
|
if v.Len() < v.Cap() {
|
|
v.Set(v.Slice(0, v.Len()+1))
|
|
return v.Index(v.Len() - 1)
|
|
}
|
|
|
|
v.Set(reflect.Append(v, zero))
|
|
return v.Index(v.Len() - 1)
|
|
}
|
|
}
|