kr.jm.metric.input.StdinLineInput Maven / Gradle / Ivy

Go to download
package kr.jm.metric.input;

import kr.jm.metric.config.input.StdinLineInputConfig;
import kr.jm.metric.data.Transfer;
import kr.jm.utils.StdInLineConsumer;
import lombok.ToString;

import java.util.Optional;
import java.util.function.Consumer;

@ToString(callSuper = true)
public class StdinLineInput extends AbstractInput {
    private StdInLineConsumer stdInLineConsumer;

    public StdinLineInput(String inputId) {
        this(new StdinLineInputConfig(inputId));
    }

    public StdinLineInput(StdinLineInputConfig inputConfig) {
        super(inputConfig);
    }

    @Override
    protected void startImpl(Consumer> inputConsumer) {
        this.stdInLineConsumer =
                new StdInLineConsumer(s -> inputConsumer.accept(newTransfer(s)))
                        .consumeStdIn();
    }

    @Override
    protected void closeImpl() {
        Optional.ofNullable(this.stdInLineConsumer)
                .ifPresent(StdInLineConsumer::close);
    }

}