Bug from to fixed

verzth · Jun 29, 2020 · 418bd9a · 418bd9a
1 parent 4795f94
commit 418bd9a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/paginator/pagination.go b/paginator/pagination.go
@@ -5,6 +5,7 @@ import (
 	"github.com/jinzhu/gorm"
 	"math"
 	"net/http"
+	"reflect"
 )
 
 type Param struct {
@@ -65,7 +66,12 @@ func Paginate(p *Param, result interface{}) *Pagination {
 	} else {
 		offset = (p.Page - 1) * p.Limit
 	}
-	db.Limit(p.Limit).Offset(offset).Find(result).Count(&countInPage)
+	db.Limit(p.Limit).Offset(offset).Find(result)
+
+	indirect := reflect.ValueOf(result)
+	if indirect.IsValid() && indirect.Elem().Kind() == reflect.Slice {
+		countInPage = indirect.Elem().Len()
+	}
 	<-done
 
 	paginate.FirstPageUrl = fmt.Sprintf("%s%s?page=%d", p.Req.Host, p.Req.URL.Path, 1)