forked from gitea/gitea
90 lines
2.4 KiB
Go
90 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
|
||
|
|
}
|