package com.chen.controller;

import java.io.File;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;

import com.chen.util.Csv_Util;

/**
 * 
 * @author chenguoji
 * @qq  375136957
 */
@Controller
public class CsvController {

	/**
	 * 跳转页面
	 * 
	 * @return
	 */
	@RequestMapping("/toCsv")
	public String toExcel() {

		return "csv";
	}

	/**
	 * 导出Excel 并下载
	 */
	@RequestMapping("/exportCsv")
	public void exportExcel(HttpServletRequest request,HttpServletResponse response) {
		 // 设置表格头
        Object[] head = {"客户姓名", "证件类型", "证件号码","客户类型", "客户经理"};
        List<Object> headList = Arrays.asList(head);
        // 设置数据
        List<List<Object>> dataList = new ArrayList<List<Object>>();
        List<Object> rowList = null;
        for (int i = 0; i < 10; i++) {
            rowList = new ArrayList<Object>();
            rowList.add("张三"+i);
            rowList.add("身份证");
            rowList.add("1234567899"+i);
            rowList.add("开发商");
            rowList.add("罗源");
            dataList.add(rowList);
        }
        // 导出文件路径
        String downloadFilePath = request.getServletContext().getRealPath("/csv");
        // 导出文件名称
        SimpleDateFormat df = new SimpleDateFormat("yyyyMMddHHmmss");
        String datetimeStr = df.format(new Date());
        String fileName = "客户列表-" + datetimeStr;
        //创建csv
		File file = Csv_Util.createCSVFile(headList, dataList, downloadFilePath, fileName);//生成CSV文件
		fileName = file.getName();
		Csv_Util.exportFile(response, downloadFilePath, fileName);//下载生成的CSV文件
	}
	
}