stmt.go

package sqlmock

import "database/sql/driver"

// Stmt implements driver.Stmt and delegates calls to the mock connection.
type Stmt struct {
	query string
	conn  *Conn
}

// Close closes the prepared statement.
func (s *Stmt) Close() error {
	return nil
}

// NumInput reports variable argument support for this statement.
func (s *Stmt) NumInput() int {
	return -1
}

// Exec converts args into named values and delegates execution to the connection.
func (s *Stmt) Exec(args []driver.Value) (driver.Result, error) {
	named := make([]driver.NamedValue, len(args))
	for i, v := range args {
		named[i] = driver.NamedValue{
			Ordinal: i + 1,
			Value:   v,
		}
	}

	return s.conn.ExecContext(nil, s.query, named)
}

// Query converts args into named values and delegates querying to the connection.
func (s *Stmt) Query(args []driver.Value) (driver.Rows, error) {
	named := make([]driver.NamedValue, len(args))
	for i, v := range args {
		named[i] = driver.NamedValue{
			Ordinal: i + 1,
			Value:   v,
		}
	}

	return s.conn.QueryContext(nil, s.query, named)
}