psiprobe.controllers.threads.ThreadStackController.java Source code

Introduction

Here is the source code for psiprobe.controllers.threads.ThreadStackController.java

Source

/**
 * Licensed under the GPL License. You may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 *
 * THIS PACKAGE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
 * WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE.
 */
package psiprobe.controllers.threads;

import org.apache.commons.modeler.Registry;
import org.springframework.web.bind.ServletRequestUtils;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.ParameterizableViewController;

import psiprobe.model.ThreadStackElement;
import psiprobe.tools.JmxTools;

import java.util.ArrayList;
import java.util.List;

import javax.management.MBeanServer;
import javax.management.ObjectName;
import javax.management.openmbean.CompositeData;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * The Class ThreadStackController.
 */
public class ThreadStackController extends ParameterizableViewController {

    /** The stack element count. */
    private int stackElementCount = 20;

    /**
     * Gets the stack element count.
     *
     * @return the stack element count
     */
    public int getStackElementCount() {
        return stackElementCount;
    }

    /**
     * Sets the stack element count.
     *
     * @param stackElementCount the new stack element count
     */
    public void setStackElementCount(int stackElementCount) {
        this.stackElementCount = stackElementCount;
    }

    @Override
    protected ModelAndView handleRequestInternal(HttpServletRequest request, HttpServletResponse response)
            throws Exception {

        long threadId = ServletRequestUtils.getLongParameter(request, "id", -1);
        String threadName = ServletRequestUtils.getStringParameter(request, "name", null);

        List<ThreadStackElement> stack = null;
        MBeanServer mbeanServer = new Registry().getMBeanServer();
        ObjectName threadingOName = new ObjectName("java.lang:type=Threading");

        if (threadId == -1 && threadName != null) {
            // find thread by name
            for (long id : (long[]) mbeanServer.getAttribute(threadingOName, "AllThreadIds")) {
                CompositeData cd = (CompositeData) mbeanServer.invoke(threadingOName, "getThreadInfo",
                        new Object[] { id }, new String[] { "long" });
                String name = JmxTools.getStringAttr(cd, "threadName");
                if (threadName.equals(name)) {
                    threadId = id;
                    break;
                }
            }
        }

        if (mbeanServer.queryMBeans(threadingOName, null) != null && threadId != -1) {

            CompositeData cd = (CompositeData) mbeanServer.invoke(threadingOName, "getThreadInfo",
                    new Object[] { threadId, stackElementCount }, new String[] { "long", "int" });
            if (cd != null) {
                CompositeData[] elements = (CompositeData[]) cd.get("stackTrace");
                threadName = JmxTools.getStringAttr(cd, "threadName");

                stack = new ArrayList<>(elements.length);

                for (CompositeData cd2 : elements) {
                    ThreadStackElement tse = new ThreadStackElement();
                    tse.setClassName(JmxTools.getStringAttr(cd2, "className"));
                    tse.setFileName(JmxTools.getStringAttr(cd2, "fileName"));
                    tse.setMethodName(JmxTools.getStringAttr(cd2, "methodName"));
                    tse.setLineNumber(JmxTools.getIntAttr(cd2, "lineNumber", -1));
                    tse.setNativeMethod(JmxTools.getBooleanAttr(cd2, "nativeMethod"));
                    stack.add(tse);
                }
            }
        }

        return new ModelAndView(getViewName(), "stack", stack).addObject("threadName", threadName);
    }

}