bug fixed

Signed-off-by: 商讯在线 <swhbox@foxmail.com>
This commit is contained in:
商讯在线 2014-04-13 12:22:32 +08:00
parent 7826f88534
commit ad33a0be49
2 changed files with 6 additions and 1 deletions

View File

@ -457,7 +457,9 @@ func (engine *Engine) mapType(v reflect.Value) *core.Table {
table := engine.newTable() table := engine.newTable()
method := v.MethodByName("TableName") method := v.MethodByName("TableName")
if !method.IsValid() { if !method.IsValid() {
method = v.Addr().MethodByName("TableName") if v.CanAddr() {
method = v.Addr().MethodByName("TableName")
}
} }
if method.IsValid() { if method.IsValid() {
params := []reflect.Value{} params := []reflect.Value{}

View File

@ -539,6 +539,9 @@ func (statement *Statement) Cols(columns ...string) *Statement {
statement.columnMap[strings.ToLower(nc)] = true statement.columnMap[strings.ToLower(nc)] = true
} }
statement.ColumnStr = statement.Engine.Quote(strings.Join(newColumns, statement.Engine.Quote(", "))) statement.ColumnStr = statement.Engine.Quote(strings.Join(newColumns, statement.Engine.Quote(", ")))
if strings.Contains(statement.ColumnStr, ".") {
statement.ColumnStr = strings.Replace(statement.ColumnStr, ".", statement.Engine.Quote("."), -1)
}
return statement return statement
} }