GuideboardInfoDataToolsImpl.java 4.88 KB
package com.bsth.service.schedule.datatools;

import com.bsth.service.schedule.exception.ScheduleException;
import com.bsth.service.schedule.utils.DataToolsProperties;
import com.bsth.service.schedule.utils.DataToolsService;
import jxl.Cell;
import jxl.Sheet;
import jxl.Workbook;
import jxl.write.Label;
import jxl.write.WritableSheet;
import jxl.write.WritableWorkbook;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.stereotype.Service;

import java.io.File;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.Map;

/**
 * Created by xu on 17/5/16.
 */
@EnableConfigurationProperties(DataToolsProperties.class)
@Service(value = "gbInfo_dataTool")
public class GuideboardInfoDataToolsImpl implements DataToolsService {
    /** 日志记录器 */
    private final static Logger LOGGER = LoggerFactory.getLogger(GuideboardInfoDataToolsImpl.class);

    @Autowired
    @Qualifier(value = "dataToolsServiceImpl")
    private DataToolsService dataToolsService;

    @Autowired
    private DataToolsProperties dataToolsProperties;

    @Override
    public File uploadFile(String filename, byte[] filedata) throws ScheduleException {
        try {
            // 对上传的excel文件做处理,将第一个sheet名字设定为工作表1
            File file = dataToolsService.uploadFile(filename, filedata);
            Workbook workbook = Workbook.getWorkbook(file);
            Sheet sheet = workbook.getSheet(0);

            File fileCal = new File(file.getAbsolutePath() + "_sheetChange.xls");
            WritableWorkbook writableWorkbook = Workbook.createWorkbook(fileCal);
            WritableSheet writableSheet = writableWorkbook.createSheet("工作表1", 0);
            for (int i = 0; i < sheet.getRows(); i++) {
                Cell[] cells = sheet.getRow(i);
                for (int j = 0; j < cells.length; j++) {
                    writableSheet.addCell(new Label(j, i, cells[j].getContents()));
                }
            }
            writableWorkbook.write();
            writableWorkbook.close();

            return fileCal;

        } catch (Exception exp) {
            throw new ScheduleException(exp);
        }
    }

    @Override
    public void importData(File file, Map<String, Object> params) throws ScheduleException {
        try {
            LOGGER.info("//---------------- 导入路牌信息 start... ----------------//");
            // 创建ktr转换所需参数
            Map<String, Object> ktrParms = new HashMap<>();
            File ktrFile = new File(this.getClass().getResource(
                    dataToolsProperties.getGuideboardsDatainputktr()).toURI());

            // 通用参数,转换文件路径,excel输入文件路径,错误输出文件路径
            ktrParms.put("transpath", ktrFile.getAbsolutePath());
            ktrParms.put("filepath", file.getAbsolutePath());
            ktrParms.put("erroroutputdir", dataToolsProperties.getTransErrordir());

            ktrParms.putAll(params);

            dataToolsService.importData(file, ktrParms);

            LOGGER.info("//---------------- 导入路牌信息 success... ----------------//");
        } catch (Exception exp) {
            LOGGER.info("//---------------- 导入路牌信息 failed... ----------------//");

            StringWriter sw = new StringWriter();
            exp.printStackTrace(new PrintWriter(sw));
            LOGGER.info(sw.toString());

            throw new ScheduleException(exp.getMessage());
        }
    }

    @Override
    public File exportData(Map<String, Object> params) throws ScheduleException {
        try {
            LOGGER.info("//---------------- 导出路牌信息 start... ----------------//");
            // 创建ktr转换所需参数
            Map<String, Object> ktrParms = new HashMap<>();
            File ktrFile = new File(this.getClass().getResource(
                    dataToolsProperties.getGuideboardsDataoutputktr()).toURI());

            // 通用参数,转换文件路径,excel输出文件名
            ktrParms.put("transpath", ktrFile.getAbsolutePath());
            ktrParms.put("filename", "路牌信息_download-");

            ktrParms.putAll(params);

            File file = dataToolsService.exportData(ktrParms);

            LOGGER.info("//---------------- 导出路牌信息 success... ----------------//");

            return file;

        } catch (Exception exp) {
            LOGGER.info("//---------------- 导出路牌信息 failed... ----------------//");

            StringWriter sw = new StringWriter();
            exp.printStackTrace(new PrintWriter(sw));
            LOGGER.info(sw.toString());

            throw new ScheduleException(exp.getMessage());
        }
    }
}