com.yammer.dropwizard.tasks.GarbageCollectionTask Maven / Gradle / Ivy

Go to download
package com.yammer.dropwizard.tasks;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMultimap;

import java.io.PrintWriter;

/**
 * Performs a full JVM garbage collection (probably).
 */
public class GarbageCollectionTask extends Task {
    private final Runtime runtime;

    /**
     * Creates a new {@link GarbageCollectionTask}.
     */
    public GarbageCollectionTask() {
        this(Runtime.getRuntime());
    }

    /**
     * Creates a new {@link GarbageCollectionTask} with the given {@link Runtime} instance.
     * 
     * Use {@link GarbageCollectionTask#GarbageCollectionTask()} instead.
     *
     * @param runtime    a {@link Runtime} instance
     */
    public GarbageCollectionTask(Runtime runtime) {
        super("gc");
        this.runtime = runtime;
    }

    @Override
    @SuppressWarnings("CallToSystemGC")
    public void execute(ImmutableMultimap parameters, PrintWriter output) {
        final int count = parseRuns(parameters);
        for (int i = 0; i < count; i++) {
            output.println("Running GC...");
            output.flush();
            runtime.gc();
        }

        output.println("Done!");
    }

    private static int parseRuns(ImmutableMultimap parameters) {
        final ImmutableList runs = parameters.get("runs").asList();
        if (runs.isEmpty()) {
            return 1;
        } else {
            try {
                return Integer.parseInt(runs.get(0));
            } catch (NumberFormatException ignored) {
                return 1;
            }
        }
    }
}