package com.yizhi.application.report.trainingproject;

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

import com.yizhi.core.application.context.TaskContext;
import com.yizhi.core.application.file.constant.FileConstant;
import com.yizhi.core.application.file.task.AbstractDefaultTask;
import com.yizhi.core.application.file.util.OssUpload;
import com.yizhi.training.application.feign.TrainingProjectReportClient;
import com.yizhi.training.application.vo.manage.ReportStudyTrainingProjectVo;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.apache.poi.ss.util.CellRangeAddress;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import com.baomidou.mybatisplus.plugins.Page;


/**
 * 按照项目统计
 * @author wangfeida
 *
 */


@Component
public class TrainingProjectGroupExport extends AbstractDefaultTask<String, Map<String, Object>> {

	private static final Logger logger = LoggerFactory.getLogger(TrainingProjectGroupExport.class);

	@Autowired
	private TrainingProjectReportClient trainingProjectReportClient;

	@Override
	protected String execute(Map<String, Object> map) {
		// TODO Auto-generated method stub
		String sheetName="培训项目-项目完成情况统计";
		String upLoadUrl = null;
		Long accountId = (Long) map.get("accountId");
		Long companyId = (Long) map.get("companyId");
		List<Long> orgIds = (List<Long>) map.get("orgIds");
		Long siteId = (Long) map.get("siteId");
		String startDate = (String) map.get("startDate");
		String endDate = (String) map.get("endDate");
		String kwd = (String) map.get("kwd");
		Long taskId=(Long) map.get("taskId");
		Date submitTime=(Date) map.get("submitTime");
		String serialNo=(String) map.get("serialNo");
		String taskName=(String) map.get("taskName");
		Integer pageNo = 2;
		Integer pageSize = 30;

		Integer maxNumOneSheet=50000;   //一个sheet最多有多少数据量
		Integer numSheet=0;             //有多少个sheet
		Integer curSheet=1;
		
        TaskContext taskContext = new TaskContext(taskId, serialNo, taskName,accountId, submitTime, siteId, companyId);// 任务id  任务名字      操作人id   操作时间
		working(taskContext);
		
		
		
		Page<ReportStudyTrainingProjectVo> page = trainingProjectReportClient.trainingProjectGroup(kwd, startDate, endDate, pageNo-1, pageSize, companyId, orgIds, siteId);
		List<ReportStudyTrainingProjectVo> list = new ArrayList<ReportStudyTrainingProjectVo>();
		
		if (page != null) {
			logger.info("培训项目-项目完成情况统计-下载-总记录数：{}", page.getTotal());
			list.addAll(page.getRecords());
			int num=page.getTotal()/pageSize+1;
			for (int i = 2; i <= num; i++) {
				page=trainingProjectReportClient.trainingProjectGroup(kwd, startDate, endDate, i, pageSize, companyId, orgIds, siteId);
				if(page == null || page.getRecords().size() == 0) {
					break;
				}
				list.addAll(page.getRecords());
			}
		}
		
		SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd");
		SimpleDateFormat df2 = new SimpleDateFormat("yyyyMMddHHmmss");
		
		
		try {
			// excel生成过程: excel-->sheet-->row-->cell
			// 第一步，创建一个Excel文件
			HSSFWorkbook wb = new HSSFWorkbook();
			if(list!=null&&list.size()>0) {
				numSheet = list.size() / maxNumOneSheet + 1;  //有这么多sheet
				for (int i = 1; i <= numSheet; i++) {
					// 最大限度是50000
					// 第二步，在webbook中添加一个sheet,对应Excel文件中的sheet
					HSSFSheet sheet = wb.createSheet(sheetName + i);
					// 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制short
					HSSFRow row = sheet.createRow((int) 0);
					// 第四步，创建单元格，并设置值表头 设置表头居中
					HSSFCellStyle style = wb.createCellStyle();
					style.setAlignment(HorizontalAlignment.CENTER); // 创建一个居中格式
					// 合并单元格
					CellRangeAddress cra = new CellRangeAddress(0, 0, 0, 7); // 起始行, 终止行, 起始列, 终止列
					sheet.addMergedRegion(cra);
					HSSFCell cell = row.createCell((short) 0);
					cell.setCellValue(sheetName);
					cell.setCellStyle(style);

					// 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制short
					HSSFRow row2 = sheet.createRow((int) 1);
					// 第四步，创建单元格
					CellRangeAddress cra2 = new CellRangeAddress(1, 1, 0, 3);
					sheet.addMergedRegion(cra2);
					HSSFCell cell2 = row2.createCell((short) 0);
					/**
					 * 统计周期只对培训项目起作用，但是对培训项目的参加人数和完成人数不起作用
					 */
					cell2.setCellValue("统计周期： " + startDate + "~" + endDate);

					HSSFRow row3 = sheet.createRow((int) 2);
					row3.createCell((short) 0).setCellValue("项目名称");
					row3.createCell((short) 1).setCellValue("开始时间");
					row3.createCell((short) 2).setCellValue("结束时间");
					row3.createCell((short) 3).setCellValue("可参加人数");
					row3.createCell((short) 4).setCellValue("实际参加人数");
					row3.createCell((short) 5).setCellValue("完成人数");
					row3.createCell((short) 6).setCellValue("完成率");
					// 第五步，写入实体数据 实际应用中这些数据从数据库得到，
					if (list != null && list.size() > 0) {
						ReportStudyTrainingProjectVo rscv = null;
						for (int j = 0; j < maxNumOneSheet; j++) {
							row = sheet.createRow(j + 3);
							int a = j + (i - 1) * maxNumOneSheet;
							if (a >= list.size()) {
								break;
							}
							rscv = list.get(a);
							// 第四步，创建单元格，并设置值
							row.createCell((short) 0).setCellValue(rscv.getName());
							if (rscv != null && rscv.getStartTime() != null) {
								row.createCell((short) 1).setCellValue(df.format(rscv.getStartTime()));
							}
							if (rscv != null && rscv.getEndTime() != null) {
								row.createCell((short) 2).setCellValue(df.format(rscv.getEndTime()));
							}
							row.createCell((short) 3).setCellValue(rscv.getTotalAsk());
							row.createCell((short) 4).setCellValue(rscv.getTotalIn());
							row.createCell((short) 5).setCellValue(rscv.getTotalFinish());
							row.createCell((short) 6).setCellValue(rscv.getAvgFinish() + "%");
							//taskDetail(taskContext.getTaskId(), sheetName+"导出数据第+" + (i+1) + "+行成功");
						}
					}
				}
			}
			String fileName = sheetName + df2.format(submitTime) + ".xls";
			String requestPath = FileConstant.SAVE_PATH;
			File fileDir = new File(requestPath);
			if (!fileDir.exists()) {
				fileDir.mkdir();
			}
			String path = requestPath +File.separator+fileName;
			FileOutputStream os = null;
			File file = null;
			try {
				os = new FileOutputStream(path);
				wb.write(os);
				// 阿里云返回url
				upLoadUrl = OssUpload.upload(path, fileName);
				file = new File(path);
				success(taskContext, "成功", upLoadUrl);
			} catch (Exception e1) {
				e1.printStackTrace();
				fail(taskContext, "写入过程中发生错误");
				logger.error("上传过程中发生错误，请查看日志");
			} finally {
				if (os != null) {
					os.close();
				}
				if (wb != null) {
					wb.close();
				}
				if (file != null) {
					file.delete();
				}
			}

		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
			fail(taskContext, sheetName+"导出过程中发生错误，请查看日志"+e.getMessage());
			logger.error(sheetName+"导出过程中发生错误，请查看日志");
			
		}
		return upLoadUrl;
	 }
}
