/*
 * Copyright (c) 2003, 2018, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

package nsk.monitoring.share;

import java.lang.management.*;
import javax.management.*;
import javax.management.openmbean.*;
import java.util.*;
import nsk.share.*;

/**
 * <code>ThreadMonitor</code> class is a wrapper of <code>ThreadMXBean</code>.
 * Depending on command line arguments, an instance of this class redirects
 * invocations to the <code>ThreadMXBean</code> interface. If
 * <code>-testMode="directly"</code> option is set, this instance directly
 * invokes corresponding method of the <code>ThreadMXBean</code> interface.
 * If <code>-testMode="server"</code> option is set it will make invocations
 * via MBeanServer. If <code>-testMode="proxy"</code> option is set it will
 * make invocations via MBeanServer proxy.
 *
 * @see ArgumentHandler
 */
public class ThreadMonitor extends Monitor {

    // Names of the attributes of ThreadMXBean
    private static final String GET_THREAD_INFO = "getThreadInfo";
    private static final String GET_THREAD_CPU_TIME = "ThreadCpuTime";
    private static final String ALL_THREAD_IDS = "AllThreadIds";
    private static final String RESET_PEAK = "resetPeakThreadCount";
    private static final String GET_PEAK_COUNT = "PeakThreadCount";
    private static final String THREAD_COUNT = "ThreadCount";
    private static final String FIND_THREADS = "findMonitorDeadlockedThreads";
    private static final String IS_CURRENT = "CurrentThreadCpuTimeSupported";
    private static final String IS_CPUTIME = "ThreadCpuTimeSupported";
    private static final String IS_CONT_SUPP
        = "ThreadContentionMonitoringSupported";
    private static final String IS_CONT_ENAB
        = "ThreadContentionMonitoringEnabled";

    // An instance of ThreadMXBean
    private static ThreadMXBean mbean = ManagementFactory.getThreadMXBean();

    // proxy instance
    private ThreadMXBean proxyInstance;

    // Internal trace level
    private static final int TRACE_LEVEL = 10;

    static {
        Monitor.logPrefix = "ThreadMonitor> ";
    }

    /**
     * Creates a new <code>ThreadMonitor</code> object.
     *
     * @param log <code>Log</code> object to print info to.
     * @param argumentHandler <code>ArgumentHandler</code> object that saves
     *        all info about test's arguments.
     *
     */
    public ThreadMonitor(Log log, ArgumentHandler argumentHandler) {
        super(log, argumentHandler);
    }

    /**
     *
     * Return a proxy instance for a platform
     * {@link java.lang.management.ThreadMXBean
     * <code>ThreadMXBean</code>} interface.
     *
     */
    synchronized ThreadMXBean getProxy() {
        if (proxyInstance == null) {
            // create proxy instance
            try {
                proxyInstance = (ThreadMXBean)
                ManagementFactory.newPlatformMXBeanProxy(
                    getMBeanServer(),
                    ManagementFactory.THREAD_MXBEAN_NAME,
                    ThreadMXBean.class
                );
            } catch (Exception e) {
                throw new Failure(e);
            }
        }
        return proxyInstance;
    }

    /**
     * Redirects the invocation to {@link
     * java.lang.management.ThreadMXBean#getAllThreadIds()
     * <code>ThreadMXBean.getAllThreadIds()</code>}.
     *
     * @return an array of <code>long</code>, each is a thread ID.
     */
    public long[] getAllThreadIds() {
        int mode = getTestMode();

        switch (mode) {
        case DIRECTLY_MODE:
            logger.trace(TRACE_LEVEL,"[getAllThreadIds] getAllThreadIds()"
                            + " directly invoked");
            return mbean.getAllThreadIds();

        case SERVER_MODE:
            logger.trace(TRACE_LEVEL,"[getAllThreadIds] getAllThreadIds()"
                       + " invoked through MBeanServer");
            return getLongArrayAttribute(mbeanObjectName, ALL_THREAD_IDS);

        case PROXY_MODE:
            logger.trace(TRACE_LEVEL,"[getAllThreadIds] getAllThreadIds()"
                            + " invoked through MBeanServer proxy");
            return getProxy().getAllThreadIds();
        }

        throw new TestBug("Unknown testMode " + mode);
    }

