125 lines
2.0 KiB
Go
125 lines
2.0 KiB
Go
//
|
|
// scan_helper.go
|
|
// Copyright (C) 2023 tiglog <me@tiglog.com>
|
|
//
|
|
// Distributed under terms of the MIT license.
|
|
//
|
|
|
|
package helper
|
|
|
|
import (
|
|
"encoding"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
func Scan(b []byte, v interface{}) error {
|
|
switch v := v.(type) {
|
|
case nil:
|
|
return fmt.Errorf("golib helper: Scan(nil)")
|
|
case *string:
|
|
*v = 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 *time.Time:
|
|
var err error
|
|
*v, err = time.Parse(time.RFC3339Nano, BytesToString(b))
|
|
return err
|
|
case *time.Duration:
|
|
n, err := ParseInt(b, 10, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*v = time.Duration(n)
|
|
return nil
|
|
case encoding.BinaryUnmarshaler:
|
|
return v.UnmarshalBinary(b)
|
|
default:
|
|
return fmt.Errorf(
|
|
"golib: can't unmarshal %T (consider implementing BinaryUnmarshaler)", v)
|
|
}
|
|
}
|