mirror of
https://gitee.com/johng/gf.git
synced 2024-12-02 04:07:47 +08:00
46 lines
1.3 KiB
Go
46 lines
1.3 KiB
Go
|
// Copyright 2017 gf Author(https://github.com/gogf/gf). All Rights Reserved.
|
||
|
|
//
|
||
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
||
|
|
// If a copy of the MIT was not distributed with this file,
|
||
|
|
// You can obtain one at https://github.com/gogf/gf.
|
||
|
|
|
||
|
|
package gdb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Update does "UPDATE ... " statement for the model.
|
||
|
|
//
|
||
|
|
// If the optional parameter <dataAndWhere> is given, the dataAndWhere[0] is the updated data field,
|
||
|
|
// and dataAndWhere[1:] is treated as where condition fields.
|
||
|
|
// Also see Model.Data and Model.Where functions.
|
||
|
|
func (m *Model) Update(dataAndWhere ...interface{}) (result sql.Result, err error) {
|
||
|
|
if len(dataAndWhere) > 0 {
|
||
|
|
if len(dataAndWhere) > 2 {
|
||
|
|
return m.Data(dataAndWhere[0]).Where(dataAndWhere[1], dataAndWhere[2:]...).Update()
|
||
|
|
} else if len(dataAndWhere) == 2 {
|
||
|
|
return m.Data(dataAndWhere[0]).Where(dataAndWhere[1]).Update()
|
||
|
|
} else {
|
||
|
|
return m.Data(dataAndWhere[0]).Update()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
defer func() {
|
||
|
|
if err == nil {
|
||
|
|
m.checkAndRemoveCache()
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
if m.data == nil {
|
||
|
|
return nil, errors.New("updating table with empty data")
|
||
|
|
}
|
||
|
|
condition, conditionArgs := m.formatCondition(false)
|
||
|
|
return m.db.DoUpdate(
|
||
|
|
m.getLink(true),
|
||
|
|
m.tables,
|
||
|
|
m.filterDataForInsertOrUpdate(m.data),
|
||
|
|
condition,
|
||
|
|
m.mergeArguments(conditionArgs)...,
|
||
|
|
)
|
||
|
|
}
|