gorm/field.go

67 lines
1.5 KiB
Go
Raw Normal View History

2013-11-14 17:26:02 +04:00
package gorm
import (
"database/sql"
"database/sql/driver"
"errors"
"fmt"
2013-11-15 07:36:27 +04:00
"reflect"
2013-11-14 17:26:02 +04:00
)
2016-03-07 09:54:20 +03:00
// Field model field definition
2013-11-14 17:26:02 +04:00
type Field struct {
2015-02-15 18:01:09 +03:00
*StructField
2015-02-16 07:04:46 +03:00
IsBlank bool
Field reflect.Value
2014-01-26 15:34:06 +04:00
}
2014-09-02 15:03:01 +04:00
2016-03-07 09:54:20 +03:00
// Set set a value to the field
2016-01-16 06:37:16 +03:00
func (field *Field) Set(value interface{}) (err error) {
if !field.Field.IsValid() {
return errors.New("field value not valid")
2014-09-02 15:03:01 +04:00
}
if !field.Field.CanAddr() {
return ErrUnaddressable
}
2016-01-04 16:49:04 +03:00
reflectValue, ok := value.(reflect.Value)
if !ok {
reflectValue = reflect.ValueOf(value)
2014-09-02 15:03:01 +04:00
}
2016-01-16 06:37:16 +03:00
fieldValue := field.Field
2016-01-04 16:49:04 +03:00
if reflectValue.IsValid() {
2016-01-16 06:37:16 +03:00
if reflectValue.Type().ConvertibleTo(fieldValue.Type()) {
fieldValue.Set(reflectValue.Convert(fieldValue.Type()))
2015-02-17 17:55:14 +03:00
} else {
2016-01-16 06:37:16 +03:00
if fieldValue.Kind() == reflect.Ptr {
if fieldValue.IsNil() {
fieldValue.Set(reflect.New(field.Struct.Type.Elem()))
}
fieldValue = fieldValue.Elem()
}
if reflectValue.Type().ConvertibleTo(fieldValue.Type()) {
fieldValue.Set(reflectValue.Convert(fieldValue.Type()))
} else if scanner, ok := fieldValue.Addr().Interface().(sql.Scanner); ok {
v := reflectValue.Interface()
if valuer, ok := v.(driver.Valuer); ok {
if v, err = valuer.Value(); err == nil {
err = scanner.Scan(v)
}
} else {
err = scanner.Scan(v)
}
2016-01-16 06:37:16 +03:00
} else {
err = fmt.Errorf("could not convert argument of field %s from %s to %s", field.Name, reflectValue.Type(), fieldValue.Type())
}
2015-02-17 17:55:14 +03:00
}
2016-01-04 16:49:04 +03:00
} else {
field.Field.Set(reflect.Zero(field.Field.Type()))
}
field.IsBlank = isBlank(field.Field)
return err
2014-09-02 15:03:01 +04:00
}