TimedRunnable.java

package org.googlecode.perftrace.perf4j;


import java.io.Serializable;


/**
 * This helper wrapper class can be used to add timing statements to an existing Runnable instance, logging how long
 * it takes for the run method to execute. Note that instances of this class are only serializable if the wrapped
 * Runnable is serializable.
 *
 */
public class TimedRunnable implements Runnable, Serializable {
    private static final long serialVersionUID = -3506292767816484297L;
    private Runnable wrappedTask;
    private LoggingStopWatch stopWatch;

    /**
     * Wraps the existing Runnable in order to time its run method.
     *
     * @param task      The existing Runnable whose run method is to be timed and executed. May not be null.
     * @param stopWatch The LoggingStopWatch to use to time the run method execution. Note that this stop watch should
     *                  already have its tag and message set to what should be logged when the task is run. May not
     *                  be null.
     */
    public TimedRunnable(Runnable task, LoggingStopWatch stopWatch) {
        this.wrappedTask = task;
        this.stopWatch = stopWatch;
    }

    /**
     * Gets the Runnable task that is wrapped by this TimedRunnable.
     *
     * @return The wrapped Runnable whose execution time is to be logged.
     */
    public Runnable getWrappedTask() {
        return wrappedTask;
    }

    /**
     * Gets the LoggingStopWatch that will be used to time the run method execution.
     *
     * @return The LoggingStopWatch to use to log execution time.
     */
    public LoggingStopWatch getStopWatch() {
        return stopWatch;
    }

    /**
     * Executes the run method of the underlying task, using the LoggingStopWatch to track the execution time.
     */
    public void run() {
        try {
            stopWatch.start();
            wrappedTask.run();
        } finally {
            stopWatch.stop();
        }
    }
}