delete_query.go

package gosql

import (
	"database/sql"
	"strings"
)

// DeleteQuery .
type DeleteQuery struct {
	db        *DB
	table     string
	joins     []string
	wheres    []*where
	whereArgs []interface{}
}

// Where .
func (dq *DeleteQuery) Where(condition string, args ...interface{}) *DeleteQuery {
	w := &where{
		conjunction: " and ",
		condition:   condition,
	}
	dq.wheres = append(dq.wheres, w)
	dq.whereArgs = append(dq.whereArgs, args...)
	return dq
}

// OrWhere .
func (dq *DeleteQuery) OrWhere(condition string, args ...interface{}) *DeleteQuery {
	w := &where{
		conjunction: " or ",
		condition:   condition,
	}
	dq.wheres = append(dq.wheres, w)
	dq.whereArgs = append(dq.whereArgs, args...)
	return dq
}

// Join .
func (dq *DeleteQuery) Join(join string) *DeleteQuery {
	dq.joins = append(dq.joins, join)
	return dq
}

// Exec .
func (dq *DeleteQuery) Exec() (sql.Result, error) {
	return dq.db.db.Exec(dq.String(), dq.whereArgs...)
}

// String returns the string representation of DeleteQuery.
func (dq *DeleteQuery) String() string {
	var q strings.Builder
	q.WriteString("delete from ")
	q.WriteString(dq.table)
	for _, join := range dq.joins {
		q.WriteString(" join ")
		q.WriteString(join)
	}
	for i, where := range dq.wheres {
		if i == 0 {
			q.WriteString(" where ")
		} else {
			q.WriteString(where.conjunction)
		}
		q.WriteString(where.condition)
	}
	return q.String()
}