delete.go

package sqlbuilder

import (
	"bytes"
)

//
// Author: 陈永佳 chenyongjia@parkingwang.com, yoojiachen@gmail.com
//

type DeleteBuilder struct {
	ctx    *SQLContext
	table  string
	ensure bool // 删除全表时，需要强制设置一个标记位。
}

func newDeleteBuilder(ctx *SQLContext, table string) *DeleteBuilder {
	return &DeleteBuilder{
		ctx:    ctx,
		table:  table,
		ensure: false,
	}
}

func (slf *DeleteBuilder) Table(table string) *DeleteBuilder {
	slf.table = table
	return slf
}

func (slf *DeleteBuilder) compile() *bytes.Buffer {
	if "" == slf.table {
		panic("Table name not found, you should call 'Table(table)' method to set it")
	}
	buf := new(bytes.Buffer)
	buf.WriteString("DELETE FROM ")
	buf.WriteString(slf.ctx.escapeName(slf.table))
	return buf
}

func (slf *DeleteBuilder) YesImSureDeleteTable() *DeleteBuilder {
	slf.ensure = true
	return slf
}

func (slf *DeleteBuilder) Where(conditions SQLStatement) *WhereBuilder {
	return newWhereBuilder(slf.ctx, slf.Compile(), conditions)
}

func (slf *DeleteBuilder) Compile() string {
	return slf.compile().String()
}

func (slf *DeleteBuilder) ToSQL() string {
	sqlTxt := sqlEndpoint(slf.compile())
	if slf.ensure {
		return sqlTxt
	} else {
		panic("Warning for FULL-DELETE the table, you must call 'YesImSureDeleteTable(bool)' to ensure. SQLText: " + sqlTxt)
	}
}

func (slf *DeleteBuilder) Execute() *Executor {
	return newExecute(slf.ToSQL(), slf.ctx.prepare)
}