From a57d58f38fc728bc062170842e7b22f7d9d263fb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Matczuk?= Date: Fri, 8 Nov 2019 23:48:15 +0100 Subject: [PATCH] Iterx inherit mapper from Queryx MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Michał Matczuk --- queryx.go | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/queryx.go b/queryx.go index ade2908..c9c815d 100644 --- a/queryx.go +++ b/queryx.go @@ -210,7 +210,7 @@ func (q *Queryx) Get(dest interface{}) error { if q.err != nil { return q.err } - return Iter(q.Query).Get(dest) + return q.Iter().Get(dest) } // GetRelease calls Get and releases the query, a released query cannot be @@ -230,7 +230,7 @@ func (q *Queryx) Select(dest interface{}) error { if q.err != nil { return q.err } - return Iter(q.Query).Select(dest) + return q.Iter().Select(dest) } // SelectRelease calls Select and releases the query, a released query cannot be @@ -244,5 +244,7 @@ func (q *Queryx) SelectRelease(dest interface{}) error { // big to be loaded with Select in order to do row by row iteration. // See Iterx StructScan function. func (q *Queryx) Iter() *Iterx { - return Iter(q.Query) + i := Iter(q.Query) + i.Mapper = q.Mapper + return i }