springboot 之 使用poi進行數據的導出(一)

使用的是idea+restful風格

第一:引入依賴為:

        <!--poi-->
        <dependency>
            <groupId>org.apache.xmlbeans</groupId>
            <artifactId>xmlbeans</artifactId>
            <version>2.6.0</version>
        </dependency>
        <dependency>
            <groupId>dom4j</groupId>
            <artifactId>dom4j</artifactId>
            <version>1.6.1</version>
        </dependency>
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.9</version>
        </dependency>
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>3.9</version>
        </dependency>
        <!--poi-->

第二步:

上傳文件的工具類

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
 
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
 
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
 
/**
 * @author	qingmu
 * @version 創建時間：2017年12月28日 下午4:53:29 
 * 類說明: POI導出工具類
 */
public class ExportPOIUtils {
 
	//參數說明:  fileName：文件名   projects：對象集合  columnNames： 列名   keys： map中的key
	public static void start_download(HttpServletResponse response, String fileName, List<?> projects,
			String[] columnNames, String[] keys) throws IOException {
		
		//將集合中對象的屬性  對應到  List<Map<String,Object>>
		List<Map<String,Object>> list=createExcelRecord(projects, keys);
 
		ByteArrayOutputStream os = new ByteArrayOutputStream();
		try {
			//將轉換成的Workbook對象通過流形式下載
			createWorkBook(list,keys,columnNames).write(os);
		} catch (IOException e) {
			e.printStackTrace();
		}
		byte[] content = os.toByteArray();
		InputStream is = new ByteArrayInputStream(content);
		// 設置response參數，可以打開下載頁面
		response.reset();
		response.setContentType("application/vnd.ms-excel;charset=utf-8");
		response.setHeader("Content-Disposition", "attachment;filename="+ new String((fileName + ".xls").getBytes(), "iso-8859-1"));
		ServletOutputStream out = response.getOutputStream();
		BufferedInputStream bis = null;
		BufferedOutputStream bos = null;
		try {
			bis = new BufferedInputStream(is);
			bos = new BufferedOutputStream(out);
			byte[] buff = new byte[2048];
			int bytesRead;
			while (-1 != (bytesRead = bis.read(buff, 0, buff.length))) {
				bos.write(buff, 0, bytesRead);
			}
		} catch (final IOException e) {
			throw e;
		} finally {
			if (bis != null)
				bis.close();
			if (bos != null)
				bos.close();
		}
	}
	
