使用 Java ArrayList 列出用户的共同关注者

本文将介绍如何使用 Java 和 ArrayList 来查找并列出一个用户在社交网络中的共同关注者。通过优化的 SQL 查询,我们可以高效地检索数据库中相关信息,并将其存储在 ArrayList 中,最终返回共同关注者的列表。

实现共同关注者列表

以下代码展示了如何使用 SQL 查询来获取用户的共同关注者,并将结果存储在 ArrayList 中。

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import javax.swing.JOptionPane;

public class SocialNetwork {

    private Connection connection; // 假设已建立数据库连接

    public SocialNetwork(Connection connection) {
        this.connection = connection;
    }

    public ArrayList listMutualFollowers(int id_user) {
        ArrayList data = new ArrayList();

        try {
            PreparedStatement ps = connection.prepareStatement(
                    "SELECT id_follower " +
                            "FROM followers a " +
                            "WHERE id_follower IN ( " +
                            "SELECT id_follower " +
                            "FROM followers b " +
                            "WHERE a.id_user = b.id_user " +
                            ") and id_user = ?");
            ps.setInt(1, id_user); // 使用参数化查询,防止 SQL 注入
            ResultSet rs = ps.executeQuery();

            while (rs.next()) {
                data.add(new Object[]{
                        getFollowerName(rs.getInt("id_follower"))
                });
            }

            rs.close();
            ps.close();

        } catch (SQLException e) {
            e.printStackTrace();
            JOptionPane.showMessageDialog(null, "listMutualFollowers():" + e.getMessage());
            return null;
        }

        return data;
    }

    // 辅助方法:根据 id 获取关注者姓名 (需要根据实际情况修改)
    private String getFollowerName(int id_follower) {
        String followerName = "Unknown";
        try {
            PreparedStatement ps = connection.prepareStatement("SELECT name FROM users WHERE id = ?");
            ps.setInt(1, id_follower);
            ResultSet rs = ps.executeQuery();
            if (rs.next()) {
                followerName = rs.getString("name");
            }
            rs.close();
            ps.close();
        } catch (SQLException e) {
            e.printStackTrace();
            JOptionPane.showMessageDialog(null, "getFollowerName():" + e.getMessage());
        }
        return followerName;
    }

    // 示例用法 (需要根据实际情况修改)
    public static void main(String[] args) {
        // 假设已经建立了数据库连接
        Connection connection = null; // 替换为你的数据库连接代码
        SocialNetwork socialNetwork = new SocialNetwork(connection);
        ArrayList mutualFollowers = socialNetwork.listMutualFollowers(123); // 假设用户ID为 123

        if (mutualFollowers != null) {
            System.out.println("Mutual Followers:");
            for (Object[] follower : (ArrayList) mutualFollowers) {
                System.out.println(follower[0]);
            }
        } else {
            System.out.println("Failed to retrieve mutual followers.");
        }
    }
}

代码解释:

  1. SQL 查询优化: 使用包含子查询的 SQL 语句可以有效地找出共同关注者。 SELECT id_follower FROM followers a WHERE id_follower IN (SELECT id_follower FROM followers b WHERE a.id_user = b.id_user) and id_user = ? 这个查询语句首先找出所有关注者ID,这些关注者同时也是其他用户的关注者,并且这些其他用户也关注了指定的 id_user。
  2. PreparedStatement: 使用 PreparedStatement 可以防止 SQL 注入攻击,并且可以提高查询效率。 在代码中,ps.setInt(1, id_user); 将 id_user 的值安全地设置到 SQL 查询中。
  3. ResultSet: 使用 ResultSet 迭代查询结果,并将每个共同关注者的姓名添加到 ArrayList 中。
  4. 异常处理: 使用 try-catch 块来处理 SQLException 异常,并在出现错误时显示错误消息。
  5. 资源关闭: 务必在 finally 块中关闭 ResultSet 和 PreparedStatement,以释放数据库资源。 虽然示例中没有 finally 块,但在实际项目中强烈建议添加。
  6. 参数化查询: 示例代码使用了参数化查询,这是防止SQL注入的最佳实践。 直接将用户输入拼接到SQL语句中是非常危险的。

注意事项:

  • 确保数据库连接已经建立,并且 followers 表和 users 表存在,并且包含 id_user、id_follower 和 name 字段。
  • 根据实际情况修改 getFollowerName() 方法,以正确地从数据库中获取关注者的姓名。
  • 在实际应用中,可能需要分页显示共同关注者列表,以提高用户体验。
  • 数据库连接 connection 需要在代码执行完毕后关闭,以释放资源。可以在 finally 块中关闭连接。
  • 示例代码使用了 JOptionPane 显示错误信息,在非 GUI 应用中,应该使用日志记录器来记录错误信息。

总结:

通过使用优化的 SQL 查询和 Java ArrayList,可以有效地列出一个用户的共同关注者。 关键在于正确构造 SQL 查询,并使用 PreparedStatement 来防止 SQL 注入攻击。同时,良好的异常处理和资源管理也是必不可少的。 请根据实际情况修改代码,以满足特定的需求。