Skip to content

Commit

Permalink
FINERACT-1724: Fix pagination
Browse files Browse the repository at this point in the history
  • Loading branch information
adamsaghy committed Sep 14, 2023
1 parent 9547550 commit f7d41b5
Show file tree
Hide file tree
Showing 2 changed files with 68 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -91,15 +91,17 @@ public String calcFoundRows() {
}
}

public String countLastExecutedQueryResult(String sql) {
public String countLastExecutedQueryResult(@NotNull String sql) {
if (databaseTypeResolver.isMySQL()) {
return "SELECT FOUND_ROWS()";
} else {
return countQueryResult(sql);
}
}

public String countQueryResult(String sql) {
public String countQueryResult(@NotNull String sql) {
// Needs to remove the limit and offset
sql = sql.replaceAll("LIMIT \\d+", "").replaceAll("OFFSET \\d+", "").trim();
return format("SELECT COUNT(*) FROM (%s) AS temp", sql);
}

Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,64 @@
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.fineract.infrastructure.core.service.database;

import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;

public class DatabaseSpecificSQLGeneratorTest {

private final DatabaseTypeResolver databaseTypeResolver = Mockito.mock(DatabaseTypeResolver.class);
private final DatabaseSpecificSQLGenerator databaseSpecificSQLGenerator = new DatabaseSpecificSQLGenerator(databaseTypeResolver);

@Test
public void testCountQueryResultOnEmptyString() {
String sql = "";
String countQuery = databaseSpecificSQLGenerator.countQueryResult(sql);
Assertions.assertEquals("SELECT COUNT(*) FROM () AS temp", countQuery);
}

@Test
public void testCountQueryResultOnSqlWithoutLimitOrOffset() {
String sql = "SELECT 1 FROM test_table WHERE asd=2";
String countQuery = databaseSpecificSQLGenerator.countQueryResult(sql);
Assertions.assertEquals("SELECT COUNT(*) FROM (" + sql + ") AS temp", countQuery);
}

@Test
public void testCountQueryResultOnSqlWithLimit() {
String sql = "SELECT 1 FROM test_table WHERE asd=2 LIMIT 2";
String countQuery = databaseSpecificSQLGenerator.countQueryResult(sql);
Assertions.assertEquals("SELECT COUNT(*) FROM (SELECT 1 FROM test_table WHERE asd=2) AS temp", countQuery);
}

@Test
public void testCountQueryResultOnSqlWithOffset() {
String sql = "SELECT 1 FROM test_table WHERE asd=2 OFFSET 2";
String countQuery = databaseSpecificSQLGenerator.countQueryResult(sql);
Assertions.assertEquals("SELECT COUNT(*) FROM (SELECT 1 FROM test_table WHERE asd=2) AS temp", countQuery);
}

@Test
public void testCountQueryResultOnSqlWithLimitAndOffset() {
String sql = "SELECT 1 FROM test_table WHERE asd=2 OFFSET 2 LIMIT 50";
String countQuery = databaseSpecificSQLGenerator.countQueryResult(sql);
Assertions.assertEquals("SELECT COUNT(*) FROM (SELECT 1 FROM test_table WHERE asd=2) AS temp", countQuery);
}
}

0 comments on commit f7d41b5

Please sign in to comment.