在开发过程中,我们经常需要从数据库中查询多条记录,并对这些记录进行逐一处理。一个常见场景是查询一批用户的电子邮件地址,然后向他们发送通知。然而,如果数据库访问层(dao)设计不当,可能会导致只获取并处理了查询结果集中的第一条数据,而忽略了后续的所有数据。
原始代码示例中,getEmail 方法虽然执行了查询,但其私有辅助方法 getEmail(ResultSet searchResultSet) 最终只返回了 result.get(0),这导致无论查询到多少条记录,都只处理了第一条。在调用端,userDto = delegate.getEmail() 接收到的也只有一个 UserDto 对象,因此后续的邮件发送也只会针对这一个地址。
// 原始的getEmail方法(仅返回第一条数据)
private UserDto getEmail(ResultSet searchResultSet) throws SQLException {
List result = new ArrayList<>();
UserDto userDto = null;
while (searchResultSet.next()) {
userDto = new UserDto();
userDto.setEmailAddress(searchResultSet.getString(1));
result.add(userDto);
}
// 错误:只返回列表中的第一个元素
return result == null ? null : result.size() == 0 ? null : result.get(0);
}
// 调用端(仅处理第一条数据)
Delegate delegate = new Delegate();
UserDto userDto = delegate.getEmail(); // 仅获取到第一个UserDto
String toEmail = userDto.getEmailAddress();
delegate.sendNotification("...", "...", toEmail, "...", "...", "...", body); 要解决上述问题,核心在于两点:
下面我们将逐步修改代码以实现这一目标。
首先,我们需要修改 getEmail 方法,使其能够返回一个 UserDto 对象的列表,而不是单个对象。
修改 public UserDto getEmail() 方法:
将公共方法的返回类型从 UserDto 改为 List
public ListgetEmail() { Connection connection = null; PreparedStatement preparedStatement = null; ResultSet searchResultSet = null; try { connection = getConnection(); // 假设getConnection()方法已定义 preparedStatement = connection.prepareStatement( "SELECT EMAIL FROM USER WHERE USER.U_SEQ IN ('1','650')"); searchResultSet = preparedStatement.executeQuery(); // 调用私有辅助方法,现在它将返回一个列表 return getEmailList(searchResultSet); // 修改方法名以示区分,或直接重载 } catch (Exception e) { throw new RuntimeException("Error retrieving emails: " + e.getMessage(), e); } finally { // 确保资源被关闭,避免内存泄漏和数据库连接耗尽 try { if (searchResultSet != null) searchResultSet.close(); if (preparedStatement != null) preparedStatement.close(); if (connection != null) connection.close(); // 通常在更高级别的Service或DAO管理连接 } catch (SQLException e) { System.err.println("Error closing resources: " + e.getMessage()); } } }
修改 private UserDto getEmail(ResultSet searchResultSet) 辅助方法:
将私有辅助方法的返回类型从 UserDto 改为 List
private ListgetEmailList(ResultSet searchResultSet) throws SQLException { List result = new ArrayList<>(); // 遍历ResultSet中的每一行 while (searchResultSet.next()) { UserDto userDto = new UserDto(); // 从当前行获取EMAIL列的值,列索引通常从1开始 userDto.setEmailAddress(searchResultSet.getString(1)); result.add(userDto); } // 返回 包含所有UserDto对象的列表 return result; }
UserDto 示例:
为了完整性,这里提供一个简单的 UserDto 类结构。
public class UserDto {
private String emailAddress;
public String getEmailAddress() {
return emailAddress;
}
public void setEmailAddress(String emailAddress) {
this.emailAddress = emailAddress;
}
// 可以添加其他属性和方法
}在 Delegate 类或其他业务逻辑层中,现在可以调用修改后的 getEmail 方法获取 UserDto 列表,然后遍历该列表,对每个 UserDto 对象执行发送邮件的操作。
// 在Delegate类或业务逻辑中
public void sendBulkNotifications(String from, String subject, String body) {
Delegate delegate = new Delegate(); // 假设Delegate实例已创建或通过依赖注入获取
// 调用修改后的getEmail方法,获取所有用户的列表
List users = delegate.getEmail();
if (users != null && !users.isEmpty()) {
for (UserDto user : users) {
String toEmail = user.getEmailAddress();
// 为每个用户发送邮件
// 假设sendNotification方法签名不变
delegate.sendNotification(from, subject, toEmail, "", "", "", body);
System.out.println("Email sent to: " + toEmail);
}
} else {
System.out.println("No emails found to send notifications.");
}
}
// 假设Delegate类中存在sendNotification方法
// public void sendNotification(String from, String subject, String to, String cc, String bcc, String attachment, String body) {
// // 实际的邮件发送逻辑
// } // 使用try-with-resources的示例 public ListgetEmail() { String sql = "SELECT EMAIL FROM USER WHERE USER.U_SEQ IN ('1','650')"; try (Connection connection = getConnection(); PreparedStatement preparedStatement = connection.prepareStatement(sql); ResultSet searchResultSet = preparedStatement.executeQuery()) { return getEmailList(searchResultSet); } catch (SQLException e) { throw new RuntimeException("Error retrieving emails: " + e.getMessage(), e); } }
通过将数据库访问方法设计为返回一个数据集合(如 List