    /**
     * Redirects the invocation to {@link
     * java.lang.management.ThreadMXBean#getThreadInfo
     * <code>ThreadMXBean.getThreadInfo()</code>}.
     *
     * @param id the thread ID of the thread.
     * @param maxDepth the maximum number of entries in the stack trace to
     *        be dumped. <code>Integer.MAX_VALUE</code> could be used to request
     *        entire stack to be dumped.
     *
     * @return A <code>ThreadInfo</code> of the thread of the given ID.
     *          <code>null</code> if the thread of the given ID is not alive
     *          or it does not exist
     */
    public ThreadInfo getThreadInfo(long id, int maxDepth) {
        int mode = getTestMode();

        switch (mode) {
        case DIRECTLY_MODE:
            logger.trace(TRACE_LEVEL, "[getThreadInfo] getThreadInfo(long, "
                       + "int) directly invoked");
            return mbean.getThreadInfo(id, maxDepth);

        case SERVER_MODE:
            Object[] params = {new Long(id), new Integer(maxDepth)};
            String[] signature = {"long", "int"};

            try {
                logger.trace(TRACE_LEVEL, "[getThreadInfo] getThreadInfo(long, "
                           + "int) invoked through MBeanServer");
                Object value = getMBeanServer().invoke(mbeanObjectName,
                                                             GET_THREAD_INFO,
                                                             params, signature);
                if (value instanceof ThreadInfo)
                        return (ThreadInfo) value;
                else {
                        CompositeData data = (CompositeData) value;
                        return ThreadInfo.from(data);
                }
            } catch (Exception e) {
                e.printStackTrace();
                throw new Failure(e);
            }
        case PROXY_MODE:
            logger.trace(TRACE_LEVEL, "[getThreadInfo] getThreadInfo(long, "
                       + "int) invoked through MBeanServer proxy");
            return getProxy().getThreadInfo(id, maxDepth);
        }

        throw new TestBug("Unknown testMode " + mode);
    }

    /**
     * Redirects the invocation to {@link
     * java.lang.management.ThreadMXBean#isCurrentThreadCpuTimeSupported()
     * <code>ThreadMXBean.isCurrentThreadCpuTimeSupported()</code>}.
     *
     * @return <code>true</code>, if the JVM supports CPU time measurement for
     *         current thread, <code>false</code> otherwise.
     */
    public boolean isCurrentThreadCpuTimeSupported() {
        int mode = getTestMode();

        switch (mode) {
        case DIRECTLY_MODE:
            return mbean.isCurrentThreadCpuTimeSupported();

        case SERVER_MODE:
            return getBooleanAttribute(mbeanObjectName, IS_CURRENT);

        case PROXY_MODE:
            return getProxy().isCurrentThreadCpuTimeSupported();
        }

        throw new TestBug("Unknown testMode " + mode);
    }

    /**
     * Redirects the invocation to {@link
     * java.lang.management.ThreadMXBean#isThreadCpuTimeSupported()
     * <code>ThreadMXBean.isThreadCpuTimeSupported()</code>}.
     *
     * @return <code>true</code>, if the JVM supports CPU time measurement for
     *         any threads, <code>false</code> otherwise.
     */
    public boolean isThreadCpuTimeSupported() {
        int mode = getTestMode();

        switch (mode) {
        case DIRECTLY_MODE:
            return mbean.isThreadCpuTimeSupported();

        case SERVER_MODE:
            return getBooleanAttribute(mbeanObjectName, IS_CPUTIME);

        case PROXY_MODE:
            return getProxy().isThreadCpuTimeSupported();
        }

        throw new TestBug("Unknown testMode " + mode);
    }

    /**
     * Redirects the invocation to {@link
     * java.lang.management.ThreadMXBean#isThreadContentionMonitoringSupported()
     * <code>ThreadMXBean.isThreadContentionMonitoringSupported()</code>}.
     *
     * @return <code>true</code>, if the JVM supports thread contantion
     *         monitoring, <code>false</code> otherwise.
     */
    public boolean isThreadContentionMonitoringSupported() {
        int mode = getTestMode();

        switch (mode) {
        case DIRECTLY_MODE:
            return mbean.isThreadContentionMonitoringSupported();

        case SERVER_MODE:
            return getBooleanAttribute(mbeanObjectName, IS_CONT_SUPP);

        case PROXY_MODE:
            return getProxy().isThreadContentionMonitoringSupported();
        }

        throw new TestBug("Unknown testMode " + mode);
    }

    /**
     * Redirects the invocation to {@link
     * java.lang.management.ThreadMXBean#setThreadContentionMonitoringEnabled
     * <code>ThreadMXBean.setThreadContentionMonitoringEnabled()</code>}.
     *
     * @param enable <code>true</code> to enable, <code>false</code> to disable.
     */
    public void setThreadContentionMonitoringEnabled(boolean enable) {
        int mode = getTestMode();

        switch (mode) {
        case DIRECTLY_MODE:
            mbean.setThreadContentionMonitoringEnabled(enable);
            break;

        case SERVER_MODE:
            setBooleanAttribute(mbeanObjectName, IS_CONT_ENAB, enable);
            break;

        case PROXY_MODE:
            getProxy().setThreadContentionMonitoringEnabled(enable);
            break;

        default:
            throw new TestBug("Unknown testMode " + mode);
        }
    }

