jet/sqlbuilder/delete_statement.go

package sqlbuilder

import (
	"database/sql"
	"github.com/dropbox/godropbox/errors"
	"github.com/sub0zero/go-sqlbuilder/types"
)

type deleteStatement interface {
	Statement

	WHERE(expression boolExpression) deleteStatement
}

func newDeleteStatement(table writableTable) deleteStatement {
	return &deleteStatementImpl{
		table: table,
	}
}

type deleteStatementImpl struct {
	table writableTable
	where boolExpression
}

func (d *deleteStatementImpl) WHERE(expression boolExpression) deleteStatement {
	d.where = expression
	return d
}

func (d *deleteStatementImpl) serializeImpl(out *queryData) error {
	out.nextLine()
	out.writeString("DELETE FROM")

	if d.table == nil {
		return errors.New("nil tableName.")
	}

	if err := d.table.serialize(delete_statement, out); err != nil {
		return err
	}

	if d.where == nil {
		return errors.New("Deleting without a WHERE clause.")
	}

	if err := out.writeWhere(delete_statement, d.where); err != nil {
		return err
	}

	return nil
}

func (d *deleteStatementImpl) Sql() (query string, args []interface{}, err error) {
	queryData := &queryData{}

	err = d.serializeImpl(queryData)

	if err != nil {
		return
	}

	query, args = queryData.finalize()
	return
}

func (d *deleteStatementImpl) DebugSql() (query string, err error) {
	return DebugSql(d)
}

func (u *deleteStatementImpl) Query(db types.Db, destination interface{}) error {
	return Query(u, db, destination)
}

func (u *deleteStatementImpl) Execute(db types.Db) (res sql.Result, err error) {
	return Execute(u, db)
}