310 lines
8.3 KiB
Go
310 lines
8.3 KiB
Go
package boil
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"math"
|
|
"math/rand"
|
|
"reflect"
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/pobri19/sqlboiler/strmangle"
|
|
"gopkg.in/BlackBaronsTux/null-extended.v1"
|
|
)
|
|
|
|
var (
|
|
typeNullFloat32 = reflect.TypeOf(null.Float32{})
|
|
typeNullFloat64 = reflect.TypeOf(null.Float64{})
|
|
typeNullInt = reflect.TypeOf(null.Int{})
|
|
typeNullInt8 = reflect.TypeOf(null.Int8{})
|
|
typeNullInt16 = reflect.TypeOf(null.Int16{})
|
|
typeNullInt32 = reflect.TypeOf(null.Int32{})
|
|
typeNullInt64 = reflect.TypeOf(null.Int64{})
|
|
typeNullUint = reflect.TypeOf(null.Uint{})
|
|
typeNullUint8 = reflect.TypeOf(null.Uint8{})
|
|
typeNullUint16 = reflect.TypeOf(null.Uint16{})
|
|
typeNullUint32 = reflect.TypeOf(null.Uint32{})
|
|
typeNullUint64 = reflect.TypeOf(null.Uint64{})
|
|
typeNullString = reflect.TypeOf(null.String{})
|
|
typeNullBool = reflect.TypeOf(null.Bool{})
|
|
typeNullTime = reflect.TypeOf(null.Time{})
|
|
typeTime = reflect.TypeOf(time.Time{})
|
|
)
|
|
|
|
// Bind executes the query and inserts the
|
|
// result into the passed in object pointer
|
|
func (q *Query) Bind(obj interface{}) error {
|
|
return nil
|
|
}
|
|
|
|
// BindOne inserts the returned row columns into the
|
|
// passed in object pointer
|
|
func BindOne(row *sql.Row, selectCols []string, obj interface{}) error {
|
|
kind := reflect.ValueOf(obj).Kind()
|
|
if kind != reflect.Ptr {
|
|
return fmt.Errorf("BindOne given a non-pointer type")
|
|
}
|
|
|
|
pointers := GetStructPointers(obj, selectCols...)
|
|
if err := row.Scan(pointers...); err != nil {
|
|
return fmt.Errorf("Unable to scan into pointers: %s", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// BindAll inserts the returned rows columns into the
|
|
// passed in slice of object pointers
|
|
func BindAll(rows *sql.Rows, selectCols []string, obj interface{}) error {
|
|
val := reflect.ValueOf(obj)
|
|
typ := reflect.TypeOf(obj)
|
|
kind := val.Kind()
|
|
|
|
if kind != reflect.Slice {
|
|
return fmt.Errorf("BindAll given a non-slice type")
|
|
}
|
|
|
|
spare := reflect.New(typ.Elem().Elem())
|
|
fmt.Printf("%T, %#v, %s\n", spare, spare, spare.Type().String())
|
|
|
|
index := 0
|
|
for rows.Next() {
|
|
val = reflect.Append(val, spare)
|
|
pointers := GetStructPointers(val.Index(index), selectCols...)
|
|
if err := rows.Scan(pointers...); err != nil {
|
|
return fmt.Errorf("Unable to scan into pointers: %s", err)
|
|
}
|
|
index++
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func checkType(obj interface{}) (reflect.Type, bool, error) {
|
|
val := reflect.ValueOf(obj)
|
|
typ := val.Type()
|
|
kind := val.Kind()
|
|
|
|
if kind != reflect.Ptr {
|
|
return nil, false, fmt.Errorf("Bind must be given pointers to structs but got type: %s, kind: %s", typ.String(), kind)
|
|
}
|
|
|
|
typ = typ.Elem()
|
|
kind = typ.Kind()
|
|
isSlice := false
|
|
|
|
switch kind {
|
|
case reflect.Slice:
|
|
typ = typ.Elem()
|
|
kind = typ.Kind()
|
|
isSlice = true
|
|
case reflect.Struct:
|
|
return typ, isSlice, nil
|
|
default:
|
|
return nil, false, fmt.Errorf("Bind was given an invalid object must be []*T or *T but got type: %s, kind: %s", typ.String(), kind)
|
|
}
|
|
|
|
if kind != reflect.Ptr {
|
|
return nil, false, fmt.Errorf("Bind must be given pointers to structs but got type: %s, kind: %s", typ.String(), kind)
|
|
}
|
|
|
|
typ = typ.Elem()
|
|
kind = typ.Kind()
|
|
|
|
if kind != reflect.Struct {
|
|
return nil, false, fmt.Errorf("Bind must be a struct but got type: %s, kind: %s", typ.String(), kind)
|
|
}
|
|
|
|
return typ, isSlice, nil
|
|
}
|
|
|
|
// GetStructValues returns the values (as interface) of the matching columns in obj
|
|
func GetStructValues(obj interface{}, columns ...string) []interface{} {
|
|
ret := make([]interface{}, len(columns))
|
|
val := reflect.Indirect(reflect.ValueOf(obj))
|
|
|
|
for i, c := range columns {
|
|
field := val.FieldByName(strmangle.TitleCase(c))
|
|
ret[i] = field.Interface()
|
|
}
|
|
|
|
return ret
|
|
}
|
|
|
|
// GetStructPointers returns a slice of pointers to the matching columns in obj
|
|
func GetStructPointers(obj interface{}, columns ...string) []interface{} {
|
|
val := reflect.ValueOf(obj).Elem()
|
|
ret := make([]interface{}, len(columns))
|
|
|
|
for i, c := range columns {
|
|
field := val.FieldByName(strmangle.TitleCase(c))
|
|
if !field.IsValid() {
|
|
panic(fmt.Sprintf("Could not find field on struct %T for field %s", obj, strmangle.TitleCase(c)))
|
|
}
|
|
|
|
field = field.Addr()
|
|
ret[i] = field.Interface()
|
|
}
|
|
|
|
return ret
|
|
}
|
|
|
|
// RandomizeStruct takes an object and fills it with random data.
|
|
// It will ignore the fields in the blacklist.
|
|
func RandomizeStruct(str interface{}, blacklist ...string) error {
|
|
// Don't modify blacklist
|
|
copyBlacklist := make([]string, len(blacklist))
|
|
copy(copyBlacklist, blacklist)
|
|
blacklist = copyBlacklist
|
|
|
|
sort.Strings(blacklist)
|
|
|
|
// Check if it's pointer
|
|
value := reflect.ValueOf(str)
|
|
kind := value.Kind()
|
|
if kind != reflect.Ptr {
|
|
return fmt.Errorf("can only randomize pointers to structs, given: %T", str)
|
|
}
|
|
|
|
// Check if it's a struct
|
|
value = value.Elem()
|
|
kind = value.Kind()
|
|
if kind != reflect.Struct {
|
|
return fmt.Errorf("can only randomize pointers to structs, given: %T", str)
|
|
}
|
|
|
|
typ := value.Type()
|
|
nFields := value.NumField()
|
|
|
|
// Iterate through fields, randomizing
|
|
for i := 0; i < nFields; i++ {
|
|
fieldVal := value.Field(i)
|
|
fieldTyp := typ.Field(i)
|
|
|
|
found := sort.Search(len(blacklist), func(i int) bool {
|
|
return blacklist[i] == fieldTyp.Name
|
|
})
|
|
if found != len(blacklist) {
|
|
continue
|
|
}
|
|
|
|
if err := randomizeField(fieldVal); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func randomizeField(field reflect.Value) error {
|
|
kind := field.Kind()
|
|
typ := field.Type()
|
|
|
|
var newVal interface{}
|
|
|
|
if kind == reflect.Struct {
|
|
switch typ {
|
|
case typeNullBool:
|
|
newVal = null.NewBool(rand.Intn(2) == 1, rand.Intn(2) == 1)
|
|
case typeNullString:
|
|
newVal = null.NewString(randStr(5+rand.Intn(25)), rand.Intn(2) == 1)
|
|
case typeNullTime:
|
|
newVal = null.NewTime(time.Now().Add(time.Duration(rand.Intn((int(time.Hour * 24 * 10))))), rand.Intn(2) == 1)
|
|
case typeTime:
|
|
newVal = time.Now().Add(time.Duration(rand.Intn((int(time.Hour * 24 * 10)))))
|
|
case typeNullFloat32:
|
|
newVal = null.NewFloat32(rand.Float32(), rand.Intn(2) == 1)
|
|
case typeNullFloat64:
|
|
newVal = null.NewFloat64(rand.Float64(), rand.Intn(2) == 1)
|
|
case typeNullInt:
|
|
newVal = null.NewInt(rand.Int(), rand.Intn(2) == 1)
|
|
case typeNullInt8:
|
|
newVal = null.NewInt8(int8(rand.Intn(int(math.MaxInt8))), rand.Intn(2) == 1)
|
|
case typeNullInt16:
|
|
newVal = null.NewInt16(int16(rand.Intn(int(math.MaxInt16))), rand.Intn(2) == 1)
|
|
case typeNullInt32:
|
|
newVal = null.NewInt32(rand.Int31(), rand.Intn(2) == 1)
|
|
case typeNullInt64:
|
|
newVal = null.NewInt64(rand.Int63(), rand.Intn(2) == 1)
|
|
case typeNullUint:
|
|
newVal = null.NewUint(uint(rand.Int()), rand.Intn(2) == 1)
|
|
case typeNullUint8:
|
|
newVal = null.NewUint8(uint8(rand.Intn(int(math.MaxInt8))), rand.Intn(2) == 1)
|
|
case typeNullUint16:
|
|
newVal = null.NewUint16(uint16(rand.Intn(int(math.MaxInt16))), rand.Intn(2) == 1)
|
|
case typeNullUint32:
|
|
newVal = null.NewUint32(uint32(rand.Int31()), rand.Intn(2) == 1)
|
|
case typeNullUint64:
|
|
newVal = null.NewUint64(uint64(rand.Int63()), rand.Intn(2) == 1)
|
|
}
|
|
} else {
|
|
switch kind {
|
|
case reflect.Float32:
|
|
newVal = rand.Float32()
|
|
case reflect.Float64:
|
|
newVal = rand.Float64()
|
|
case reflect.Int:
|
|
newVal = rand.Int()
|
|
case reflect.Int8:
|
|
newVal = int8(rand.Intn(int(math.MaxInt8)))
|
|
case reflect.Int16:
|
|
newVal = int16(rand.Intn(int(math.MaxInt16)))
|
|
case reflect.Int32:
|
|
newVal = rand.Int31()
|
|
case reflect.Int64:
|
|
newVal = rand.Int63()
|
|
case reflect.Uint:
|
|
newVal = uint(rand.Int())
|
|
case reflect.Uint8:
|
|
newVal = uint8(rand.Intn(int(math.MaxInt8)))
|
|
case reflect.Uint16:
|
|
newVal = uint16(rand.Intn(int(math.MaxInt16)))
|
|
case reflect.Uint32:
|
|
newVal = uint32(rand.Int31())
|
|
case reflect.Uint64:
|
|
newVal = uint64(rand.Int63())
|
|
case reflect.Bool:
|
|
var b bool
|
|
if rand.Intn(2) == 1 {
|
|
b = true
|
|
}
|
|
newVal = b
|
|
case reflect.String:
|
|
newVal = randStr(5 + rand.Intn(20))
|
|
case reflect.Slice:
|
|
sliceVal := typ.Elem()
|
|
if sliceVal.Kind() != reflect.Uint8 {
|
|
return fmt.Errorf("unsupported slice type: %T", typ.String())
|
|
}
|
|
newVal = randByteSlice(5 + rand.Intn(20))
|
|
default:
|
|
return fmt.Errorf("unsupported type: %T", typ.String())
|
|
}
|
|
}
|
|
|
|
field.Set(reflect.ValueOf(newVal))
|
|
|
|
return nil
|
|
}
|
|
|
|
const alphabet = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
|
|
func randStr(ln int) string {
|
|
str := make([]byte, ln)
|
|
for i := 0; i < ln; i++ {
|
|
str[i] = byte(alphabet[rand.Intn(len(alphabet))])
|
|
}
|
|
|
|
return string(str)
|
|
}
|
|
|
|
func randByteSlice(ln int) []byte {
|
|
str := make([]byte, ln)
|
|
for i := 0; i < ln; i++ {
|
|
str[i] = byte(rand.Intn(256))
|
|
}
|
|
|
|
return str
|
|
}
|