导入csv文件到数据库

csv:逗号分隔值（Comma-Separated Values，CSV，有时也称为字符分隔值，因为分隔字符也可以不是逗号），其文件以纯文本形式存储表格数据（数字和文本）。纯文本意味着该文件是一个字符序列，不含必须像二进制数字那样被解读的数据。CSV文件由任意数目的记录组成，记录间以某种换行符分隔；每条记录由字段组成，字段间的分隔符是其它字符或字符串，最常见的是逗号或制表符。通常，所有记录都有完全相同的字段序列。

这是使用记事本打开大一个csv文件：

另外，csv也可以像excel那样作为一个表格的形式打开：

当然，首要先要一个前提，你要添加到数据库的数据，必须和数据库中标的字段互相对应：

注意：这里我故意将表格中某些字段没有输入数据，尤其是表格的第五行，主键没有，所以，这个是不能插入到数据中的，我们将在后台进行判断。

下面开始代码：（数据库连接部分省略）

public static void main(String[] args) throws IOException {
        // 获取要录入的文件
        BufferedReader br = new BufferedReader(new FileReader(new File("C:/Users/Nemo/Desktop/testCSV.csv")));
        // 读取直到最后一行
        String line = "";
        boolean isFirstRow = true; // 判断是不是表头
        int collumNum = 0;
        while ((line = br.readLine()) != null) {
            if (isFirstRow) { // 如果是表头，根据表头判断这个表一共有几个字段，并跳过表头的插入
                isFirstRow = false;
                collumNum = line.split(",").length;
                continue;
            }
            String[] fields = new String[collumNum];
            fields = line.split(",");
            if (fields.length == 7) { // 判断最后一个字段有没有内容
                User user = new User(); // 封装要添加到数据库的对象
                if (fields[0] == null || fields[0].equals("")) {
                    System.out.println("**************此行没有主键，跳过**************");
                    continue;
                } else
                    user.setId(Integer.parseInt(fields[0]));
                if (fields[1] == null || fields[1].equals(""))
                    user.setName(null);
                else
                    user.setName(fields[1]);
                if (fields[2] == null || fields[2].equals(""))
                    user.setPassword(null);
                else
                    user.setPassword(fields[2]);
                if (fields[3] == null || fields[3].equals(""))
                    user.setNickName(null);
                else
                    user.setNickName(fields[3]);
                if (fields[4] == null || fields[4].equals(""))
                    ;
                else
                    user.setAge(Integer.parseInt(fields[4]));
                if (fields[5] == null || fields[5].equals(""))
                    user.setGender(null);
                else
                    user.setGender(fields[5]);
                if (fields[6] == null || fields[6].equals(""))
                    user.setAddress(null);
                else
                    user.setAddress(fields[6]);
                addToDb(user); // 添加到数据库
            } else { // 最后一个字段没有输入内容
                User user = new User(); // 封装要添加到数据库的对象
                if (fields[0] == null || fields[0].equals("")) {
                    System.out.println("**************此行没有主键，跳过**************");
                    continue;
                } else
                    user.setId(Integer.parseInt(fields[0]));
                if (fields[1] == null || fields[1].equals(""))
                    user.setName(null);
                else
                    user.setName(fields[1]);
                if (fields[2] == null || fields[2].equals(""))
                    user.setPassword(null);
                else
                    user.setPassword(fields[2]);
                if (fields[3] == null || fields[3].equals(""))
                    user.setNickName(null);
                else
                    user.setNickName(fields[3]);
                if (fields[4] == null || fields[4].equals(""))
                    ;
                else
                    user.setAge(Integer.parseInt(fields[4]));
                if (fields[5] == null || fields[5].equals(""))
                    user.setGender(null);
                else
                    user.setGender(fields[5]);
                addToDb(user); // 添加到数据库
            }
        }
        br.close();
        System.out.println("**************录入完毕**************");
    }

/**
     * 向数据库中添加记录
     * 
     * @param user
     */
    private static void addToDb(User user) {
        // 判断数据库当中是否存在该条记录
        if (isRepeat(user))
            return;
        Connection conn = null;
        PreparedStatement pstmt = null;
        try {
            conn = DbUtil.getConnection();
            String sql = "insert into t_user(dbid, name, pwd, nickName, age, gender, addres) values(?, ?, ?, ?, ?, ?, ?)";
            pstmt = conn.prepareStatement(sql);
            pstmt.setInt(1, user.getId());
            pstmt.setString(2, user.getName());
            pstmt.setString(3, user.getPassword());
            pstmt.setString(4, user.getNickName());
            pstmt.setInt(5, user.getAge());
            pstmt.setString(6, user.getGender());
            pstmt.setString(7, user.getAddress());
            pstmt.execute();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

/**
     * 判断数据库中是否有该条记录
     * 
     * @param user
     * @return
     */
    private static boolean isRepeat(User user) {
        ResultSet rs = null;
        Connection conn = null;
        PreparedStatement pstmt = null;
        try {
            conn = DbUtil.getConnection();
            String sql = "select * from t_user";
            pstmt = conn.prepareStatement(sql);
            rs = pstmt.executeQuery();
            while (rs.next())
                if (rs.getString("dbid").equals(user.getId()))
                    return true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }