com.exasol.projectkeeper.stream.CollectingConsumer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of project-keeper-core Show documentation

Project keeper is a tool that verifies and fixes project setups.

There is a newer version: 4.4.0

package com.exasol.projectkeeper.stream;

import java.io.IOException;
import java.time.Duration;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

import com.exasol.errorreporting.ExaError;

/**
 * This {@link StreamConsumer} collects all lines into a {@link StringBuilder} and provides a method for check if
 * reading has finished.
 */
public class CollectingConsumer implements StreamConsumer {
    private final CountDownLatch countDownLatch = new CountDownLatch(1);
    private final StringBuilder stringBuilder = new StringBuilder();

    @Override
    public void accept(final String line) {
        this.stringBuilder.append(line).append("\n");
    }

    @Override
    public void readFinished() {
        this.countDownLatch.countDown();
    }

    @Override
    public void readFailed(final IOException exception) {
        this.countDownLatch.countDown();
    }

    /**
     * Waits until the stream was read completely and returns the read content from the stream.
     *
     * @param timeout the maximum time to wait
     * @return the content collected from the stream
     * @throws InterruptedException if the current thread is interrupted while waiting
     */
    public String getContent(final Duration timeout) throws InterruptedException {
        final boolean result = this.countDownLatch.await(timeout.toMillis(), TimeUnit.MILLISECONDS);
        if (!result) {
            throw new IllegalStateException(ExaError.messageBuilder("E-PK-CORE-99").message(
                    "Stream reading did not finish after timeout of {{timeout}}. Content collected until now: {{content}}.",
                    timeout, this.stringBuilder.toString()).ticketMitigation().toString());
        }
        return this.stringBuilder.toString();
    }
}