导入csv文件到数据库

时间:2023-01-20 06:29:18

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

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

导入csv文件到数据库

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

导入csv文件到数据库

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

导入csv文件到数据库

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

下面开始代码:(数据库连接部分省略)

 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;
     }