fix mssql test

This commit is contained in:
Lunny Xiao 2019-10-23 16:43:31 +08:00
parent 361409ba89
commit 0e27d27f9e
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
1 changed files with 5 additions and 2 deletions

View File

@ -378,19 +378,22 @@ func (session *Session) Update(bean interface{}, condiBean ...interface{}) (int6
} }
var tableAlias = session.engine.Quote(tableName) var tableAlias = session.engine.Quote(tableName)
var fromSQL string
if session.statement.TableAlias != "" { if session.statement.TableAlias != "" {
switch session.engine.dialect.DBType() { switch session.engine.dialect.DBType() {
case core.MSSQL: case core.MSSQL:
tableAlias = fmt.Sprintf("%s %s", tableAlias, session.statement.TableAlias) tableAlias = fmt.Sprintf("%s", tableAlias)
fromSQL = fmt.Sprintf("FROM %s %s ", session.statement.TableAlias, tableAlias)
default: default:
tableAlias = fmt.Sprintf("%s AS %s", tableAlias, session.statement.TableAlias) tableAlias = fmt.Sprintf("%s AS %s", tableAlias, session.statement.TableAlias)
} }
} }
sqlStr = fmt.Sprintf("UPDATE %v%v SET %v %v", sqlStr = fmt.Sprintf("UPDATE %v%v SET %v %v%v",
top, top,
tableAlias, tableAlias,
strings.Join(colNames, ", "), strings.Join(colNames, ", "),
fromSQL,
condSQL) condSQL)
res, err := session.exec(sqlStr, append(args, condArgs...)...) res, err := session.exec(sqlStr, append(args, condArgs...)...)