diff --git a/rspagination/rspagination.go b/rspagination/rspagination.go index 61e7c04..70f1de1 100644 --- a/rspagination/rspagination.go +++ b/rspagination/rspagination.go @@ -11,9 +11,14 @@ import ( func GetTotalPages(database *gorm.DB, paginationLimit int, any interface{}, whereQuery interface{}, args ...interface{}) int { var totalPages int64 - database.Model(any). - Where(whereQuery, args). - Count(&totalPages) + if whereQuery == nil { + database.Model(any). + Count(&totalPages) + } else { + database.Model(any). + Where(whereQuery, args). + Count(&totalPages) + } return int(math.Ceil(float64(totalPages) / float64(paginationLimit))) }