springboot 使用poi進(jìn)行數(shù)據(jù)的導(dǎo)出過程詳解

<!--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 創(chuàng)建時(shí)間：2017年12月28日 下午4:53:29
 * 類說明: POI導(dǎo)出工具類
 */
public class ExportPOIUtils {
 
  //參數(shù)說明: fileName：文件名  projects：對(duì)象集合 columnNames： 列名  keys： map中的key
  public static void start_download(HttpServletResponse response, String fileName, List<?> projects,
      String[] columnNames, String[] keys) throws IOException {
     
    //將集合中對(duì)象的屬性 對(duì)應(yīng)到 List<Map<String,Object>>
    List<Map<String,Object>> list=createExcelRecord(projects, keys);
 
    ByteArrayOutputStream os = new ByteArrayOutputStream();
    try {
      //將轉(zhuǎn)換成的Workbook對(duì)象通過流形式下載
      createWorkBook(list,keys,columnNames).write(os);
    } catch (IOException e) {
      e.printStackTrace();
    }
    byte[] content = os.toByteArray();
    InputStream is = new ByteArrayInputStream(content);
    // 設(shè)置response參數(shù)，可以打開下載頁面
    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;
  }
  /**
   * 利用反射 根據(jù)屬性名獲取屬性值
   * */ 
  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;  
    }  
  }
  /**
   * 創(chuàng)建excel文檔對(duì)象
   * @param keys list中map的key數(shù)組集合
   * @param columnNames excel的列名
   * */
  private static Workbook createWorkBook(List<Map<String, Object>> list,String []keys,String columnNames[]) {
    // 創(chuàng)建excel工作簿
    Workbook wb = new HSSFWorkbook();
    // 創(chuàng)建第一個(gè)sheet（頁），并命名
    Sheet sheet = wb.createSheet(list.get(0).get("sheetName").toString());
    // 手動(dòng)設(shè)置列寬。第一個(gè)參數(shù)表示要為第幾列設(shè)；，第二個(gè)參數(shù)表示列的寬度，n為列高的像素?cái)?shù)。
    for(int i=0;i<keys.length;i++){
      sheet.setColumnWidth((short) i, (short) (35.7 * 150));
    }
 
    // 創(chuàng)建第一行
    Row row = sheet.createRow((short) 0);
 
    // 創(chuàng)建兩種單元格格式
    CellStyle cs = wb.createCellStyle();
    CellStyle cs2 = wb.createCellStyle();
 
    // 創(chuàng)建兩種字體
    Font f = wb.createFont();
    Font f2 = wb.createFont();
 
    // 創(chuàng)建第一種字體樣式（用于列名）
    f.setFontHeightInPoints((short) 10);
    f.setColor(IndexedColors.BLACK.getIndex());
    f.setBoldweight(Font.BOLDWEIGHT_BOLD);
 
    // 創(chuàng)建第二種字體樣式（用于值）
    f2.setFontHeightInPoints((short) 10);
    f2.setColor(IndexedColors.BLACK.getIndex());
 
    // 設(shè)置第一種單元格的樣式（用于列名）
    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);
 
    // 設(shè)置第二種單元格的樣式（用于值）
    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);
    //設(shè)置列名
    for(int i=0;i<columnNames.length;i++){
      Cell cell = row.createCell(i);
      cell.setCellValue(columnNames[i]);
      cell.setCellStyle(cs);
    }
    //設(shè)置每行每列的值
    for (short i = 1; i < list.size(); i++) {
      // Row 行,Cell 方格 , Row 和 Cell 都是從0開始計(jì)數(shù)的
      // 創(chuàng)建一行，在頁sheet上
      Row row1 = sheet.createRow((short) i);
      // 在row行上創(chuàng)建一個(gè)方格
      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;
  }
}

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", "姓名", "性別", "所屬部門", "所屬單位", "郵箱",
        "電話", "手機(jī)", "學(xué)歷/學(xué)位", "專業(yè)/專科方向", "直屬上級(jí)", "賬號(hào)鎖定"};
    // 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();
    }
  }
}