BAEL-3961 Using a list of values in a JDBCTemplate IN clause
This commit is contained in:
parent
a057b3af44
commit
63027fd269
|
@ -9,6 +9,13 @@ public class Employee {
|
||||||
|
|
||||||
private String address;
|
private String address;
|
||||||
|
|
||||||
|
public Employee(int id, String firstName, String lastName, String address) {
|
||||||
|
setId(id);
|
||||||
|
setFirstName(firstName);
|
||||||
|
setLastName(lastName);
|
||||||
|
setAddress(address);
|
||||||
|
}
|
||||||
|
|
||||||
public int getId() {
|
public int getId() {
|
||||||
return id;
|
return id;
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,21 +1,62 @@
|
||||||
package com.baeldung.jdbc;
|
package com.baeldung.jdbc;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Collections;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
import javax.sql.DataSource;
|
import javax.sql.DataSource;
|
||||||
|
|
||||||
import org.springframework.jdbc.core.JdbcTemplate;
|
import org.springframework.jdbc.core.JdbcTemplate;
|
||||||
|
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
|
||||||
|
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
|
||||||
|
import org.springframework.jdbc.core.namedparam.SqlParameterSource;
|
||||||
import org.springframework.stereotype.Repository;
|
import org.springframework.stereotype.Repository;
|
||||||
|
|
||||||
@Repository
|
@Repository
|
||||||
public class EmployeeDAO {
|
public class EmployeeDAO {
|
||||||
private JdbcTemplate jdbcTemplate;
|
private JdbcTemplate jdbcTemplate;
|
||||||
|
private NamedParameterJdbcTemplate namedJdbcTemplate;
|
||||||
|
|
||||||
public void setDataSource(DataSource dataSource) {
|
public void setDataSource(DataSource dataSource) {
|
||||||
jdbcTemplate = new JdbcTemplate(dataSource);
|
jdbcTemplate = new JdbcTemplate(dataSource);
|
||||||
|
namedJdbcTemplate = new NamedParameterJdbcTemplate(dataSource);
|
||||||
}
|
}
|
||||||
|
|
||||||
public int getCountOfEmployees() {
|
public int getCountOfEmployees() {
|
||||||
return jdbcTemplate.queryForObject("SELECT COUNT(*) FROM EMPLOYEE", Integer.class);
|
return jdbcTemplate.queryForObject("SELECT COUNT(*) FROM EMPLOYEE", Integer.class);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public List<Employee> getEmployeesFromIdListNamed(List<Integer> ids) {
|
||||||
|
SqlParameterSource parameters = new MapSqlParameterSource("ids", ids);
|
||||||
|
List<Employee> employees = namedJdbcTemplate.query("SELECT * FROM EMPLOYEE WHERE id IN (:ids)",
|
||||||
|
parameters,
|
||||||
|
(rs, rowNum) -> new Employee(rs.getInt("id"), rs.getString("first_name"), rs.getString("last_name"), rs.getString("address")));
|
||||||
|
|
||||||
|
return employees;
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<Employee> getEmployeesFromIdList(List<Integer> ids) {
|
||||||
|
String inSql = String.join(",", Collections.nCopies(ids.size(), "?"));
|
||||||
|
List<Employee> employees = jdbcTemplate.query("SELECT * FROM EMPLOYEE WHERE id IN (" + inSql +")",
|
||||||
|
ids.toArray(),
|
||||||
|
(rs, rowNum) -> new Employee(rs.getInt("id"), rs.getString("first_name"), rs.getString("last_name"), rs.getString("address")));
|
||||||
|
|
||||||
|
return employees;
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<Employee> getEmployeesFromLargeIdList(List<Integer> ids) {
|
||||||
|
jdbcTemplate.execute("CREATE TEMPORARY TABLE employee_tmp (id INT NOT NULL)");
|
||||||
|
|
||||||
|
List<Object[]> employeeIds = new ArrayList<>();
|
||||||
|
for (Integer id : ids) {
|
||||||
|
employeeIds.add(new Object[] { id });
|
||||||
|
}
|
||||||
|
jdbcTemplate.batchUpdate("INSERT INTO employee_tmp VALUES(?)", employeeIds);
|
||||||
|
|
||||||
|
List<Employee> employees = jdbcTemplate.query("SELECT * FROM EMPLOYEE WHERE id IN (SELECT id FROM employee_tmp)",
|
||||||
|
(rs, rowNum) -> new Employee(rs.getInt("id"), rs.getString("first_name"), rs.getString("last_name"), rs.getString("address")));
|
||||||
|
|
||||||
|
return employees;
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -2,8 +2,12 @@ package com.baeldung.jdbc;
|
||||||
|
|
||||||
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
import javax.sql.DataSource;
|
import javax.sql.DataSource;
|
||||||
|
|
||||||
|
import org.junit.Before;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
import org.junit.runner.RunWith;
|
import org.junit.runner.RunWith;
|
||||||
import org.mockito.Mock;
|
import org.mockito.Mock;
|
||||||
|
@ -14,13 +18,24 @@ import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseBuilder;
|
||||||
import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseType;
|
import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseType;
|
||||||
import org.springframework.test.util.ReflectionTestUtils;
|
import org.springframework.test.util.ReflectionTestUtils;
|
||||||
|
|
||||||
|
|
||||||
@RunWith(MockitoJUnitRunner.class)
|
@RunWith(MockitoJUnitRunner.class)
|
||||||
|
|
||||||
public class EmployeeDAOUnitTest {
|
public class EmployeeDAOUnitTest {
|
||||||
@Mock
|
@Mock
|
||||||
JdbcTemplate jdbcTemplate;
|
JdbcTemplate jdbcTemplate;
|
||||||
|
|
||||||
|
DataSource dataSource;
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void setup() {
|
||||||
|
dataSource = new EmbeddedDatabaseBuilder().setType(EmbeddedDatabaseType.H2)
|
||||||
|
.generateUniqueName(true)
|
||||||
|
.addScript("classpath:jdbc/schema.sql")
|
||||||
|
.addScript("classpath:jdbc/test-data.sql")
|
||||||
|
.build();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void whenMockJdbcTemplate_thenReturnCorrectEmployeeCount() {
|
public void whenMockJdbcTemplate_thenReturnCorrectEmployeeCount() {
|
||||||
EmployeeDAO employeeDAO = new EmployeeDAO();
|
EmployeeDAO employeeDAO = new EmployeeDAO();
|
||||||
|
@ -38,14 +53,50 @@ public class EmployeeDAOUnitTest {
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void whenInjectInMemoryDataSource_thenReturnCorrectEmployeeCount() {
|
public void whenInjectInMemoryDataSource_thenReturnCorrectEmployeeCount() {
|
||||||
DataSource dataSource = new EmbeddedDatabaseBuilder().setType(EmbeddedDatabaseType.H2)
|
|
||||||
.addScript("classpath:jdbc/schema.sql")
|
|
||||||
.addScript("classpath:jdbc/test-data.sql")
|
|
||||||
.build();
|
|
||||||
|
|
||||||
EmployeeDAO employeeDAO = new EmployeeDAO();
|
EmployeeDAO employeeDAO = new EmployeeDAO();
|
||||||
employeeDAO.setDataSource(dataSource);
|
employeeDAO.setDataSource(dataSource);
|
||||||
|
|
||||||
assertEquals(4, employeeDAO.getCountOfEmployees());
|
assertEquals(4, employeeDAO.getCountOfEmployees());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void givenSmallIdList_whenGetEmployeesFromIdList_thenReturnCorrectEmployees() {
|
||||||
|
List<Integer> ids = new ArrayList<>();
|
||||||
|
ids.add(1);
|
||||||
|
ids.add(3);
|
||||||
|
ids.add(4);
|
||||||
|
EmployeeDAO employeeDAO = new EmployeeDAO();
|
||||||
|
employeeDAO.setDataSource(dataSource);
|
||||||
|
|
||||||
|
List<Employee> employees = employeeDAO.getEmployeesFromIdList(ids);
|
||||||
|
|
||||||
|
assertEquals(3, employees.size());
|
||||||
|
assertEquals(1, employees.get(0).getId());
|
||||||
|
assertEquals(3, employees.get(1).getId());
|
||||||
|
assertEquals(4, employees.get(2).getId());
|
||||||
|
|
||||||
|
employees = employeeDAO.getEmployeesFromIdListNamed(ids);
|
||||||
|
|
||||||
|
assertEquals(3, employees.size());
|
||||||
|
assertEquals(1, employees.get(0).getId());
|
||||||
|
assertEquals(3, employees.get(1).getId());
|
||||||
|
assertEquals(4, employees.get(2).getId());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void givenLargeIdList_whenGetEmployeesFromIdList_thenReturnCorrectEmployees() {
|
||||||
|
List<Integer> ids = new ArrayList<>();
|
||||||
|
ids.add(1);
|
||||||
|
ids.add(3);
|
||||||
|
ids.add(4);
|
||||||
|
EmployeeDAO employeeDAO = new EmployeeDAO();
|
||||||
|
employeeDAO.setDataSource(dataSource);
|
||||||
|
|
||||||
|
List<Employee> employees = employeeDAO.getEmployeesFromLargeIdList(ids);
|
||||||
|
|
||||||
|
assertEquals(3, employees.size());
|
||||||
|
assertEquals(1, employees.get(0).getId());
|
||||||
|
assertEquals(3, employees.get(1).getId());
|
||||||
|
assertEquals(4, employees.get(2).getId());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue