querier_selects.go

package reform

import (
	"database/sql"
	"fmt"
	mysqlDriver "github.com/go-sql-driver/mysql"
	"strings"
)

// NextRow scans next result row from rows to str. If str has valid method "AfterFind", it also calls AfterFind().
// It is caller's responsibility to call rows.Close().
//
// If there is no next result row, it returns ErrNoRows. It also may return rows.Next(), rows.Scan()
// and AfterFind() errors.
//
// See SelectRows example for idiomatic usage.
func (q *Querier) NextRow(str Struct, rows *sql.Rows) error {
	var err error
	next := rows.Next()
	if !next {
		err = rows.Err()
		if err == nil {
			err = ErrNoRows
		}
		return err
	}

	if err = rows.Scan(str.Pointers()...); err != nil {
		return err
	}

	return q.callStructMethod(str, "AfterFind")
}

// selectQuery returns full SELECT query for given view and tail.
func (q *Querier) selectQuery(view View, tail string, limit1 bool, forceAnotherTable *string, forceFields []string) string {
	queryStart := q.startQuery("SELECT")

	if limit1 && q.SelectLimitMethod() == SelectTop {
		queryStart += " TOP 1"
	}

	var columnsQuoted []string
	if len(forceFields) > 0 {
		for _, field := range forceFields {
			column := view.ColumnNameByFieldName(field)
			if column == "" {
				panic("Unknown field:" + field)
			}
			columnsQuoted = append(columnsQuoted, q.QuoteIdentifier(column))
		}
	} else {
		columnsQuoted = q.QualifiedColumns(view)
	}
	columnsQuery := strings.Join(columnsQuoted, ", ")

	var tableQuery string
	if forceAnotherTable == nil {
		tableQuery = q.QualifiedView(view)
	} else {
		tableQuery = *forceAnotherTable
	}

	return fmt.Sprintf("%s %s FROM %s %s", queryStart, columnsQuery, tableQuery, tail)
}

// FlexSelectOneTo queries str's View with tail, args, forceAnotherTable and forceFields and scans first result to str.
// If str has valid method "AfterFind", it also calls AfterFind().
//
// If there are no rows in result, it returns ErrNoRows. It also may return QueryRow(), Scan()
// and AfterFind() errors.
func (q *Querier) FlexSelectOneTo(str Struct, forceAnotherTable *string, forceFields []string, tail string, args ...interface{}) error {
	query := q.selectQuery(str.View(), tail, true, forceAnotherTable, forceFields)
	for {
		err := q.QueryRow(query, args...).Scan(str.FieldPointersByNames(forceFields)...)
		if err == mysqlDriver.ErrInvalidConn {
			continue
		}
		if err != nil {
			return err
		}
		break
	}

	return q.callStructMethod(str, "AfterFind")
}

// SelectOneTo queries str's View with tail and args and scans first result to str.
// If str has valid method "AfterFind", it also calls AfterFind().
//
// If there are no rows in result, it returns ErrNoRows. It also may return QueryRow(), Scan()
// and AfterFind() errors.
func (q *Querier) SelectOneTo(str Struct, tail string, args ...interface{}) error {
	return q.FlexSelectOneTo(str, nil, nil, tail, args...)
}

// SelectOneFrom queries view with tail and args and scans first result to new Struct str.
// If str has valid method "AfterFind", it also calls AfterFind().
//
// If there are no rows in result, it returns nil, ErrNoRows. It also may return QueryRow(), Scan()
// and AfterFind() errors.
func (q *Querier) SelectOneFrom(view View, tail string, args ...interface{}) (Struct, error) {
	str := view.NewStruct()
	if err := q.SelectOneTo(str, tail, args...); err != nil {
		return nil, err
	}
	return str, nil
}

// FlexSelectRows queries view with tail, args, forceAnotherTable and forceFields and returns rows. They can then be iterated with NextRow().
// It is caller's responsibility to call rows.Close().
//
// In case of error rows will be nil. Error is never ErrNoRows.
//
// See example for idiomatic usage.
func (q *Querier) FlexSelectRows(view View, forceAnotherTable *string, forceFields []string, tail string, args ...interface{}) (*sql.Rows, error) {
	query := q.selectQuery(view, tail, false, forceAnotherTable, forceFields)
	return q.Query(query, args...)
}

// SelectRows queries view with tail and args and returns rows. They can then be iterated with NextRow().
// It is caller's responsibility to call rows.Close().
//
// In case of error rows will be nil. Error is never ErrNoRows.
//
// See example for idiomatic usage.
func (q *Querier) SelectRows(view View, tail string, args ...interface{}) (*sql.Rows, error) {
	query := q.selectQuery(view, tail, false, nil, nil)
	return q.Query(query, args...)
}

