update_builder.go

package db

import (
	"database/sql"

	"github.com/stellar/go/support/errors"
)

// Exec executes the query that has been previously configured on the
// UpdateBuilder.
func (ub *UpdateBuilder) Exec() (sql.Result, error) {
	r, err := ub.Table.Session.Exec(ub.sql)
	if err != nil {
		return nil, errors.Wrap(err, "select failed")
	}

	return r, nil
}

// Limit is a passthrough call to the squirrel.  See
// https://godoc.org/github.com/Masterminds/squirrel#UpdateBuilder.Limit
func (ub *UpdateBuilder) Limit(limit uint64) *UpdateBuilder {
	ub.sql = ub.sql.Limit(limit)
	return ub
}

// Offset is a passthrough call to the squirrel.  See
// https://godoc.org/github.com/Masterminds/squirrel#UpdateBuilder.Offset
func (ub *UpdateBuilder) Offset(offset uint64) *UpdateBuilder {
	ub.sql = ub.sql.Offset(offset)
	return ub
}

// OrderBy is a passthrough call to the squirrel.  See
// https://godoc.org/github.com/Masterminds/squirrel#UpdateBuilder.OrderBy
func (ub *UpdateBuilder) OrderBy(
	orderBys ...string,
) *UpdateBuilder {
	ub.sql = ub.sql.OrderBy(orderBys...)
	return ub
}

// Prefix is a passthrough call to the squirrel.  See
// https://godoc.org/github.com/Masterminds/squirrel#UpdateBuilder.Prefix
func (ub *UpdateBuilder) Prefix(
	sql string,
	args ...interface{},
) *UpdateBuilder {
	ub.sql = ub.sql.Prefix(sql, args...)
	return ub
}

// Set is a passthrough call to the squirrel.  See
// https://godoc.org/github.com/Masterminds/squirrel#UpdateBuilder.Suffix
func (ub *UpdateBuilder) Set(column string, value interface{}) *UpdateBuilder {
	ub.sql = ub.sql.Set(column, value)
	return ub
}

// SetMap is a passthrough call to the squirrel.  See
// https://godoc.org/github.com/Masterminds/squirrel#UpdateBuilder.Suffix
func (ub *UpdateBuilder) SetMap(clauses map[string]interface{}) *UpdateBuilder {
	ub.sql = ub.sql.SetMap(clauses)
	return ub
}

// Suffix is a passthrough call to the squirrel.  See
// https://godoc.org/github.com/Masterminds/squirrel#UpdateBuilder.Suffix
func (ub *UpdateBuilder) Suffix(
	sql string,
	args ...interface{},
) *UpdateBuilder {
	ub.sql = ub.sql.Suffix(sql, args...)
	return ub
}

// Where is a passthrough call to the squirrel.  See
// https://godoc.org/github.com/Masterminds/squirrel#UpdateBuilder.Where
func (ub *UpdateBuilder) Where(
	pred interface{},
	args ...interface{},
) *UpdateBuilder {
	ub.sql = ub.sql.Where(pred, args...)
	return ub
}