org.carewebframework.ui.util.MemoryLeakPreventionUtil.java Source code

Introduction

Here is the source code for org.carewebframework.ui.util.MemoryLeakPreventionUtil.java

Source

/**
 * This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0. 
 * If a copy of the MPL was not distributed with this file, You can obtain one at 
 * http://mozilla.org/MPL/2.0/.
 * 
 * This Source Code Form is also subject to the terms of the Health-Related Additional
 * Disclaimer of Warranty and Limitation of Liability available at
 * http://www.carewebframework.org/licensing/disclaimer.
 */
package org.carewebframework.ui.util;

import java.lang.reflect.Field;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 *
 */
public final class MemoryLeakPreventionUtil {

    private static final Log log = LogFactory.getLog(MemoryLeakPreventionUtil.class);

    /**
     * Attempt all known cleanup measures.
     * 
     * @see #clearReferencesHttpClientKeepAliveThread() 
     */
    /*@see #shutdownLogging()*/
    public static void clean() {
        log.info("Attempting to prevent known memory leaks");
        clearReferencesHttpClientKeepAliveThread();
        /*TODO Create some type of registry so that there is no dependency to the logging extension module.  shutdownLogging();*/
    }

    /**
     * @see LogUtil#shutdown()
     */
    /*    public static void shutdownLogging() {
        LogUtil.shutdown();
    }*/

    /**
     * CWI-1409
     * <p>
     * This does what Tomcat 7 context attribute clearReferencesHttpClientKeepAliveThread is
     * suggested to do but doesn't. Using reflection in this manner is a last resort.
     * <ul>
     * <li>JDK 7 is said to fix the keepAliveTimer thread bug.</li>
     * <li>Tomcat 7.0.x context attribute clearReferencesHttpClientKeepAliveThread is suggested to
     * fix this leak but does not</li>
     * </p>
     */
    @SuppressWarnings("restriction")
    protected static void clearReferencesHttpClientKeepAliveThread() {
        try {
            final Field kac = sun.net.www.http.HttpClient.class.getDeclaredField("kac");
            kac.setAccessible(true);
            final Field keepAliveTimer = sun.net.www.http.KeepAliveCache.class.getDeclaredField("keepAliveTimer");
            keepAliveTimer.setAccessible(true);

            final Thread t = (Thread) keepAliveTimer.get(kac.get(null));//kac is a static field, hence null argument
            if (t != null) {//May not actually be running
                log.debug("KeepAliveTimer contextClassLoader: " + t.getContextClassLoader());
                if (t.getContextClassLoader() == Thread.currentThread().getContextClassLoader()) {
                    t.setContextClassLoader(ClassLoader.getSystemClassLoader());
                    log.info("Changed KeepAliveTimer classloader to system to prevent leak.");
                }
            }
        } catch (final Exception e) {
            log.warn(
                    "Exception occurred attempting to prevent sun.net.www.http.HttpClient keepAliveTimer memory leak. Note: this code should not be necessary w/ java7",
                    e);
        }

    }
}