| | |
|---|
| | | package com.mes.largenscreen.service.impl; |
|---|
| | | |
|---|
| | | import com.alibaba.excel.EasyExcel; |
|---|
| | | import com.alibaba.excel.support.ExcelTypeEnum; |
|---|
| | | import com.baomidou.mybatisplus.extension.plugins.pagination.Page; |
|---|
| | | import com.mes.largenscreen.entity.DailyProductionData; |
|---|
| | | import com.mes.largenscreen.entity.DailyProductionVO; |
|---|
| | | import com.mes.largenscreen.entity.DateRequest; |
|---|
| | | import com.mes.largenscreen.mapper.LargenScreenMapper; |
|---|
| | | import com.mes.largenscreen.service.LargenScreenService; |
|---|
| | | import com.mes.tools.PageUtil; |
|---|
| | | import com.mes.utils.excel.ExcelMergeStrategy; |
|---|
| | | import com.mes.utils.excel.ExcelUtil; |
|---|
| | | import org.springframework.stereotype.Service; |
|---|
| | | import org.springframework.util.StringUtils; |
|---|
| | | |
|---|
| | | import javax.annotation.Resource; |
|---|
| | | import javax.servlet.http.HttpServletResponse; |
|---|
| | | import java.io.IOException; |
|---|
| | | import java.text.SimpleDateFormat; |
|---|
| | | import java.util.Calendar; |
|---|
| | | import java.util.Date; |
|---|
| | |
|---|
| | | |
|---|
| | | @Resource |
|---|
| | | private LargenScreenMapper largenScreenMapper; |
|---|
| | | @Resource |
|---|
| | | private HttpServletResponse response; |
|---|
| | | |
|---|
| | | @Override |
|---|
| | | public Page<DailyProductionVO> queryDailyProduction(DateRequest query) { |
|---|
| | |
|---|
| | | return list; |
|---|
| | | } |
|---|
| | | |
|---|
| | | @Override |
|---|
| | | public void exportDailyProduction(DateRequest query) { |
|---|
| | | try { |
|---|
| | | if (StringUtils.isEmpty(query.getBeginDate())) { |
|---|
| | | SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd"); |
|---|
| | | Calendar calendar = Calendar.getInstance(); |
|---|
| | | calendar.add(Calendar.DAY_OF_MONTH, -6); |
|---|
| | | query.setBeginDate(sdf.format(calendar.getTime())); |
|---|
| | | query.setEndDate(sdf.format(new Date())); |
|---|
| | | } |
|---|
| | | ExcelUtil.setExcelResponseProp(response, "供应商列表-合并"); |
|---|
| | | List<DailyProductionData> providerList = largenScreenMapper.exportDailyProduction(query.getBeginDate(), query.getEndDate()); |
|---|
| | | EasyExcel.write(response.getOutputStream()) |
|---|
| | | .head(DailyProductionData.class) |
|---|
| | | .registerWriteHandler(new ExcelMergeStrategy(DailyProductionData.class)) |
|---|
| | | .excelType(ExcelTypeEnum.XLSX).sheet().doWrite(providerList); |
|---|
| | | } catch (IOException e) { |
|---|
| | | throw new RuntimeException(e); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | @Override |
|---|
| | | public List<DailyProductionData> queryProduction(DateRequest query) { |
|---|
| | | if (StringUtils.isEmpty(query.getBeginDate())) { |
|---|
| | | SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd"); |
|---|
| | | Calendar calendar = Calendar.getInstance(); |
|---|
| | | calendar.add(Calendar.DAY_OF_MONTH, -6); |
|---|
| | | query.setBeginDate(sdf.format(calendar.getTime())); |
|---|
| | | query.setEndDate(sdf.format(new Date())); |
|---|
| | | } |
|---|
| | | return largenScreenMapper.exportDailyProduction(query.getBeginDate(), query.getEndDate()); |
|---|
| | | |
|---|
| | | |
|---|
| | | } |
|---|
| | | |
|---|
| | | |
|---|
| | | } |
|---|
