103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
// Copyright 2018 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 (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"reflect"
|
|
"strings"
|
|
|
|
"github.com/go-xorm/core"
|
|
)
|
|
|
|
// TableNameWithSchema will automatically add schema prefix on table name
|
|
func (engine *Engine) TableNameWithSchema(v string) string {
|
|
// Add schema name as prefix of table name.
|
|
// Only for postgres database.
|
|
if engine.dialect.DBType() == core.POSTGRES &&
|
|
engine.dialect.URI().Schema != "" &&
|
|
engine.dialect.URI().Schema != postgresPublicSchema &&
|
|
strings.Index(v, ".") == -1 {
|
|
return engine.dialect.URI().Schema + "." + v
|
|
}
|
|
return v
|
|
}
|
|
|
|
func (engine *Engine) tbName(v reflect.Value) string {
|
|
return engine.TableNameWithSchema(engine.tbNameNoSchemaString(v.Interface()))
|
|
}
|
|
|
|
func (engine *Engine) tbNameForMap(v reflect.Value) string {
|
|
t := v.Type()
|
|
if tb, ok := v.Interface().(TableName); ok {
|
|
return tb.TableName()
|
|
}
|
|
if v.CanAddr() {
|
|
if tb, ok := v.Addr().Interface().(TableName); ok {
|
|
return tb.TableName()
|
|
}
|
|
}
|
|
return engine.TableMapper.Obj2Table(t.Name())
|
|
}
|
|
|
|
func (engine *Engine) tbNameNoSchema(w io.Writer, tablename interface{}) {
|
|
switch tablename.(type) {
|
|
case []string:
|
|
t := tablename.([]string)
|
|
if len(t) > 1 {
|
|
fmt.Fprintf(w, "%v AS %v", engine.Quote(t[0]), engine.Quote(t[1]))
|
|
} else if len(t) == 1 {
|
|
fmt.Fprintf(w, engine.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 TableName:
|
|
table = f.(TableName).TableName()
|
|
default:
|
|
v := rValue(f)
|
|
t := v.Type()
|
|
if t.Kind() == reflect.Struct {
|
|
fmt.Fprintf(w, engine.tbNameForMap(v))
|
|
} else {
|
|
fmt.Fprintf(w, engine.Quote(fmt.Sprintf("%v", f)))
|
|
}
|
|
}
|
|
}
|
|
if l > 1 {
|
|
fmt.Fprintf(w, "%v AS %v", engine.Quote(table),
|
|
engine.Quote(fmt.Sprintf("%v", t[1])))
|
|
} else if l == 1 {
|
|
fmt.Fprintf(w, engine.Quote(table))
|
|
}
|
|
case TableName:
|
|
fmt.Fprintf(w, tablename.(TableName).TableName())
|
|
case string:
|
|
fmt.Fprintf(w, tablename.(string))
|
|
default:
|
|
v := rValue(tablename)
|
|
t := v.Type()
|
|
if t.Kind() == reflect.Struct {
|
|
fmt.Fprintf(w, engine.tbNameForMap(v))
|
|
} else {
|
|
fmt.Fprintf(w, engine.Quote(fmt.Sprintf("%v", tablename)))
|
|
}
|
|
}
|
|
}
|
|
|
|
func (engine *Engine) tbNameNoSchemaString(tablename interface{}) string {
|
|
var buf bytes.Buffer
|
|
engine.tbNameNoSchema(&buf, tablename)
|
|
return buf.String()
|
|
}
|