Improve get bean

This commit is contained in:
Lunny Xiao 2021-06-11 13:35:50 +08:00
parent 611e378b7b
commit 7b84aa150b
6 changed files with 53 additions and 74 deletions

View File

@ -462,7 +462,6 @@ func convertAssign(dest, src interface{}, originalLocation *time.Location, conve
} }
var sv reflect.Value var sv reflect.Value
switch d := dest.(type) { switch d := dest.(type) {
case *string: case *string:
sv = reflect.ValueOf(src) sv = reflect.ValueOf(src)
@ -496,7 +495,7 @@ func convertAssign(dest, src interface{}, originalLocation *time.Location, conve
func convertAssignV(dpv reflect.Value, src interface{}, originalLocation, convertedLocation *time.Location) error { func convertAssignV(dpv reflect.Value, src interface{}, originalLocation, convertedLocation *time.Location) error {
if dpv.Kind() != reflect.Ptr { if dpv.Kind() != reflect.Ptr {
return errors.New("destination not a pointer") return fmt.Errorf("destination %s not a pointer", dpv.Kind())
} }
if dpv.IsNil() { if dpv.IsNil() {
return errNilPtr return errNilPtr

38
convert/string.go Normal file
View File

@ -0,0 +1,38 @@
// Copyright 2021 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 convert
import (
"database/sql"
"fmt"
"strconv"
)
func ConvertAssignString(v interface{}) (string, error) {
switch vv := v.(type) {
case *sql.NullString:
if vv.Valid {
return vv.String, nil
}
return "", nil
case *int64:
if vv != nil {
return strconv.FormatInt(*vv, 10), nil
}
return "", nil
case *int8:
if vv != nil {
return strconv.FormatInt(int64(*vv), 10), nil
}
return "", nil
case *sql.RawBytes:
if vv != nil && len([]byte(*vv)) > 0 {
return string(*vv), nil
}
return "", nil
default:
return "", fmt.Errorf("unsupported type: %#v", vv)
}
}

View File

@ -671,6 +671,12 @@ func (p *mysqlDriver) Parse(driverName, dataSourceName string) (*URI, error) {
return uri, nil return uri, nil
} }
func (b *mysqlDriver) Features() DriverFeatures {
return DriverFeatures{
SupportNullable: false,
}
}
func (p *mysqlDriver) GenScanResult(colType string) (interface{}, error) { func (p *mysqlDriver) GenScanResult(colType string) (interface{}, error) {
switch colType { switch colType {
case "CHAR", "VARCHAR", "TINYTEXT", "TEXT", "MEDIUMTEXT", "LONGTEXT", "ENUM", "SET": case "CHAR", "VARCHAR", "TINYTEXT", "TEXT", "MEDIUMTEXT", "LONGTEXT", "ENUM", "SET":
@ -698,6 +704,7 @@ func (p *mysqlDriver) GenScanResult(colType string) (interface{}, error) {
var r sql.RawBytes var r sql.RawBytes
return &r, nil return &r, nil
default: default:
fmt.Printf("unknow mysql database type: %v\n", colType)
var r sql.RawBytes var r sql.RawBytes
return &r, nil return &r, nil
} }

View File

@ -551,6 +551,12 @@ func (p *sqlite3Driver) Parse(driverName, dataSourceName string) (*URI, error) {
return &URI{DBType: schemas.SQLITE, DBName: dataSourceName}, nil return &URI{DBType: schemas.SQLITE, DBName: dataSourceName}, nil
} }
func (b *sqlite3Driver) Features() DriverFeatures {
return DriverFeatures{
SupportNullable: false,
}
}
func (p *sqlite3Driver) GenScanResult(colType string) (interface{}, error) { func (p *sqlite3Driver) GenScanResult(colType string) (interface{}, error) {
switch colType { switch colType {
case "TEXT": case "TEXT":

View File

@ -82,12 +82,7 @@ func newEngine(driverName, dataSourceName string, dialect dialects.Dialect, db *
dataSourceName: dataSourceName, dataSourceName: dataSourceName,
db: db, db: db,
logSessionID: false, logSessionID: false,
} DatabaseTZ: time.Local,
if dialect.URI().DBType == schemas.SQLITE {
engine.DatabaseTZ = time.UTC
} else {
engine.DatabaseTZ = time.Local
} }
logger := log.NewSimpleLogger(os.Stdout) logger := log.NewSimpleLogger(os.Stdout)

View File

@ -5,19 +5,8 @@
package xorm package xorm
import ( import (
<<<<<<< HEAD
=======
<<<<<<< HEAD
=======
"database/sql" "database/sql"
"errors"
>>>>>>> 6e19325 (refactor driver)
"fmt"
"reflect"
"strconv"
"time"
>>>>>>> 634f82a (refactor driver)
"xorm.io/xorm/core" "xorm.io/xorm/core"
) )
@ -35,55 +24,6 @@ func (session *Session) Query(sqlOrArgs ...interface{}) ([]map[string][]byte, er
return session.queryBytes(sqlStr, args...) return session.queryBytes(sqlStr, args...)
} }
<<<<<<< HEAD
=======
func value2String(rawValue *reflect.Value) (str string, err error) {
aa := reflect.TypeOf((*rawValue).Interface())
vv := reflect.ValueOf((*rawValue).Interface())
switch aa.Kind() {
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
str = strconv.FormatInt(vv.Int(), 10)
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
str = strconv.FormatUint(vv.Uint(), 10)
case reflect.Float32, reflect.Float64:
str = strconv.FormatFloat(vv.Float(), 'f', -1, 64)
case reflect.String:
str = vv.String()
case reflect.Array, reflect.Slice:
switch aa.Elem().Kind() {
case reflect.Uint8:
data := rawValue.Interface().([]byte)
str = string(data)
if str == "\x00" {
str = "0"
}
default:
err = fmt.Errorf("Unsupported struct type %v", vv.Type().Name())
}
// time type
case reflect.Struct:
if aa.ConvertibleTo(schemas.TimeType) {
str = vv.Convert(schemas.TimeType).Interface().(time.Time).Format(time.RFC3339Nano)
} else {
err = fmt.Errorf("Unsupported struct type %v", vv.Type().Name())
}
case reflect.Bool:
str = strconv.FormatBool(vv.Bool())
case reflect.Complex128, reflect.Complex64:
str = fmt.Sprintf("%v", vv.Complex())
/* TODO: unsupported types below
case reflect.Map:
case reflect.Ptr:
case reflect.Uintptr:
case reflect.UnsafePointer:
case reflect.Chan, reflect.Func, reflect.Interface:
*/
default:
err = fmt.Errorf("Unsupported struct type %v", vv.Type().Name())
}
return
}
// genRowsScanResults according // genRowsScanResults according
func (session *Session) genRowsScanResults(rows *core.Rows, types []*sql.ColumnType) ([]interface{}, error) { func (session *Session) genRowsScanResults(rows *core.Rows, types []*sql.ColumnType) ([]interface{}, error) {
var scanResults = make([]interface{}, len(types)) var scanResults = make([]interface{}, len(types))
@ -97,7 +37,6 @@ func (session *Session) genRowsScanResults(rows *core.Rows, types []*sql.ColumnT
return scanResults, nil return scanResults, nil
} }
>>>>>>> 634f82a (refactor driver)
func (session *Session) rows2Strings(rows *core.Rows) (resultsSlice []map[string]string, err error) { func (session *Session) rows2Strings(rows *core.Rows) (resultsSlice []map[string]string, err error) {
fields, err := rows.Columns() fields, err := rows.Columns()
if err != nil { if err != nil {
@ -107,11 +46,6 @@ func (session *Session) rows2Strings(rows *core.Rows) (resultsSlice []map[string
if err != nil { if err != nil {
return nil, err return nil, err
} }
types, err := rows.ColumnTypes()
if err != nil {
return nil, err
}
for rows.Next() { for rows.Next() {
result, err := row2mapStr(rows, types, fields) result, err := row2mapStr(rows, types, fields)
if err != nil { if err != nil {