JAVA项目后台查询的数据生成Excel表格并提供给用户下载

最近需求：把Hive语句查询出来的数据在生成Excel，并提供给用户下载

想了2方案，第一种自然就是用js，但是有个问题，浏览器需要安装插件

第二种方法就是在后台生成文件，存放到服务器里，通过数据集来把数据塞到文件里，后缀名为.xls，这种方法明显靠谱得多嘛

第一步：JSP页面添加一个按钮，给个URL能跳转到相应的Controller的方法里

"<a href=\"/hiveadmin/exportExcel.do?id="+haJob.getId()+"&jobType="+ haJob.getJobType()+"&jobName="+ haJob.getJobName() + " \" title='导出数据'>导出数据</a>"

传了三个参数 ，id,name,type，id用于后面查找到底是哪个数据(因为查询出来的数据我先保存为txt文件，这个txt文件带有固定编号，id用于定位txt文件)，name就是生成的excel的名字，jobType用来获取excel表格表头的名字(这些表头名称存放在一个单独的表中)

第二步：在Controller里新加方法，用于查询数据

@RequestMapping(value = "/exportExcel")
    public ModelAndView exportExcel(HttpServletRequest request, HttpServletResponse response, ModelMap model) {
    	
    	List<Map<String, Object>> rowlst = new ArrayList<Map<String, Object>>();
    	
    	String jobId = request.getParameter("id");
    	String jobType = request.getParameter("jobType") ;
    	String jobName = request.getParameter("jobName") ;
    	
    	// 读取数据文本内容
    	File file = new File(HiveAdminJobConstants.JOB_WORK_DATA_DIR + "/data-" + jobId + ".txt");
        BufferedReader reader = null;
        try {
            reader = new BufferedReader(new FileReader(file));
            String line = null;
            int n = 0;
            // 一次读入一行，直到读入null为文件结束
            while ((line = reader.readLine()) != null) {
                String[] arryLine = line.split(",");
                Map<String, Object> map = new HashMap<String, Object>();
                for (int i = 0; i < arryLine.length; i++) {
                    map.put(String.valueOf(i), arryLine[i]);
                }
                rowlst.add(map);
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (reader != null) {
                try {
                    reader.close();
                } catch (IOException e1) {
                }
            }
        }
        
        if(jobName == null){
        	jobName = "未命名" ;
        }
        
        //获取表头
        HiveAdminJobTemplate template = hiveAdminJobTemplateService.getHiveAdminJobTemplateByUid(jobType);
        List<String> colTitleList = new ArrayList<String>();
        if (template != null) {
            String strListTitle = template.getShowListTitle();
            String[] arryListTitle = strListTitle.split(",");
            for (int i = 0; i < arryListTitle.length; i++) {
                colTitleList.add(arryListTitle[i]);
            }
        }
        
        model.put("colTitleList", colTitleList) ;
        model.put("exclName", jobName) ;
        model.put("list", rowlst) ;
    	
    	return new ModelAndView(new HiveAdminStatDateExportExcelView(), model);
    }

注意上面的方法的返回值类型：ModelAndView，不是String

return一个 new ModelAndView，事实上new的事HiveAdminStatDateExportExcelView()，我把参数model给他，model里存了数据集、文件名、表头名

那么HiveAdminStatDateExportExcelView()是什么呢，他其实是个类，是excel文件的生成策略

/**
 * 生成excel视图，可用excel工具打开或者保存 由ViewController的return new ModelAndView(viewExcel,
 * model)生成
 */
public class HiveAdminStatDateExportExcelView extends AbstractExcelView {
	@SuppressWarnings("unchecked")
	public void buildExcelDocument(Map model, HSSFWorkbook workbook, HttpServletRequest request,
			HttpServletResponse response) throws Exception {

		// 设置文件名称
		String jobName = model.get("exclName").toString() ;
		String excelName = jobName+".xls";
		// 设置sheet名称
		String sheetName = jobName;

		// 设置response方式,使执行此controller时候自动出现下载页面,而非直接使用excel打开
		response.setContentType("APPLICATION/OCTET-STREAM");
		response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(excelName, "UTF-8"));

		// Excel表头
		HSSFSheet sheet = workbook.createSheet(sheetName);

		// 产生标题列
		List<String> headerNamesList = new ArrayList<String>() ;
		headerNamesList = (List<String>) model.get("colTitleList") ;
		String[] headerNames = new String[headerNamesList.size()] ;
		for(int i=0; i<headerNamesList.size(); i++){
			headerNames[i] = headerNamesList.get(i) ;
		}
		SpringExcelUtil excelUtil = new SpringExcelUtil();
		excelUtil.setHeader(sheet, headerNames);

		//
		HSSFCellStyle cellStyle = workbook.createCellStyle();
		cellStyle.setDataFormat(HSSFDataFormat.getBuiltinFormat("mm/dd/yyyy"));

		// 填充数据
		List<Map<String, Object>> stuList = (List<Map<String, Object>>) model.get("list");
		for (Map<String, Object> map : stuList) {
			if (map != null) {
				String[] data = new String[map.size()];
				for(int j=0; j<map.size(); j++){
					data[j] = String.valueOf(map.get(String.valueOf(j))) ;
				}
				excelUtil.addRecord(sheet, data);
			}
		}
	}
}

完成以上步骤，就实现了excel表格后台生成，前台获取的功能了！

点击导出数据：

下载成功，打开看看：

OK。