com.adrninistrator.javacg2.writer.WriterSupportSkip Maven / Gradle / Ivy

Go to download
package com.adrninistrator.javacg2.writer;

import com.adrninistrator.javacg2.util.JavaCG2FileUtil;

import java.io.Closeable;
import java.io.IOException;
import java.io.Writer;

/**
 * @author adrninistrator
 * @date 2023/4/26
 * @description: 包装后的Writer，若未写文件则不创建文件
 */
public class WriterSupportSkip implements Closeable {

    private final String filePath;
    private final boolean buffered;

    private Writer writer;

    public WriterSupportSkip(String filePath) {
        this(filePath, true);
    }

    public WriterSupportSkip(String filePath, boolean buffered) {
        this.filePath = filePath;
        this.buffered = buffered;
    }

    @Override
    public void close() throws IOException {
        if (writer != null) {
            writer.close();
        }
    }

    public void write(String data) throws IOException {
        if (writer == null) {
            writer = JavaCG2FileUtil.genBufferedWriter(filePath);
        }
        writer.write(data);
        if (!buffered) {
            // 不缓存时，每次写文件后flush
            writer.flush();
        }
    }
}