	private static List<Map<String, Object>> createExcelRecord(List<?> projects, String[] keys) {
		List<Map<String, Object>> listmap = new ArrayList<Map<String, Object>>();
		Map<String, Object> map = new HashMap<String, Object>();
		map.put("sheetName", "sheet");
		listmap.add(map);
		Object project=null;
		for (int j = 0; j < projects.size(); j++) {
			project=projects.get(j);
			Map<String, Object> mapValue = new HashMap<String, Object>();
			for(int i=0; i<keys.length; i++){
				mapValue.put(keys[i], getFieldValueByName(keys[i], project));
			}
 
			listmap.add(mapValue);
		}
		return listmap;
	}
	/** 
	 * 利用反射  根據屬性名獲取屬性值 
	 * */  
	private static Object getFieldValueByName(String fieldName, Object o) {  
		try {    
			String firstLetter = fieldName.substring(0, 1).toUpperCase();    
			String getter = "get" + firstLetter + fieldName.substring(1);    
			Method method = o.getClass().getMethod(getter, new Class[] {});    
			Object value = method.invoke(o, new Object[] {});    
			return value;    
		} catch (Exception e) {    
			e.printStackTrace();  
			return null;    
		}    
	}
    /**
     * 創建excel文檔對象
     * @param keys list中map的key數組集合
     * @param columnNames excel的列名
     * */
	private static Workbook createWorkBook(List<Map<String, Object>> list,String []keys,String columnNames[]) {
        // 創建excel工作簿
        Workbook wb = new HSSFWorkbook();
        // 創建第一個sheet（頁），並命名
        Sheet sheet = wb.createSheet(list.get(0).get("sheetName").toString());
        // 手動設置列寬。第一個參數表示要為第幾列設；，第二個參數表示列的寬度，n為列高的像素數。
        for(int i=0;i<keys.length;i++){
            sheet.setColumnWidth((short) i, (short) (35.7 * 150));
        }
 
        // 創建第一行
        Row row = sheet.createRow((short) 0);
 
        // 創建兩種單元格格式
        CellStyle cs = wb.createCellStyle();
        CellStyle cs2 = wb.createCellStyle();
 
        // 創建兩種字體
        Font f = wb.createFont();
        Font f2 = wb.createFont();
 
        // 創建第一種字體樣式（用於列名）
        f.setFontHeightInPoints((short) 10);
        f.setColor(IndexedColors.BLACK.getIndex());
        f.setBoldweight(Font.BOLDWEIGHT_BOLD);
 
        // 創建第二種字體樣式（用於值）
        f2.setFontHeightInPoints((short) 10);
        f2.setColor(IndexedColors.BLACK.getIndex());
 
        // 設置第一種單元格的樣式（用於列名）
        cs.setFont(f);
        cs.setBorderLeft(CellStyle.BORDER_THIN);
        cs.setBorderRight(CellStyle.BORDER_THIN);
        cs.setBorderTop(CellStyle.BORDER_THIN);
        cs.setBorderBottom(CellStyle.BORDER_THIN);
        cs.setAlignment(CellStyle.ALIGN_CENTER);
 
        // 設置第二種單元格的樣式（用於值）
        cs2.setFont(f2);
        cs2.setBorderLeft(CellStyle.BORDER_THIN);
        cs2.setBorderRight(CellStyle.BORDER_THIN);
        cs2.setBorderTop(CellStyle.BORDER_THIN);
        cs2.setBorderBottom(CellStyle.BORDER_THIN);
        cs2.setAlignment(CellStyle.ALIGN_CENTER);
        //設置列名
        for(int i=0;i<columnNames.length;i++){
            Cell cell = row.createCell(i);
            cell.setCellValue(columnNames[i]);
            cell.setCellStyle(cs);
        }
        //設置每行每列的值
        for (short i = 1; i < list.size(); i++) {
            // Row 行,Cell 方格 , Row 和 Cell 都是從0開始計數的
            // 創建一行，在頁sheet上
            Row row1 = sheet.createRow((short) i);
            // 在row行上創建一個方格
            for(short j=0;j<keys.length;j++){
                Cell cell = row1.createCell(j);
                cell.setCellValue(list.get(i).get(keys[j]) == null?" ": list.get(i).get(keys[j]).toString());
                cell.setCellStyle(cs2);
            }
        }
        return wb;
    }
 
}

　第三步:controller層的編寫

import com.example.uploaddemo.util.ExportPOIUtils;
import org.springframework.web.bind.annotation.RequestMapping;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * @author qingmu
 * @date 
 */
public class UserController {
    @RequestMapping("exportList")
    public void exportList(HttpServletResponse response, String ids) {

        String fileName = "人員檔案列表";

        List<User> users = new ArrayList<User>();

        // 列名
        String columnNames[] = {"ID", "姓名", "性別", "所屬部門", "所屬單位", "郵箱",
                "電話", "手機", "學歷/學位", "專業/專科方向", "直屬上級", "賬號鎖定"};
        // map中的key
        String keys[] = {"id", "userName", "gender", "dept", "unit", "email",
                "tel", "phone", "degree", "major", "parentName", "isLocked"};
        try {
            ExportPOIUtils.start_download(response, fileName, users, columnNames, keys);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

這里少了一個user實體類,你可以直接替換成你的要導出的實體類,就ok了.

簡潔的代碼,又高效的代碼.