160 lines
3.2 KiB
Go
160 lines
3.2 KiB
Go
package jet
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
"github.com/go-jet/jet/execution"
|
|
)
|
|
|
|
type UpdateStatement interface {
|
|
Statement
|
|
|
|
SET(value interface{}, values ...interface{}) UpdateStatement
|
|
MODEL(data interface{}) UpdateStatement
|
|
|
|
WHERE(expression BoolExpression) UpdateStatement
|
|
RETURNING(projections ...projection) UpdateStatement
|
|
}
|
|
|
|
func newUpdateStatement(table WritableTable, columns []column) UpdateStatement {
|
|
return &updateStatementImpl{
|
|
table: table,
|
|
columns: columns,
|
|
row: make([]clause, 0, len(columns)),
|
|
}
|
|
}
|
|
|
|
type updateStatementImpl struct {
|
|
table WritableTable
|
|
columns []column
|
|
row []clause
|
|
where BoolExpression
|
|
returning []projection
|
|
}
|
|
|
|
func (u *updateStatementImpl) SET(value interface{}, values ...interface{}) UpdateStatement {
|
|
u.row = unwindRowFromValues(value, values)
|
|
|
|
return u
|
|
}
|
|
|
|
func (u *updateStatementImpl) MODEL(data interface{}) UpdateStatement {
|
|
u.row = unwindRowFromModel(u.columns, data)
|
|
|
|
return u
|
|
}
|
|
|
|
func (u *updateStatementImpl) WHERE(expression BoolExpression) UpdateStatement {
|
|
u.where = expression
|
|
return u
|
|
}
|
|
|
|
func (u *updateStatementImpl) RETURNING(projections ...projection) UpdateStatement {
|
|
u.returning = projections
|
|
return u
|
|
}
|
|
|
|
func (u *updateStatementImpl) Sql() (sql string, args []interface{}, err error) {
|
|
out := &queryData{}
|
|
|
|
out.newLine()
|
|
out.writeString("UPDATE")
|
|
|
|
if isNil(u.table) {
|
|
return "", nil, errors.New("table to update is nil")
|
|
}
|
|
|
|
if err = u.table.serialize(update_statement, out); err != nil {
|
|
return
|
|
}
|
|
|
|
if len(u.columns) == 0 {
|
|
return "", nil, errors.New("no columns selected")
|
|
}
|
|
|
|
if len(u.row) == 0 {
|
|
return "", nil, errors.New("no values to updated")
|
|
}
|
|
|
|
out.newLine()
|
|
out.writeString("SET")
|
|
|
|
if len(u.columns) > 1 {
|
|
out.writeString("(")
|
|
}
|
|
|
|
err = serializeColumnNames(u.columns, out)
|
|
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if len(u.columns) > 1 {
|
|
out.writeString(")")
|
|
}
|
|
|
|
out.writeString("=")
|
|
|
|
if len(u.row) > 1 {
|
|
out.writeString("(")
|
|
}
|
|
|
|
err = serializeClauseList(update_statement, u.row, out)
|
|
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if len(u.row) > 1 {
|
|
out.writeString(")")
|
|
}
|
|
|
|
if u.where == nil {
|
|
return "", nil, errors.New("WHERE clause not set")
|
|
}
|
|
|
|
if err = out.writeWhere(update_statement, u.where); err != nil {
|
|
return
|
|
}
|
|
|
|
if err = out.writeReturning(update_statement, u.returning); err != nil {
|
|
return
|
|
}
|
|
|
|
//if len(u.returning) > 0 {
|
|
// out.newLine()
|
|
// out.writeString("RETURNING")
|
|
// out.increaseIdent()
|
|
// out.increaseIdent()
|
|
//
|
|
// err = serializeProjectionList(update_statement, u.returning, out)
|
|
//
|
|
// if err != nil {
|
|
// return
|
|
// }
|
|
//}
|
|
|
|
sql, args = out.finalize()
|
|
return
|
|
}
|
|
|
|
func (u *updateStatementImpl) DebugSql() (query string, err error) {
|
|
return debugSql(u)
|
|
}
|
|
|
|
func (u *updateStatementImpl) Query(db execution.DB, destination interface{}) error {
|
|
return query(u, db, destination)
|
|
}
|
|
|
|
func (u *updateStatementImpl) QueryContext(db execution.DB, context context.Context, destination interface{}) error {
|
|
return queryContext(u, db, context, destination)
|
|
}
|
|
|
|
func (u *updateStatementImpl) Exec(db execution.DB) (res sql.Result, err error) {
|
|
return exec(u, db)
|
|
}
|
|
|
|
func (u *updateStatementImpl) ExecContext(db execution.DB, context context.Context) (res sql.Result, err error) {
|
|
return execContext(u, db, context)
|
|
}
|