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
89 lines
2.4 KiB
Go
89 lines
2.4 KiB
Go
// Copyright 2015 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 dialects
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"strings"
|
|
|
|
"xorm.io/xorm/internal/utils"
|
|
"xorm.io/xorm/names"
|
|
)
|
|
|
|
// TableNameWithSchema will add schema prefix on table name if possible
|
|
func TableNameWithSchema(dialect Dialect, tableName string) string {
|
|
// Add schema name as prefix of table name.
|
|
// Only for postgres database.
|
|
if dialect.URI().Schema != "" &&
|
|
strings.Index(tableName, ".") == -1 {
|
|
return fmt.Sprintf("%s.%s", dialect.URI().Schema, tableName)
|
|
}
|
|
return tableName
|
|
}
|
|
|
|
// TableNameNoSchema returns table name with given tableName
|
|
func TableNameNoSchema(dialect Dialect, mapper names.Mapper, tableName interface{}) string {
|
|
quote := dialect.Quoter().Quote
|
|
switch tableName.(type) {
|
|
case []string:
|
|
t := tableName.([]string)
|
|
if len(t) > 1 {
|
|
return fmt.Sprintf("%v AS %v", quote(t[0]), quote(t[1]))
|
|
} else if len(t) == 1 {
|
|
return quote(t[0])
|
|
}
|
|
case []interface{}:
|
|
t := tableName.([]interface{})
|
|
l := len(t)
|
|
var table string
|
|
if l > 0 {
|
|
f := t[0]
|
|
switch f.(type) {
|
|
case string:
|
|
table = f.(string)
|
|
case names.TableName:
|
|
table = f.(names.TableName).TableName()
|
|
default:
|
|
v := utils.ReflectValue(f)
|
|
t := v.Type()
|
|
if t.Kind() == reflect.Struct {
|
|
table = names.GetTableName(mapper, v)
|
|
} else {
|
|
table = quote(fmt.Sprintf("%v", f))
|
|
}
|
|
}
|
|
}
|
|
if l > 1 {
|
|
return fmt.Sprintf("%v AS %v", quote(table), quote(fmt.Sprintf("%v", t[1])))
|
|
} else if l == 1 {
|
|
return quote(table)
|
|
}
|
|
case names.TableName:
|
|
return tableName.(names.TableName).TableName()
|
|
case string:
|
|
return tableName.(string)
|
|
case reflect.Value:
|
|
v := tableName.(reflect.Value)
|
|
return names.GetTableName(mapper, v)
|
|
default:
|
|
v := utils.ReflectValue(tableName)
|
|
t := v.Type()
|
|
if t.Kind() == reflect.Struct {
|
|
return names.GetTableName(mapper, v)
|
|
}
|
|
return quote(fmt.Sprintf("%v", tableName))
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// FullTableName returns table name with quote and schema according parameter
|
|
func FullTableName(dialect Dialect, mapper names.Mapper, bean interface{}, includeSchema ...bool) string {
|
|
tbName := TableNameNoSchema(dialect, mapper, bean)
|
|
if len(includeSchema) > 0 && includeSchema[0] && !utils.IsSubQuery(tbName) {
|
|
tbName = TableNameWithSchema(dialect, tbName)
|
|
}
|
|
return tbName
|
|
}
|