mirror of
https://github.com/go-gitea/gitea.git
synced 2024-12-04 14:46:57 -05:00
bea343ce09
Co-authored-by: techknowlogick <techknowlogick@gitea.io>
158 lines
4.1 KiB
Go
Vendored
158 lines
4.1 KiB
Go
Vendored
// Copyright 2017 The Xorm Authors. All rights reserved.
|
||
// Use of this source code is governed by a BSD-style
|
||
// license that can be found in the LICENSE file.
|
||
|
||
package statements
|
||
|
||
import (
|
||
"database/sql"
|
||
"database/sql/driver"
|
||
"fmt"
|
||
"reflect"
|
||
"time"
|
||
|
||
"xorm.io/xorm/convert"
|
||
"xorm.io/xorm/dialects"
|
||
"xorm.io/xorm/internal/json"
|
||
"xorm.io/xorm/schemas"
|
||
)
|
||
|
||
var (
|
||
nullFloatType = reflect.TypeOf(sql.NullFloat64{})
|
||
)
|
||
|
||
// Value2Interface convert a field value of a struct to interface for puting into database
|
||
func (statement *Statement) Value2Interface(col *schemas.Column, fieldValue reflect.Value) (interface{}, error) {
|
||
if fieldValue.CanAddr() {
|
||
if fieldConvert, ok := fieldValue.Addr().Interface().(convert.Conversion); ok {
|
||
data, err := fieldConvert.ToDB()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if col.SQLType.IsBlob() {
|
||
return data, nil
|
||
}
|
||
return string(data), nil
|
||
}
|
||
}
|
||
|
||
isNil := fieldValue.Kind() == reflect.Ptr && fieldValue.IsNil()
|
||
if !isNil {
|
||
if fieldConvert, ok := fieldValue.Interface().(convert.Conversion); ok {
|
||
data, err := fieldConvert.ToDB()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if col.SQLType.IsBlob() {
|
||
return data, nil
|
||
}
|
||
if nil == data {
|
||
return nil, nil
|
||
}
|
||
return string(data), nil
|
||
}
|
||
}
|
||
|
||
fieldType := fieldValue.Type()
|
||
k := fieldType.Kind()
|
||
if k == reflect.Ptr {
|
||
if fieldValue.IsNil() {
|
||
return nil, nil
|
||
} else if !fieldValue.IsValid() {
|
||
return nil, nil
|
||
} else {
|
||
// !nashtsai! deference pointer type to instance type
|
||
fieldValue = fieldValue.Elem()
|
||
fieldType = fieldValue.Type()
|
||
k = fieldType.Kind()
|
||
}
|
||
}
|
||
|
||
switch k {
|
||
case reflect.Bool:
|
||
return fieldValue.Bool(), nil
|
||
case reflect.String:
|
||
return fieldValue.String(), nil
|
||
case reflect.Struct:
|
||
if fieldType.ConvertibleTo(schemas.TimeType) {
|
||
t := fieldValue.Convert(schemas.TimeType).Interface().(time.Time)
|
||
tf := dialects.FormatColumnTime(statement.dialect, statement.defaultTimeZone, col, t)
|
||
return tf, nil
|
||
} else if fieldType.ConvertibleTo(nullFloatType) {
|
||
t := fieldValue.Convert(nullFloatType).Interface().(sql.NullFloat64)
|
||
if !t.Valid {
|
||
return nil, nil
|
||
}
|
||
return t.Float64, nil
|
||
}
|
||
|
||
if !col.SQLType.IsJson() {
|
||
// !<winxxp>! 增加支持driver.Valuer接口的结构,如sql.NullString
|
||
if v, ok := fieldValue.Interface().(driver.Valuer); ok {
|
||
return v.Value()
|
||
}
|
||
|
||
fieldTable, err := statement.tagParser.ParseWithCache(fieldValue)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if len(fieldTable.PrimaryKeys) == 1 {
|
||
pkField := reflect.Indirect(fieldValue).FieldByName(fieldTable.PKColumns()[0].FieldName)
|
||
return pkField.Interface(), nil
|
||
}
|
||
return nil, fmt.Errorf("no primary key for col %v", col.Name)
|
||
}
|
||
|
||
if col.SQLType.IsText() {
|
||
bytes, err := json.DefaultJSONHandler.Marshal(fieldValue.Interface())
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return string(bytes), nil
|
||
} else if col.SQLType.IsBlob() {
|
||
bytes, err := json.DefaultJSONHandler.Marshal(fieldValue.Interface())
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return bytes, nil
|
||
}
|
||
return nil, fmt.Errorf("Unsupported type %v", fieldValue.Type())
|
||
case reflect.Complex64, reflect.Complex128:
|
||
bytes, err := json.DefaultJSONHandler.Marshal(fieldValue.Interface())
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return string(bytes), nil
|
||
case reflect.Array, reflect.Slice, reflect.Map:
|
||
if !fieldValue.IsValid() {
|
||
return fieldValue.Interface(), nil
|
||
}
|
||
|
||
if col.SQLType.IsText() {
|
||
bytes, err := json.DefaultJSONHandler.Marshal(fieldValue.Interface())
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return string(bytes), nil
|
||
} else if col.SQLType.IsBlob() {
|
||
var bytes []byte
|
||
var err error
|
||
if (k == reflect.Slice) &&
|
||
(fieldValue.Type().Elem().Kind() == reflect.Uint8) {
|
||
bytes = fieldValue.Bytes()
|
||
} else {
|
||
bytes, err = json.DefaultJSONHandler.Marshal(fieldValue.Interface())
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
}
|
||
return bytes, nil
|
||
}
|
||
return nil, ErrUnSupportedType
|
||
case reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uint:
|
||
return fieldValue.Uint(), nil
|
||
default:
|
||
return fieldValue.Interface(), nil
|
||
}
|
||
}
|