81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
package mysql
|
|
|
|
import "github.com/Gleipnir-Technology/jet/internal/jet"
|
|
|
|
// UpdateStatement is interface of SQL UPDATE statement
|
|
type UpdateStatement interface {
|
|
jet.Statement
|
|
|
|
OPTIMIZER_HINTS(hints ...OptimizerHint) UpdateStatement
|
|
|
|
SET(value interface{}, values ...interface{}) UpdateStatement
|
|
MODEL(data interface{}) UpdateStatement
|
|
|
|
WHERE(expression BoolExpression) UpdateStatement
|
|
LIMIT(limit int64) UpdateStatement
|
|
}
|
|
|
|
type updateStatementImpl struct {
|
|
jet.SerializerStatement
|
|
|
|
Update jet.ClauseUpdate
|
|
Set jet.SetClause
|
|
SetNew jet.SetClauseNew
|
|
Where jet.ClauseWhere
|
|
Limit jet.ClauseLimit
|
|
}
|
|
|
|
func newUpdateStatement(table Table, columns []jet.Column) UpdateStatement {
|
|
update := &updateStatementImpl{}
|
|
update.SerializerStatement = jet.NewStatementImpl(Dialect, jet.UpdateStatementType, update,
|
|
&update.Update,
|
|
&update.Set,
|
|
&update.SetNew,
|
|
&update.Where,
|
|
&update.Limit)
|
|
|
|
update.Update.Table = table
|
|
update.Set.Columns = columns
|
|
update.Where.Mandatory = true
|
|
update.Limit.Count = -1 // Initialize to -1 to indicate no LIMIT
|
|
|
|
return update
|
|
}
|
|
|
|
func (u *updateStatementImpl) OPTIMIZER_HINTS(hints ...OptimizerHint) UpdateStatement {
|
|
u.Update.OptimizerHints = hints
|
|
return u
|
|
}
|
|
|
|
func (u *updateStatementImpl) SET(value interface{}, values ...interface{}) UpdateStatement {
|
|
columnAssigment, isColumnAssigment := value.(ColumnAssigment)
|
|
|
|
if isColumnAssigment {
|
|
u.SetNew = []ColumnAssigment{columnAssigment}
|
|
for _, value := range values {
|
|
u.SetNew = append(u.SetNew, value.(ColumnAssigment))
|
|
}
|
|
} else {
|
|
u.Set.Values = jet.UnwindRowFromValues(value, values)
|
|
}
|
|
|
|
return u
|
|
}
|
|
|
|
func (u *updateStatementImpl) MODEL(data interface{}) UpdateStatement {
|
|
u.Set.Values = jet.UnwindRowFromModel(u.Set.Columns, data)
|
|
return u
|
|
}
|
|
|
|
func (u *updateStatementImpl) WHERE(expression BoolExpression) UpdateStatement {
|
|
u.Where.Condition = expression
|
|
return u
|
|
}
|
|
|
|
func (u *updateStatementImpl) LIMIT(limit int64) UpdateStatement {
|
|
if _, isJoinTable := u.Update.Table.(*joinTable); isJoinTable {
|
|
panic("jet: MySQL does not support LIMIT with multi-table UPDATE statements")
|
|
}
|
|
u.Limit.Count = limit
|
|
return u
|
|
}
|