117 lines
2.9 KiB
Go
117 lines
2.9 KiB
Go
// 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 xorm
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
"xorm.io/xorm/convert"
|
|
"xorm.io/xorm/core"
|
|
"xorm.io/xorm/dialects"
|
|
)
|
|
|
|
func row2mapStr(rows *core.Rows, types []*sql.ColumnType, fields []string) (map[string]string, error) {
|
|
var scanResults = make([]interface{}, len(fields))
|
|
for i := 0; i < len(fields); i++ {
|
|
var s sql.NullString
|
|
scanResults[i] = &s
|
|
}
|
|
|
|
if err := rows.Scan(scanResults...); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result := make(map[string]string, len(fields))
|
|
for ii, key := range fields {
|
|
s := scanResults[ii].(*sql.NullString)
|
|
result[key] = s.String
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func row2mapBytes(rows *core.Rows, types []*sql.ColumnType, fields []string) (map[string][]byte, error) {
|
|
var scanResults = make([]interface{}, len(fields))
|
|
for i := 0; i < len(fields); i++ {
|
|
var s sql.NullString
|
|
scanResults[i] = &s
|
|
}
|
|
|
|
if err := rows.Scan(scanResults...); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result := make(map[string][]byte, len(fields))
|
|
for ii, key := range fields {
|
|
s := scanResults[ii].(*sql.NullString)
|
|
result[key] = []byte(s.String)
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func row2sliceStr(rows *core.Rows, types []*sql.ColumnType, fields []string) ([]string, error) {
|
|
results := make([]string, 0, len(fields))
|
|
var scanResults = make([]interface{}, len(fields))
|
|
for i := 0; i < len(fields); i++ {
|
|
var s sql.NullString
|
|
scanResults[i] = &s
|
|
}
|
|
|
|
if err := rows.Scan(scanResults...); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for i := 0; i < len(fields); i++ {
|
|
results = append(results, scanResults[i].(*sql.NullString).String)
|
|
}
|
|
return results, nil
|
|
}
|
|
|
|
func rows2maps(rows *core.Rows) (resultsSlice []map[string][]byte, err error) {
|
|
fields, err := rows.Columns()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
types, err := rows.ColumnTypes()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for rows.Next() {
|
|
result, err := row2mapBytes(rows, types, fields)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
resultsSlice = append(resultsSlice, result)
|
|
}
|
|
|
|
return resultsSlice, nil
|
|
}
|
|
|
|
func (engine *Engine) row2mapInterface(rows *core.Rows, types []*sql.ColumnType, fields []string) (map[string]interface{}, error) {
|
|
var resultsMap = make(map[string]interface{}, len(fields))
|
|
var scanResultContainers = make([]interface{}, len(fields))
|
|
for i := 0; i < len(fields); i++ {
|
|
scanResult, err := engine.driver.GenScanResult(types[i].DatabaseTypeName())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
scanResultContainers[i] = scanResult
|
|
}
|
|
if err := engine.driver.Scan(&dialects.ScanContext{
|
|
DBLocation: engine.DatabaseTZ,
|
|
UserLocation: engine.TZLocation,
|
|
}, rows, types, scanResultContainers...); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for ii, key := range fields {
|
|
res, err := convert.Interface2Interface(engine.TZLocation, scanResultContainers[ii])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
resultsMap[key] = res
|
|
}
|
|
return resultsMap, nil
|
|
}
|