能實現一對多關聯的通用EXCEL導入功能

      發一個本人寫的通用EXCEL導入功能，看了一下其他人發過的類似功能的導入組件，相比起來，我寫的這個優點在於，能實現數據的一對一、一對多關聯；直接使用SQL語句插入數據，不用通過實體類，執行效率會高一些。缺點在於，需要進行XML文件的配置，會有點麻煩。

      首先舉例說下使用方法。要導入以下兩個EXCEL文件

--------student.xls--------------

學號	姓名	性別
2010001	張三	男
2010002	李四	女

------------sex.xls---------------

代碼       標題

 1            男

 2            女

其中學生信息表中的性別字段爲外鍵，XML配置如下

EXCEL文件與數據表映射關係excel-table-mapping.xml

<?xml version="1.0" encoding="UTF-8"?> <mappings> <mapping><!-- 性別代碼 --> <import-name>sex</import-name><!-- 導入名稱 --> <first-row>1</first-row> <!-- 導入數據的開始行--> <last-row>0</last-row> <!-- 結束行，爲0表示到最後一行--> <table-name>t_sex</table-name><!-- 要導入的數據庫表名--> <xml-name>com/ronghz/excel/t_sex.xml</xml-name><!-- EXCEL行與字段的映射文件，與src的相對路徑--> </mapping> <mapping><!-- 學生資料 --> <import-name>student</import-name> <first-row>1</first-row> <last-row>0</last-row> <table-name>t_student</table-name> <xml-name>com/ronghz/excel/t_student.xml</xml-name> </mapping> </mappings>

EXCEL文件列與數據庫字段的映射關係

t_sex.xml

 <?xml version="1.0" encoding="UTF-8"?> <mappings> <mapping> <excel-order>0</excel-order> <!-- 數據在EXCEL文件中的列序號--> <excel-name>代碼</excel-name> <!-- 列名，可以爲空--> <column-name>code</column-name> <!-- 該列對應的字段名--> <column-type>varchar</column-type> <!-- 數據類型，可能的取值爲int,double,varchar,date,datetime,foreignkey，其它值都會當作varchar進行處理--> </mapping> <mapping> <excel-order>1</excel-order> <excel-name>內容</excel-name> <column-name>title</column-name> <column-type>varchar</column-type> </mapping> </mappings>

t_student.xml

 <?xml version="1.0" encoding="UTF-8"?> <mappings> <mapping> <excel-order>0</excel-order> <excel-name>學號</excel-name> <column-name>code</column-name> <column-type>varchar</column-type> </mapping> <mapping> <excel-order>1</excel-order> <excel-name>姓名</excel-name> <column-name>name</column-name> <column-type>varchar</column-type> </mapping> <mapping> <excel-order>2</excel-order> <excel-name>性別</excel-name> <column-name>sex</column-name> <column-type>foreignkey</column-type> <!-- 聲明該字段爲外鍵--> <outtable-name>t_sex</outtable-name> <!-- 外表的表名--> <outtable-key>t_sex_id</outtable-key> <!-- 外表的主鍵--> <outtable-value>title</outtable-value> <!-- 主表導入的數據對應於外表的哪個字段--> </mapping> </mappings>

調用相關類執行導入

public static void main(String[] args) { ImportExcel ie = new ImportExcel(); try { ie.execute("sex","D:/sex.xls"); ie.execute("student","D:/student.xls"); } catch (SQLException e) { e.printStackTrace(); } catch (ParseException e) { e.printStackTrace(); } }

 

建立數據庫

