Merge pull request #620 from upper/issue-616_add-limit-to-one

Set "LIMIT 1" when using One()
upper · Jan 9, 2021 · 0f7146a · 0f7146a
2 parents bc21d72 + d3f71bb
commit 0f7146a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/internal/sqladapter/result.go b/internal/sqladapter/result.go
@@ -234,7 +234,8 @@ func (r *Result) All(dst interface{}) error {
 
 // One fetches only one Result from the set.
 func (r *Result) One(dst interface{}) error {
-	query, err := r.buildPaginator()
+	one := r.Limit(1).(*Result)
+	query, err := one.buildPaginator()
 	if err != nil {
 		r.setErr(err)
 		return err

diff --git a/internal/testsuite/sql_suite.go b/internal/testsuite/sql_suite.go
@@ -124,7 +124,7 @@ func (s *SQLTestSuite) TestPreparedStatementsCache() {
 
 			// This query is different on each iteration and generates a new
 			// prepared statement everytime it's called.
-			res := sess.Collection("artist").Find().Select(db.Raw(fmt.Sprintf("count(%d)", i)))
+			res := sess.Collection("artist").Find().Select(db.Raw(fmt.Sprintf("count(%d) AS c", i)))
 
 			var count map[string]uint64
 			err := res.One(&count)