package org.embeddedt.modernfix.util;

import com.google.common.base.Stopwatch;

import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicLong;

public class AsyncStopwatch {
    private final AtomicLong cpuTimeMs = new AtomicLong(0);
    private final ThreadLocal<Stopwatch> threadStopwatch = ThreadLocal.withInitial(Stopwatch::createUnstarted);

    public void startMeasuringAsync() {
        threadStopwatch.get().start();
    }

    public void stopMeasuringAsync() {
        Stopwatch watch = threadStopwatch.get();
        watch.stop();
        long elapsed = watch.elapsed(TimeUnit.MILLISECONDS);
        cpuTimeMs.addAndGet(elapsed);
        watch.reset();
    }

    public long getCpuTime() {
        return cpuTimeMs.get();
    }
}