d28f005552
Fix limit for databases other than sqlite go mod tidy && go mod vendor Remove unneeded break statements Make everything work with the new xorm version Fix xorm logging Fix lint Fix redis init Fix using id field Fix database init for testing Change default database log level Add xorm logger Use const for postgres go mod tidy Merge branch 'master' into update/xorm # Conflicts: # go.mod # go.sum # vendor/modules.txt go mod vendor Fix loading fixtures for postgres Go mod vendor1 Update xorm to version 1 Co-authored-by: kolaente <k@knt.li> Reviewed-on: https://kolaente.dev/vikunja/api/pulls/323
358 lines
8.5 KiB
Go
358 lines
8.5 KiB
Go
// Copyright 2016 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"
|
|
"errors"
|
|
"fmt"
|
|
"reflect"
|
|
"strconv"
|
|
|
|
"xorm.io/xorm/caches"
|
|
"xorm.io/xorm/internal/utils"
|
|
"xorm.io/xorm/schemas"
|
|
)
|
|
|
|
// Get retrieve one record from database, bean's non-empty fields
|
|
// will be as conditions
|
|
func (session *Session) Get(bean interface{}) (bool, error) {
|
|
if session.isAutoClose {
|
|
defer session.Close()
|
|
}
|
|
return session.get(bean)
|
|
}
|
|
|
|
func (session *Session) get(bean interface{}) (bool, error) {
|
|
defer session.resetStatement()
|
|
|
|
if session.statement.LastError != nil {
|
|
return false, session.statement.LastError
|
|
}
|
|
|
|
beanValue := reflect.ValueOf(bean)
|
|
if beanValue.Kind() != reflect.Ptr {
|
|
return false, errors.New("needs a pointer to a value")
|
|
} else if beanValue.Elem().Kind() == reflect.Ptr {
|
|
return false, errors.New("a pointer to a pointer is not allowed")
|
|
}
|
|
|
|
if beanValue.Elem().Kind() == reflect.Struct {
|
|
if err := session.statement.SetRefBean(bean); err != nil {
|
|
return false, err
|
|
}
|
|
}
|
|
|
|
var sqlStr string
|
|
var args []interface{}
|
|
var err error
|
|
|
|
if session.statement.RawSQL == "" {
|
|
if len(session.statement.TableName()) <= 0 {
|
|
return false, ErrTableNotFound
|
|
}
|
|
session.statement.Limit(1)
|
|
sqlStr, args, err = session.statement.GenGetSQL(bean)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
} else {
|
|
sqlStr = session.statement.GenRawSQL()
|
|
args = session.statement.RawParams
|
|
}
|
|
|
|
table := session.statement.RefTable
|
|
|
|
if session.statement.ColumnMap.IsEmpty() && session.canCache() && beanValue.Elem().Kind() == reflect.Struct {
|
|
if cacher := session.engine.GetCacher(session.statement.TableName()); cacher != nil &&
|
|
!session.statement.GetUnscoped() {
|
|
has, err := session.cacheGet(bean, sqlStr, args...)
|
|
if err != ErrCacheFailed {
|
|
return has, err
|
|
}
|
|
}
|
|
}
|
|
|
|
context := session.statement.Context
|
|
if context != nil {
|
|
res := context.Get(fmt.Sprintf("%v-%v", sqlStr, args))
|
|
if res != nil {
|
|
session.engine.logger.Debugf("hit context cache: %s", sqlStr)
|
|
|
|
structValue := reflect.Indirect(reflect.ValueOf(bean))
|
|
structValue.Set(reflect.Indirect(reflect.ValueOf(res)))
|
|
session.lastSQL = ""
|
|
session.lastSQLArgs = nil
|
|
return true, nil
|
|
}
|
|
}
|
|
|
|
has, err := session.nocacheGet(beanValue.Elem().Kind(), table, bean, sqlStr, args...)
|
|
if err != nil || !has {
|
|
return has, err
|
|
}
|
|
|
|
if context != nil {
|
|
context.Put(fmt.Sprintf("%v-%v", sqlStr, args), bean)
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
func (session *Session) nocacheGet(beanKind reflect.Kind, table *schemas.Table, bean interface{}, sqlStr string, args ...interface{}) (bool, error) {
|
|
rows, err := session.queryRows(sqlStr, args...)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
if !rows.Next() {
|
|
if rows.Err() != nil {
|
|
return false, rows.Err()
|
|
}
|
|
return false, nil
|
|
}
|
|
|
|
switch bean.(type) {
|
|
case sql.NullInt64, sql.NullBool, sql.NullFloat64, sql.NullString:
|
|
return true, rows.Scan(&bean)
|
|
case *sql.NullInt64, *sql.NullBool, *sql.NullFloat64, *sql.NullString:
|
|
return true, rows.Scan(bean)
|
|
case *string:
|
|
var res sql.NullString
|
|
if err := rows.Scan(&res); err != nil {
|
|
return true, err
|
|
}
|
|
if res.Valid {
|
|
*(bean.(*string)) = res.String
|
|
}
|
|
return true, nil
|
|
case *int:
|
|
var res sql.NullInt64
|
|
if err := rows.Scan(&res); err != nil {
|
|
return true, err
|
|
}
|
|
if res.Valid {
|
|
*(bean.(*int)) = int(res.Int64)
|
|
}
|
|
return true, nil
|
|
case *int8:
|
|
var res sql.NullInt64
|
|
if err := rows.Scan(&res); err != nil {
|
|
return true, err
|
|
}
|
|
if res.Valid {
|
|
*(bean.(*int8)) = int8(res.Int64)
|
|
}
|
|
return true, nil
|
|
case *int16:
|
|
var res sql.NullInt64
|
|
if err := rows.Scan(&res); err != nil {
|
|
return true, err
|
|
}
|
|
if res.Valid {
|
|
*(bean.(*int16)) = int16(res.Int64)
|
|
}
|
|
return true, nil
|
|
case *int32:
|
|
var res sql.NullInt64
|
|
if err := rows.Scan(&res); err != nil {
|
|
return true, err
|
|
}
|
|
if res.Valid {
|
|
*(bean.(*int32)) = int32(res.Int64)
|
|
}
|
|
return true, nil
|
|
case *int64:
|
|
var res sql.NullInt64
|
|
if err := rows.Scan(&res); err != nil {
|
|
return true, err
|
|
}
|
|
if res.Valid {
|
|
*(bean.(*int64)) = int64(res.Int64)
|
|
}
|
|
return true, nil
|
|
case *uint:
|
|
var res sql.NullInt64
|
|
if err := rows.Scan(&res); err != nil {
|
|
return true, err
|
|
}
|
|
if res.Valid {
|
|
*(bean.(*uint)) = uint(res.Int64)
|
|
}
|
|
return true, nil
|
|
case *uint8:
|
|
var res sql.NullInt64
|
|
if err := rows.Scan(&res); err != nil {
|
|
return true, err
|
|
}
|
|
if res.Valid {
|
|
*(bean.(*uint8)) = uint8(res.Int64)
|
|
}
|
|
return true, nil
|
|
case *uint16:
|
|
var res sql.NullInt64
|
|
if err := rows.Scan(&res); err != nil {
|
|
return true, err
|
|
}
|
|
if res.Valid {
|
|
*(bean.(*uint16)) = uint16(res.Int64)
|
|
}
|
|
return true, nil
|
|
case *uint32:
|
|
var res sql.NullInt64
|
|
if err := rows.Scan(&res); err != nil {
|
|
return true, err
|
|
}
|
|
if res.Valid {
|
|
*(bean.(*uint32)) = uint32(res.Int64)
|
|
}
|
|
return true, nil
|
|
case *uint64:
|
|
var res sql.NullInt64
|
|
if err := rows.Scan(&res); err != nil {
|
|
return true, err
|
|
}
|
|
if res.Valid {
|
|
*(bean.(*uint64)) = uint64(res.Int64)
|
|
}
|
|
return true, nil
|
|
case *bool:
|
|
var res sql.NullBool
|
|
if err := rows.Scan(&res); err != nil {
|
|
return true, err
|
|
}
|
|
if res.Valid {
|
|
*(bean.(*bool)) = res.Bool
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
switch beanKind {
|
|
case reflect.Struct:
|
|
fields, err := rows.Columns()
|
|
if err != nil {
|
|
// WARN: Alougth rows return true, but get fields failed
|
|
return true, err
|
|
}
|
|
|
|
scanResults, err := session.row2Slice(rows, fields, bean)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
// close it before covert data
|
|
rows.Close()
|
|
|
|
dataStruct := utils.ReflectValue(bean)
|
|
_, err = session.slice2Bean(scanResults, fields, bean, &dataStruct, table)
|
|
if err != nil {
|
|
return true, err
|
|
}
|
|
|
|
return true, session.executeProcessors()
|
|
case reflect.Slice:
|
|
err = rows.ScanSlice(bean)
|
|
case reflect.Map:
|
|
err = rows.ScanMap(bean)
|
|
case reflect.String, reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64,
|
|
reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
err = rows.Scan(bean)
|
|
default:
|
|
err = rows.Scan(bean)
|
|
}
|
|
|
|
return true, err
|
|
}
|
|
|
|
func (session *Session) cacheGet(bean interface{}, sqlStr string, args ...interface{}) (has bool, err error) {
|
|
// if has no reftable, then don't use cache currently
|
|
if !session.canCache() {
|
|
return false, ErrCacheFailed
|
|
}
|
|
|
|
for _, filter := range session.engine.dialect.Filters() {
|
|
sqlStr = filter.Do(sqlStr)
|
|
}
|
|
newsql := session.statement.ConvertIDSQL(sqlStr)
|
|
if newsql == "" {
|
|
return false, ErrCacheFailed
|
|
}
|
|
|
|
tableName := session.statement.TableName()
|
|
cacher := session.engine.cacherMgr.GetCacher(tableName)
|
|
|
|
session.engine.logger.Debugf("[cache] Get SQL: %s, %v", newsql, args)
|
|
table := session.statement.RefTable
|
|
ids, err := caches.GetCacheSql(cacher, tableName, newsql, args)
|
|
if err != nil {
|
|
var res = make([]string, len(table.PrimaryKeys))
|
|
rows, err := session.NoCache().queryRows(newsql, args...)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
if rows.Next() {
|
|
err = rows.ScanSlice(&res)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
} else {
|
|
return false, ErrCacheFailed
|
|
}
|
|
|
|
var pk schemas.PK = make([]interface{}, len(table.PrimaryKeys))
|
|
for i, col := range table.PKColumns() {
|
|
if col.SQLType.IsText() {
|
|
pk[i] = res[i]
|
|
} else if col.SQLType.IsNumeric() {
|
|
n, err := strconv.ParseInt(res[i], 10, 64)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
pk[i] = n
|
|
} else {
|
|
return false, errors.New("unsupported")
|
|
}
|
|
}
|
|
|
|
ids = []schemas.PK{pk}
|
|
session.engine.logger.Debugf("[cache] cache ids: %s, %v", newsql, ids)
|
|
err = caches.PutCacheSql(cacher, ids, tableName, newsql, args)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
} else {
|
|
session.engine.logger.Debugf("[cache] cache hit: %s, %v", newsql, ids)
|
|
}
|
|
|
|
if len(ids) > 0 {
|
|
structValue := reflect.Indirect(reflect.ValueOf(bean))
|
|
id := ids[0]
|
|
session.engine.logger.Debugf("[cache] get bean: %s, %v", tableName, id)
|
|
sid, err := id.ToString()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
cacheBean := cacher.GetBean(tableName, sid)
|
|
if cacheBean == nil {
|
|
cacheBean = bean
|
|
has, err = session.nocacheGet(reflect.Struct, table, cacheBean, sqlStr, args...)
|
|
if err != nil || !has {
|
|
return has, err
|
|
}
|
|
|
|
session.engine.logger.Debugf("[cache] cache bean: %s, %v, %v", tableName, id, cacheBean)
|
|
cacher.PutBean(tableName, sid, cacheBean)
|
|
} else {
|
|
session.engine.logger.Debugf("[cache] cache hit: %s, %v, %v", tableName, id, cacheBean)
|
|
has = true
|
|
}
|
|
structValue.Set(reflect.Indirect(reflect.ValueOf(cacheBean)))
|
|
|
|
return has, nil
|
|
}
|
|
return false, nil
|
|
}
|