43 lines
706 B
Go
43 lines
706 B
Go
package sql
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"github.com/baez90/nurse/grammar"
|
|
"github.com/baez90/nurse/validation"
|
|
)
|
|
|
|
var _ validation.FromCall[*sql.Rows] = (*RowsValidator)(nil)
|
|
|
|
type RowsValidator struct {
|
|
Want int
|
|
}
|
|
|
|
func (r *RowsValidator) Validate(in *sql.Rows) error {
|
|
readRows := 0
|
|
for in.Next() {
|
|
readRows++
|
|
}
|
|
|
|
if err := in.Err(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if readRows != r.Want {
|
|
return fmt.Errorf("expected %d rows but got %d", r.Want, readRows)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *RowsValidator) UnmarshalCall(c grammar.Call) error {
|
|
if err := grammar.ValidateParameterCount(c.Params, 1); err != nil {
|
|
return err
|
|
}
|
|
|
|
var err error
|
|
r.Want, err = c.Params[0].AsInt()
|
|
|
|
return err
|
|
}
|