    /**
     * Redirects the invocation to {@link
     * java.lang.management.ThreadMXBean#getThreadCpuTime
     * <code>ThreadMXBean.getThreadCpuTime()</code>}.
     *
     * @param id the id of a thread
     *
     * @return the CPU time for a thread of the specified ID, if the thread
     *         existsand is alive and CPU time measurement is enabled, -1
     *         otherwise.
     */
    public long getThreadCpuTime(long id) {
        int mode = getTestMode();

        switch (mode) {
        case DIRECTLY_MODE:
            return mbean.getThreadCpuTime(id);

        case SERVER_MODE:
            Object[] params = {new Long(id)};
            String[] signature = {"long"};

            try {
                Long l = (Long) getMBeanServer().invoke(mbeanObjectName,
                                                            GET_THREAD_CPU_TIME,
                                                             params, signature);
                return l.longValue();
            } catch (Exception e) {
                e.printStackTrace();
                throw new Failure(e);
            }

        case PROXY_MODE:
            return getProxy().getThreadCpuTime(id);
        }

        throw new TestBug("Unknown testMode " + mode);
    }

    /**
     * Redirects the invocation to {@link
     * java.lang.management.ThreadMXBean#getThreadCount()
     * <code>ThreadMXBean.getThreadCount()</code>}.
     *
     * @return the current number of live threads.
     */
    public int getThreadCount() {
        int mode = getTestMode();

        switch (mode) {
        case DIRECTLY_MODE:
            return mbean.getThreadCount();

        case SERVER_MODE:
            return getIntAttribute(mbeanObjectName, THREAD_COUNT);

        case PROXY_MODE:
            return getProxy().getThreadCount();
        }

        throw new TestBug("Unknown testMode " + mode);
    }

    /**
     * Redirects the invocation to {@link
     * java.lang.management.ThreadMXBean#getPeakThreadCount()
     * <code>ThreadMXBean.getPeakThreadCount()</code>}.
     *
     * @return the peak live thrad count.
     */
    public int getPeakThreadCount() {
        int mode = getTestMode();

        switch (mode) {
        case DIRECTLY_MODE:
            return mbean.getPeakThreadCount();

        case SERVER_MODE:
            return getIntAttribute(mbeanObjectName, GET_PEAK_COUNT);

        case PROXY_MODE:
            return getProxy().getPeakThreadCount();
        }

        throw new TestBug("Unknown testMode " + mode);
    }

    /**
     * Redirects the invocation to {@link
     * java.lang.management.ThreadMXBean#resetPeakThreadCount()
     * <code>ThreadMXBean.resetPeakThreadCount()</code>}.
     */
    public void resetPeakThreadCount() {
        int mode = getTestMode();

        switch (mode) {
        case DIRECTLY_MODE:
            mbean.resetPeakThreadCount();
            break;

        case SERVER_MODE:
            try {
               getMBeanServer().invoke(mbeanObjectName, RESET_PEAK, null, null);
            } catch (Exception e) {
                e.printStackTrace(logger.getOutStream());
                throw new Failure(e);
            }
            break;
        case PROXY_MODE:
            getProxy().resetPeakThreadCount();
            break;

        default:
            throw new TestBug("Unknown testMode " + mode);
        }

    }

    /**
     * Redirects the invocation to {@link
     * java.lang.management.ThreadMXBean#findMonitorDeadlockedThreads
     * <code>ThreadMXBean.findMonitorDeadlockedThreads()</code>}.
     *
     * @return an array of IDs of the reads that are monitor deadlocked, if any;
     *         <code>null</code> otherwise.
     */
    public long[] findMonitorDeadlockedThreads() {
        int mode = getTestMode();

        switch (mode) {
        case DIRECTLY_MODE:
            return mbean.findMonitorDeadlockedThreads();

        case SERVER_MODE:
            try {
                Object o = (Object) getMBeanServer().invoke(mbeanObjectName,
                                                      FIND_THREADS, null, null);
                return (long[]) o;
            } catch (Exception e) {
                e.printStackTrace(logger.getOutStream());
                throw new Failure(e);
            }
        case PROXY_MODE:
            return getProxy().findMonitorDeadlockedThreads();
        }
        throw new TestBug("Unknown testMode " + mode);
    }
} // ThreadMonitor