1. 程式人生 > >用Java POI操作Excel,讀取資料匯入DB2資料庫

用Java POI操作Excel,讀取資料匯入DB2資料庫

Java操作Excel現在基本有兩種方法,就是使用JXL或POI,這兩者各有利弊,互有長短。
下載地址:POI  http://jakarta.apache.org/poi/index.html
                    JXL  http://www.andykhan.com/jexcelapi/
最近專案需要讀取Excel,將資料匯入資料庫儲存,最終可以用報表讀取資料庫並顯示,比較簡單。查了一下兩者的資料,覺的也無所謂,我需要的功能比較簡單,只要能讀取Excel資料就行,最後決定使用POI,比較信任apache
使用果然很簡單,上網找了個原始碼,Copy過來稍微改改就能用了。
下面付的程式碼是一個Demo,功能點有兩個:一就是POI讀取Excel,二是DB2資料庫的連線和SQL執行。
現在的問題就是這樣做的話程式碼太死,針對一張Excel表就要編寫一個專門的轉換類,太麻煩了,要考慮怎麼實現一個通用轉換類。

import java.io.FileInputStream;
import java.io.IOException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;

public class POITest {
 private static Connection conn = null;
 private static Statement stmt = null;
 private static boolean connectDB2() {

  String url = "";
  String username = "username";
  String password = "password";

  //載入驅動程式以連線資料庫
  try {
   //新增類庫驅動包db2jcc.jar和db2jcc_license_cu.jar
   Class.forName("com.ibm.db2.jcc.DB2Driver");
   url = "jdbc:db2://192.168.0.1:50000/dbname";
   //新增類庫驅動包db2java.jar
   //Class.forName("com.ibm.db2.jdbc.app.DB2Driver").newInstance();
   //url = "jdbc:db2:njtcdata";
   conn = DriverManager.getConnection(url, username, password);
   stmt = conn.createStatement();
  }
  //捕獲載入驅動程式異常
  catch (ClassNotFoundException cnfex) {
   System.err.println("裝載JDBC驅動程式失敗。");
   cnfex.printStackTrace();
   return false;
  }
  //捕獲連線資料庫異常
  catch (SQLException sqlex) {
   System.err.println("無法連線資料庫");
   sqlex.printStackTrace();
   //System.exit(1); // terminate program
   return false;
  }
  return true;
 }

 private static boolean readExcelToDB2() {
  POIFSFileSystem fs = null;
  HSSFWorkbook wb = null;
  try {
   fs = new POIFSFileSystem(new FileInputStream("c://test.xls"));
   wb = new HSSFWorkbook(fs);
  } catch (IOException e) {
   e.printStackTrace();
   return false;
  }
  HSSFSheet sheet = wb.getSheetAt(0);
  HSSFRow row = null;
  HSSFCell cell = null;
  String name = "";
  int id = 0;
  int rowNum, cellNum;
  int i;
  rowNum = sheet.getLastRowNum();
  for (i = 0; i <= rowNum; i++) {
   row = sheet.getRow(i);
   //cellNum = row.getLastCellNum();
   cell = row.getCell((short) 0);
   name = cell.getStringCellValue();
   cell = row.getCell((short) 1);
   id = (int) cell.getNumericCellValue();
   String sql = "insert into TEST(ID, NAME) values(" + id + ",'" + name + "')";
   try {
    stmt.executeUpdate(sql);
   } catch (SQLException e1) {
    e1.printStackTrace();
    return false;
   }
  }
  return true;
 }

 public static void main(String[] args) {
  if (connectDB2()==true){
   if (readExcelToDB2()==true)
    System.out.println("資料匯入成功");
   else
    System.out.println("資料匯入失敗");
  }
  else{
   System.out.println("資料庫連線失敗");
  }
 }
}