CREATE DATABASE `xls`; CREATE TABLE `xls`.`t_sex` ( `t_sex_id` int(10) unsigned NOT NULL auto_increment, `code` varchar(45) NOT NULL, `title` varchar(45) NOT NULL, PRIMARY KEY (`t_sex_id`) ); CREATE TABLE `xls`.`t_student` ( `t_student_id` int(10) unsigned NOT NULL auto_increment, `code` varchar(45) NOT NULL, `name` varchar(45) default NULL, `sex` int(10) unsigned default NULL, PRIMARY KEY (`t_student_id`);

      

主要代碼解釋：

一共包含5個類：

Conn.java                   數據庫連接類

FileMapping.java 

ImportExcel.java

TableMapping.java

TableMappingArray.java

 

ImportExcel.java的代碼package com.ronghz.excel; import java.io.*; import java.sql.*; import java.text.ParseException; import java.text.SimpleDateFormat; import jxl.*; import jxl.read.biff.BiffException; /** * Email:[e-mail] * Blog:http://blog.csdn.net/dreamdiary * 歡迎轉載；轉載時請註明出處 * */ public class ImportExcel { /** * 執行導入操作 * * @param importName導入的名稱 * @param excelPath文件路徑 * @return * @throws SQLException * @throws ParseException */ public boolean execute(String importName, String excelPath) throws SQLException, ParseException { FileMapping fileMapping = new FileMapping(importName);// 通過導入類型名，找到對應的表名、字段映射配置文件 String xmlName = fileMapping.getXmlName(); TableMappingArray mappingArray = new TableMappingArray(xmlName);// 把以上代碼找到的XML文件讀取到數組中 TableMapping[] array = mappingArray.getArray(); Conn conn = new Conn(); try { conn.beginTrans(); int haveFK = 0;// 外鍵關聯的數目 String addTempColumn = "ALTER TABLE " + fileMapping.getTableName(); String delTempColumn = "ALTER TABLE " + fileMapping.getTableName(); String[] updateFK = new String[array.length]; String preSql = "UPDATE $tableName$ t1 SET $tableKey$=(SELECT $outtablKey$ FROM $outtableName$ t2 WHERE t1.$tableKeyTmp$ = t2.$outtableValue$)" + " WHERE EXISTS ( SELECT 1 FROM $outtableName$ t3 WHERE t1.$tableKeyTmp$=t3.$outtableValue$)"; String columns = ""; String values = ""; for (int i = 0; i < array.length; i++) { if (array[i].getColumnType() == TableMapping.TYPE_FOREIGNKEY) {//如果該列類型屬於外鍵 // 生成插入數據的SQL語句 columns += array[i].getColumnName() + "_tmp,"; // 增加臨時字段用於保存導入的數據，導入完成後再將臨時字段刪除 addTempColumn += " ADD COLUMN " + array[i].getColumnName() + "_tmp VARCHAR(100),"; delTempColumn += " DROP COLUMN " + array[i].getColumnName() + "_tmp,"; // 把臨時字段的數據轉換成外鍵 String update = preSql.replace("$tableName$", fileMapping.getTableName()); update = update.replace("$tableKey$", array[i].getColumnName()); update = update.replace("$tableKeyTmp$", array[i].getColumnName() + "_tmp"); update = update.replace("$outtableName$", array[i].getOuttableName()); update = update.replace("$outtablKey$", array[i].getOuttableKey()); update = update.replace("$outtableValue$", array[i].getOuttableValue()); updateFK[haveFK++] = update; } else { columns += array[i].getColumnName() + ","; } values += "?,"; } if (haveFK > 0) {//如果有類型爲外鍵的列 conn.executeUpdate(addTempColumn.substring(0, addTempColumn.length() - 1)); } //讀取EXCEL文件的內容 String insertSQL = "INSERT INTO " + fileMapping.getTableName() + " (" + columns.substring(0, columns.length() - 1)+ ") " + "VALUES (" + values.substring(0, values.length() - 1) + ")"; Workbook book = null; book = Workbook.getWorkbook(new java.io.File(excelPath)); jxl.Sheet sheet = book.getSheet(0); int sheetRows = sheet.getRows() > fileMapping.getLastRow() ? sheet.getRows() : fileMapping.getLastRow(); Cell cell; PreparedStatement ps = conn.prepareStatement(insertSQL); // for (int row = fileMapping.getFirstRow(), count = 0; row < sheetRows; row++, count++) { for (int i = 0; i < array.length; i++) { cell = sheet.getCell(array[i].getExcelOrder(), row); //根據數據庫類型進行轉換 switch (array[i].getColumnType()) { case TableMapping.TYPE_INT: ps.setInt(i + 1, cell.getContents().equals("") ? 0 : Integer.parseInt(cell.getContents())); break; case TableMapping.TYPE_DOUBLE: ps.setDouble(i + 1, cell.getContents().equals("") ? 0 : Double.parseDouble(cell.getContents())); break; case TableMapping.TYPE_DATE: ps.setTimestamp(i + 1, new Timestamp(parse(cell.getContents(), array[i].getColumnFormat()).getTime())); break; case TableMapping.TYPE_DATETIME: ps.setTimestamp(i + 1, new Timestamp(parse(cell.getContents(), array[i].getColumnFormat()).getTime())); break; case TableMapping.TYPE_FOREIGNKEY:// 外鍵字段作爲字符串類型插入到臨時字段中 ps.setString(i + 1, cell.getContents()); break; case TableMapping.TYPE_STRING: ps.setString(i + 1, cell.getContents()); break; } } ps.addBatch(); if (count > 500) { count = 0; ps.executeBatch(); } } ps.executeBatch(); if (haveFK > 0) {// 把臨時字段的數據轉換成外鍵 for (int i = 0; i < haveFK; i++) { conn.executeUpdate(updateFK[i]); } //刪除臨時字段 conn.executeUpdate(delTempColumn.substring(0, delTempColumn.length() - 1)); } conn.commit(); return true; } catch (BiffException ex) { System.out.println("Read Excel file failed!"); System.out.println("File name: " + excelPath); } catch (IOException ex) { System.out.println("Read Excel file failed!"); System.out.println("File name: " + excelPath); } catch (SQLException e) { conn.rollback(); System.out.println("DateBase ERROR!"); e.printStackTrace(); throw e; } finally { if (conn != null) { conn.close(); conn = null; } } return true; } public java.util.Date parse(String time, String format) throws ParseException { if (time == null) { return null; } else { SimpleDateFormat sdf = new SimpleDateFormat(format); java.util.Date d = sdf.parse(time); return d; } } public static void main(String[] args) { ImportExcel ie = new ImportExcel(); try { ie.execute("sex","D:/sex.xls"); ie.execute("student","D:/student.xls"); } catch (SQLException e) { e.printStackTrace(); } catch (ParseException e) { e.printStackTrace(); } } }

 

 

      如果要應用於WEB項目，只需要增加上傳文件的操作，並把上傳後的文件路徑傳到execute()函數中就可以了。

      代碼比較多，不全部貼出來了，需要的請到以下地址下載http://download.csdn.net/source/2464234