com.moon.poi.excel.ProxyStyleBuilder Maven / Gradle / Ivy

Go to download
package com.moon.poi.excel;

import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.Workbook;

import java.util.function.BiConsumer;
import java.util.function.Consumer;

/**
 * 样式构建器
 *
 * @author moonsky
 */
class ProxyStyleBuilder extends ProxyBuilder {

    /**
     * 不包含字体
     */
    private final Consumer nonFontBuilder;
    /**
     * 包含字体
     */
    private final BiConsumer hasFontBuilder;

    ProxyStyleBuilder(Object classname, ProxyStyleBuilder builder) {
        super(classname);
        this.nonFontBuilder = builder.nonFontBuilder;
        this.hasFontBuilder = builder.hasFontBuilder;
    }

    ProxyStyleBuilder(Object classname, Consumer builder) {
        super(classname);
        this.nonFontBuilder = builder;
        this.hasFontBuilder = null;
    }

    ProxyStyleBuilder(Object classname, BiConsumer builder) {
        super(classname);
        this.hasFontBuilder = builder;
        this.nonFontBuilder = null;
    }

    /**
     * 构建样式
     *
     * @param workbook 工作表
     *
     * @return CellStyle 实例
     */
    @Override
    CellStyleProxy build(Workbook workbook) {
        CellStyle style = workbook.createCellStyle();
        if (hasFontBuilder != null) {
            Font font = workbook.createFont();
            hasFontBuilder.accept(style, font);
            if (style.getFontIndexAsInt() < 1) {
                style.setFont(font);
            }
        } else if (nonFontBuilder != null) {
            nonFontBuilder.accept(style);
        }
        return new CellStyleProxy(style);
    }
}