HiPoweredDev
/
xorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use alias name for field qualifiers when available

This commit is contained in:
Victor Gaydov 2016-05-12 23:29:12 +03:00
parent ee6e17b4e3
commit 7aed2a253b
1 changed files with 20 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
statement.go
View File

@ -55,7 +55,7 @@ type Statement struct {
ColumnStr string ColumnStr string
selectStr string selectStr string
columnMap map[string]bool columnMap map[string]bool
tableMap map[string]bool tableMap map[string]string
useAllCols bool useAllCols bool
OmitStr string OmitStr string
ConditionStr string ConditionStr string
@ -101,7 +101,7 @@ func (statement *Statement) Init() {
statement.ColumnStr = "" statement.ColumnStr = ""
statement.OmitStr = "" statement.OmitStr = ""
statement.columnMap = make(map[string]bool) statement.columnMap = make(map[string]bool)
statement.tableMap = make(map[string]bool) statement.tableMap = make(map[string]string)
statement.ConditionStr = "" statement.ConditionStr = ""
statement.AltTableName = "" statement.AltTableName = ""
statement.IdParam = nil statement.IdParam = nil
@ -201,7 +201,7 @@ func (statement *Statement) Or(querystring string, args ...interface{}) *Stateme
// Table tempororily set table name, the parameter could be a string or a pointer of struct // Table tempororily set table name, the parameter could be a string or a pointer of struct
func (statement *Statement) Table(tableNameOrBean interface{}) *Statement { func (statement *Statement) Table(tableNameOrBean interface{}) *Statement {
if statement.TableName() != "" { if statement.TableAlias == "" && statement.TableName() != "" {
statement.tableMapDelete(statement.TableName()) statement.tableMapDelete(statement.TableName())
} }
v := rValue(tableNameOrBean) v := rValue(tableNameOrBean)
@ -211,7 +211,9 @@ func (statement *Statement) Table(tableNameOrBean interface{}) *Statement {
} else if t.Kind() == reflect.Struct { } else if t.Kind() == reflect.Struct {
statement.RefTable = statement.Engine.autoMapType(v) statement.RefTable = statement.Engine.autoMapType(v)
} }
if statement.TableAlias == "" {
statement.tableMapAdd(statement.TableName()) statement.tableMapAdd(statement.TableName())
}
return statement return statement
} }
@ -456,7 +458,7 @@ func (statement *Statement) needTableName() bool {
func (statement *Statement) tableMapAdd(table string) { func (statement *Statement) tableMapAdd(table string) {
tableName := statement.Engine.Quote(strings.ToLower(table)) tableName := statement.Engine.Quote(strings.ToLower(table))
statement.tableMap[tableName] = true statement.tableMap[tableName] = table
} }
func (statement *Statement) tableMapDelete(table string) { func (statement *Statement) tableMapDelete(table string) {
@ -464,7 +466,7 @@ func (statement *Statement) tableMapDelete(table string) {
delete(statement.tableMap, tableName) delete(statement.tableMap, tableName)
} }
func (statement *Statement) isKnownTable(table string) bool { func (statement *Statement) isKnownTable(table string) (string, bool) {
if len(table) > 0 { if len(table) > 0 {
var mainTable string var mainTable string
@ -477,21 +479,25 @@ func (statement *Statement) isKnownTable(table string) bool {
cm := statement.Engine.Quote(strings.ToLower(mainTable)) cm := statement.Engine.Quote(strings.ToLower(mainTable))
ct := statement.Engine.Quote(strings.ToLower(table)) ct := statement.Engine.Quote(strings.ToLower(table))
if ct == cm || statement.tableMap[ct] { if name, ok := statement.tableMap[ct]; ok {
return true return name, true
}
if ct == cm {
return mainTable, true
} }
} }
return false return "", false
} }
func (statement *Statement) colName(col *core.Column) string { func (statement *Statement) colName(col *core.Column) string {
var colTable string var colTable string
if statement.needTableName() { if statement.needTableName() {
if statement.isKnownTable(col.TableName) { if name, ok := statement.isKnownTable(col.TableName); ok {
colTable = col.TableName colTable = name
} else if statement.isKnownTable(statement.outTableName()) { } else if name, ok := statement.isKnownTable(statement.outTableName()); ok {
colTable = statement.outTableName() colTable = name
} else { } else {
colTable = "" colTable = ""
} }
@ -1074,7 +1080,7 @@ func (statement *Statement) Join(joinOP string, tablename interface{}, condition
fmt.Fprintf(&buf, " ON %v", condition) fmt.Fprintf(&buf, " ON %v", condition)
statement.JoinStr = buf.String() statement.JoinStr = buf.String()
statement.joinArgs = append(statement.joinArgs, args...) statement.joinArgs = append(statement.joinArgs, args...)
statement.tableMap[statement.Engine.Quote(strings.ToLower(refName))] = true statement.tableMapAdd(refName)
return statement return statement
} }