// SelectAllFrom queries view with tail and args and returns a slice of new Structs.
// If view's Struct has valid method "AfterFind", it also calls AfterFind().
//
// In case of query error slice will be nil. If error is encountered during iteration,
// partial result and error will be returned. Error is never ErrNoRows.
func (q *Querier) SelectAllFrom(view View, tail string, args ...interface{}) (structs []Struct, err error) {
	var rows *sql.Rows
	rows, err = q.SelectRows(view, tail, args...)
	if err != nil {
		return
	}
	defer func() {
		e := rows.Close()
		if err == nil {
			err = e
		}
	}()

	for {
		str := view.NewStruct()
		if err = q.NextRow(str, rows); err != nil {
			break
		}

		structs = append(structs, str)
	}
	if err == ErrNoRows {
		err = nil
	}
	return
}

// findTail returns a tail of SELECT query for given view, column and arg.
func (q *Querier) findTail(view string, column string, arg interface{}, limit1 bool) (tail string, needArg bool) {
	qi := q.QuoteIdentifier(view) + "." + q.QuoteIdentifier(column)
	if arg == nil {
		tail = fmt.Sprintf("WHERE %s IS NULL", qi)
	} else {
		tail = fmt.Sprintf("WHERE %s = %s", qi, q.Placeholder(1))
		needArg = true
	}

	if limit1 && q.SelectLimitMethod() == Limit {
		tail += " LIMIT 1"
	}

	return
}

// FindOneTo queries str's View with column and arg and scans first result to str.
// If str has valid method "AfterFind", it also calls AfterFind().
//
// If there are no rows in result, it returns ErrNoRows. It also may return QueryRow(), Scan()
// and AfterFind() errors.
func (q *Querier) FindOneTo(str Struct, column string, arg interface{}) error {
	tail, needArg := q.findTail(str.View().Name(), column, arg, true)
	if needArg {
		return q.SelectOneTo(str, tail, arg)
	}
	return q.SelectOneTo(str, tail)
}

// FindOneFrom queries view with column and arg and scans first result to new Struct str.
// If str has valid method "AfterFind", it also calls AfterFind().
//
// If there are no rows in result, it returns nil, ErrNoRows. It also may return QueryRow(), Scan()
// and AfterFind() errors.
func (q *Querier) FindOneFrom(view View, column string, arg interface{}) (Struct, error) {
	tail, needArg := q.findTail(view.Name(), column, arg, true)
	if needArg {
		return q.SelectOneFrom(view, tail, arg)
	}
	return q.SelectOneFrom(view, tail)
}

// FindRows queries view with column and arg and returns rows. They can then be iterated with NextRow().
// It is caller's responsibility to call rows.Close().
//
// In case of error rows will be nil. Error is never ErrNoRows.
//
// See SelectRows example for idiomatic usage.
func (q *Querier) FindRows(view View, column string, arg interface{}) (*sql.Rows, error) {
	tail, needArg := q.findTail(view.Name(), column, arg, false)
	if needArg {
		return q.SelectRows(view, tail, arg)
	}
	return q.SelectRows(view, tail)
}

// FindAllFrom queries view with column and args and returns a slice of new Structs.
// If view's Struct has valid method "AfterFind", it also calls AfterFind().
//
// In case of query error slice will be nil. If error is encountered during iteration,
// partial result and error will be returned. Error is never ErrNoRows.
func (q *Querier) FindAllFrom(view View, column string, args ...interface{}) ([]Struct, error) {
	p := strings.Join(q.Placeholders(1, len(args)), ", ")
	qi := q.QualifiedView(view) + "." + q.QuoteIdentifier(column)
	tail := fmt.Sprintf("WHERE %s IN (%s)", qi, p)
	return q.SelectAllFrom(view, tail, args...)
}

// FindByPrimaryKeyTo queries record's Table with primary key and scans first result to record.
// If record has valid method "AfterFind", it also calls AfterFind().
//
// If there are no rows in result, it returns ErrNoRows. It also may return QueryRow(), Scan()
// and AfterFind() errors.
func (q *Querier) FindByPrimaryKeyTo(record Record, pk interface{}) error {
	table := record.Table()
	return q.FindOneTo(record, table.Columns()[table.PKColumnIndex()], pk)
}

// FindByPrimaryKeyFrom queries table with primary key and scans first result to new Record.
// If record has valid method "AfterFind", it also calls AfterFind().
//
// If there are no rows in result, it returns nil, ErrNoRows. It also may return QueryRow(), Scan()
// and AfterFind() errors.
func (q *Querier) FindByPrimaryKeyFrom(table Table, pk interface{}) (Record, error) {
	record := table.NewRecord()
	if err := q.FindOneTo(record, table.Columns()[table.PKColumnIndex()], pk); err != nil {
		return nil, err
	}
	return record, nil
}

// Reload is a shortcut for FindByPrimaryKeyTo for given record.
func (q *Querier) Reload(record Record) error {
	return q.FindByPrimaryKeyTo(record, record.PKValue())
}