• Home
• io.kestra.plugin
• plugin-datahub
• 0.19.0
• source code
• DataHubLogConsumer.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.kestra.plugin.datahub.DataHubLogConsumer Maven / Gradle / Ivy

package io.kestra.plugin.datahub;

import io.kestra.core.models.tasks.runners.AbstractLogConsumer;
import io.kestra.core.models.tasks.runners.PluginUtilsService;
import io.kestra.core.runners.RunContext;

import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

public class DataHubLogConsumer extends AbstractLogConsumer {

    private final RunContext runContext;
    private final AtomicInteger counter;

    public DataHubLogConsumer(RunContext runContext) {
        this.runContext = runContext;
        this.counter = new AtomicInteger(0);
    }

    @Override
    public void accept(String line, Boolean isStdErr) {
        if (line.contains("INFO")) {
            isStdErr = false;
        }

        Map outputs = PluginUtilsService.parseOut(line, runContext.logger(), runContext, isStdErr);
        if (outputs.isEmpty()) {
            super.outputs.put(String.valueOf(counter.incrementAndGet()), line);
        } else {
            super.outputs.putAll(outputs);
        }

        if (isStdErr) {
            this.stdErrCount.incrementAndGet();
        } else {
            this.stdOutCount.incrementAndGet();
        }
    }

}