♻️ Refactor(paginate.go): add replace count

booscaaa · Nov 9, 2023 · d1cb25b · d1cb25b
1 parent 4c099b1
commit d1cb25b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/v2/paginate/paginate.go b/v2/paginate/paginate.go
@@ -273,7 +273,7 @@ func GenerateCountQuery(params *paginQueryParams) (string, []interface{}) {
 	}
 
 	// Cláusula SELECT para contagem
-	countSelectClause := "SELECT COUNT(*)"
+	countSelectClause := "SELECT COUNT(id)"
 	clauses = append(clauses, countSelectClause)
 
 	// Cláusula FROM com tabela principal
@@ -317,7 +317,8 @@ func GenerateCountQuery(params *paginQueryParams) (string, []interface{}) {
 	// Verifica se VACUUM deve ser aplicado
 	if params.Vacuum {
 		countQuery := strings.Join(clauses, " ")
-		countQuery = "SELECT count_estimate(" + countQuery + ");"
+		countQuery = "SELECT count_estimate('" + countQuery + "');"
+		countQuery = strings.Replace(countQuery, "COUNT(id)", "1", -1)
 		return countQuery, args
 	}