104 lines
2.2 KiB
Go
104 lines
2.2 KiB
Go
package postgres
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/go-jet/jet/internal/jet"
|
|
)
|
|
|
|
// UpdateStatement is interface of SQL UPDATE statement
|
|
type UpdateStatement interface {
|
|
jet.Statement
|
|
|
|
SET(value interface{}, values ...interface{}) UpdateStatement
|
|
MODEL(data interface{}) UpdateStatement
|
|
|
|
WHERE(expression BoolExpression) UpdateStatement
|
|
RETURNING(projections ...jet.Projection) UpdateStatement
|
|
}
|
|
|
|
type updateStatementImpl struct {
|
|
jet.StatementImpl
|
|
|
|
Update jet.ClauseUpdate
|
|
Set ClauseSet
|
|
Where jet.ClauseWhere
|
|
Returning jet.ClauseReturning
|
|
}
|
|
|
|
func newUpdateStatement(table WritableTable, columns []jet.IColumn) UpdateStatement {
|
|
update := &updateStatementImpl{}
|
|
update.StatementImpl = jet.NewStatementImpl(Dialect, jet.UpdateStatementType, update, &update.Update,
|
|
&update.Set, &update.Where, &update.Returning)
|
|
|
|
update.Update.Table = table
|
|
update.Set.Columns = columns
|
|
update.Where.Mandatory = true
|
|
|
|
return update
|
|
}
|
|
|
|
func (u *updateStatementImpl) SET(value interface{}, values ...interface{}) UpdateStatement {
|
|
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) RETURNING(projections ...jet.Projection) UpdateStatement {
|
|
u.Returning.Projections = projections
|
|
return u
|
|
}
|
|
|
|
type ClauseSet struct {
|
|
Columns []jet.IColumn
|
|
Values []jet.Serializer
|
|
}
|
|
|
|
func (s *ClauseSet) Serialize(statementType jet.StatementType, out *jet.SqlBuilder) error {
|
|
out.NewLine()
|
|
out.WriteString("SET")
|
|
|
|
if len(s.Columns) == 0 {
|
|
return errors.New("jet: no columns selected")
|
|
}
|
|
|
|
if len(s.Columns) > 1 {
|
|
out.WriteString("(")
|
|
}
|
|
|
|
err := jet.SerializeColumnNames(s.Columns, out)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(s.Columns) > 1 {
|
|
out.WriteString(")")
|
|
}
|
|
|
|
out.WriteString("=")
|
|
|
|
if len(s.Values) > 1 {
|
|
out.WriteString("(")
|
|
}
|
|
|
|
err = jet.SerializeClauseList(statementType, s.Values, out)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(s.Values) > 1 {
|
|
out.WriteString(")")
|
|
}
|
|
|
|
return nil
|
|
}
|