MockProfiler2.java

package com.etsy.statsd.profiler.profilers;

import com.etsy.statsd.profiler.Arguments;
import com.etsy.statsd.profiler.Profiler;
import com.etsy.statsd.profiler.reporter.MockReporter;

import java.util.Set;
import java.util.concurrent.TimeUnit;

/**
 * Mock profiler for testing
 *
 * @author Andrew Johnson
 */
public class MockProfiler2 extends Profiler {
    private Set<String> output;

    public MockProfiler2(Set<String> output) {
        super(new MockReporter(), null);
        this.output = output;
    }

    @Override
    public void profile() {
        output.add(this.getClass().getSimpleName() + "-profile");
    }

    @Override
    public void flushData() {
        output.add(this.getClass().getSimpleName() + "-flushData");
    }

    @Override
    public long getPeriod() {
        return 1;
    }

    @Override
    public TimeUnit getTimeUnit() {
        return TimeUnit.MINUTES;
    }

    @Override
    protected void handleArguments(Arguments arguments) { }
}