seven.savewapper.wapperRef.sysWppers.ResExportMap Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ExcelReads Show documentation
A ExcelReads Utils By Seven
There is a newer version: 1.0.4
package seven.savewapper.wapperRef.sysWppers;
//=======================================================
//		          .----.
//		       _.'__    `.
//		   .--(^)(^^)---/!\
//		 .' @          /!!!\
//		 :         ,    !!!!
//		  `-..__.-' _.-\!!!/
//		        `;_:    `"'
//		      .'"""""`.
//		     /,  ya ,\\
//		    //狗神保佑\\
//		    `-._______.-'
//		    ___`. | .'___
//		   (______|______)
//=======================================================

import org.apache.poi.ss.usermodel.*;
import seven.savewapper.wapperRef.SaveExcelObject;
import seven.util.ExcelTool;

import java.io.OutputStream;
import java.sql.ResultSet;
import java.util.*;

/**
 * [Zhihu]https://www.zhihu.com/people/Sweets07
 * [Github]https://github.com/MatrixSeven
 * Created by seven on 2016/11/30.
 */
@SuppressWarnings("all")
public class ResExportMap extends SaveExcelObject {

    public ResExportMap(List list) {
        super(list);
    }

    public ResExportMap(List list, String path) {
        super(list, path);
    }

    public ResExportMap(ResultSet resultSet, String path) {
        super(resultSet, path);
    }

    public ResExportMap(ResultSet resultSet) {
        super(resultSet);
    }


    @Override
    @Deprecated
    public void Save() throws Exception {
        OutputStream out = createStream();
        createWK();
        tryCreateCellStyle();
        checkData();
        if (c != null) {
            list.sort(c);
        }
        Sheet sheet = wk.createSheet("sheet1");
        sheet.setDefaultColumnWidth((short) 15);
        // 生成一个样式
        CellStyle style = wk.createCellStyle();
        // 设置这些样式

        style.setAlignment(HorizontalAlignment.CENTER);
        Map m = list.get(0);
        Set setTitle = m.keySet();
        Iterator it = setTitle.iterator();
        String t;
        String[] title;
        List l = new ArrayList();

        if (!anyColByKey.isEmpty()) {
            for (Map o : list) {
                if (!filter.test(o)) {
                    HashSet strings = new HashSet<>(setTitle);
                    strings.removeAll(setTitle);
                    if (!strings.isEmpty()) {
                        throw new Exception("字段名无效");
                    }
                }
            }
            title = anyColByKey.toArray(new String[anyColByKey.size()]);
        } else {
            while (it.hasNext()) {
                t = it.next();
                if (!filterColByKey.contains(t)) {
                    l.add(t);
                }
            }
            title = l.toArray(new String[l.size()]);
        }

        Row row = sheet.createRow(0);

        initTitle(title, row, style);

        int index = 0;
        for (
                Map o : list) {
            if (!filter.test(o)) {
                continue;
            }
            row = sheet.createRow(++index);
            process.accept(o);//加工每一行
            for (int i = 0; i < title.length; i++) {
                Cell cell = row.createCell(i);
                cell.setCellStyle(style);
                if (cellStyle.containsKey(title[i])) {
                    cell.setCellStyle(cellStyle.get(title[i]).getRealyStyle());
                }
                cell.setCellValue(o.get(title[i]));
            }
        }
        try {
            wk.write(out);
            out.flush();

        } finally {
            ExcelTool.Close(wk, out);
        }
    }
}