From c4d37d1da86bc06a0f26a1e5a6d6c3caadd4911f Mon Sep 17 00:00:00 2001
From: alex <alex@roese.dev>
Date: Sun, 26 Nov 2023 09:51:19 +0100
Subject: [PATCH] more options for where clause

---
 rspagination/rspagination.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/rspagination/rspagination.go b/rspagination/rspagination.go
index b00b9c8..d52e392 100644
--- a/rspagination/rspagination.go
+++ b/rspagination/rspagination.go
@@ -14,7 +14,11 @@ func GetTotalPages(database *gorm.DB, paginationLimit int, any interface{}, wher
 	if whereQuery == nil {
 		database.Model(any).
 			Count(&totalPages)
-	} else {
+	} else if len(args) == 0 { // here used with db.Where(&User{Name: "jinzhu", Age: 20}).First(&user)
+		database.Model(any).
+			Where(whereQuery).
+			Count(&totalPages)
+	} else { // here used with "name = ?", "jinzhu"
 		database.Model(any).
 			Where(whereQuery, args).
 			Count(&totalPages)