wtf.metio.yosql.codegen.orchestration.DefaultTimer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of yosql-codegen Show documentation

There is a newer version: 2023.5.3

/*
 * This file is part of yosql. It is subject to the license terms in the LICENSE file found in the top-level
 * directory of this distribution and at https://creativecommons.org/publicdomain/zero/1.0/. No part of yosql,
 * including this file, may be copied, modified, propagated, or distributed except according to the terms contained
 * in the LICENSE file.
 */
package wtf.metio.yosql.codegen.orchestration;

import org.slf4j.cal10n.LocLogger;
import wtf.metio.yosql.codegen.lifecycle.TimerLifecycle;

import java.time.Duration;
import java.time.Instant;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.function.Supplier;

/**
 * Default implementation of a {@link Timer}.
 */
public final class DefaultTimer implements Timer {

    private final Map timings = new LinkedHashMap<>();
    private final LocLogger logger;

    public DefaultTimer(final LocLogger logger) {
        this.logger = logger;
    }

    @Override
    public void timed(final String taskName, final Runnable task) {
        if (logger.isInfoEnabled()) {
            final var preRun = Instant.now();
            task.run();
            final var postRun = Instant.now();
            timings.put(taskName, Duration.between(preRun, postRun));
        } else {
            task.run();
        }
    }

    @Override
    public  T timed(final String taskName, final Supplier supplier) {
        if (logger.isInfoEnabled()) {
            final var preRun = Instant.now();
            final var value = supplier.get();
            final var postRun = Instant.now();
            timings.put(taskName, Duration.between(preRun, postRun));
            return value;
        }
        return supplier.get();
    }

    @Override
    public void printTimings() {
        if (logger.isInfoEnabled()) {
            final var total = timings.entrySet().stream()
                    .peek(entry -> logger.info(TimerLifecycle.TASK_RUNTIME, entry.getKey(), entry.getValue().toMillis()))
                    .map(Map.Entry::getValue)
                    .reduce(Duration.ofMillis(0), Duration::plus);
            logger.info(TimerLifecycle.APPLICATION_RUNTIME, total.toMillis());
        }
        timings.clear();
    }

}