/*

 * Copyright (c) 2004, 2015, 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.

 */

/*

 * @test

 * @bug 4984057

 * @summary Test that monitors can sample a large number of attributes

 * @author Eamonn McManus

 * @modules java.management

 * @run clean MultiMonitorTest

 * @run build MultiMonitorTest

 * @run main MultiMonitorTest

 * @key randomness

 */

import java.util.*;

import javax.management.*;

import javax.management.monitor.*;

/* We create N MBeans and three monitors, one for each different

   monitor type.  Each monitor monitors a single attribute in each of

   the N MBeans.  We arrange for the trigger condition to be

   satisfied, so the listener we register on each monitor should get N

   notifications.  */

public class MultiMonitorTest {

    static final int N = 100;

    static final ObjectName[] mbeanNames = new ObjectName[N];

    static final Monitored[] monitored = new Monitored[N];

    static final int COUNTER_THRESHOLD = 1000;

    static final int OVER_COUNTER_THRESHOLD = 2000;

    static final double GAUGE_THRESHOLD = 1000.0;

    static final double OVER_GAUGE_THRESHOLD = 2000.0;

    static final String STRING_TO_COMPARE = "chou";

    static final String DIFFERENT_STRING = "chevre";

    public static void main(String[] args) throws Exception {

        System.out.println("Test that monitors can sample a large " +

                           "number of attributes");

        final MBeanServer mbs = MBeanServerFactory.createMBeanServer();

        for (int i = 0; i < N; i++) {

            mbeanNames[i] = new ObjectName(":type=Monitored,instance=" + i);

            monitored[i] = new Monitored();

            mbs.registerMBean(monitored[i], mbeanNames[i]);

        }

        final ObjectName counterMonitor =

            new ObjectName(":type=CounterMonitor");

        final ObjectName gaugeMonitor =

            new ObjectName(":type=GaugeMonitor");

        final ObjectName stringMonitor =

            new ObjectName(":type=StringMonitor");

        final ObjectName[] monitorNames =

            new ObjectName[] {counterMonitor, gaugeMonitor, stringMonitor};

        final String[] attrNames =

            new String[] {"CounterValue", "GaugeValue", "StringValue"};

        mbs.createMBean(CounterMonitor.class.getName(), counterMonitor);

        mbs.createMBean(GaugeMonitor.class.getName(), gaugeMonitor);

        mbs.createMBean(StringMonitor.class.getName(), stringMonitor);

        final CounterMonitorMBean counterProxy = (CounterMonitorMBean)

            MBeanServerInvocationHandler

            .newProxyInstance(mbs, counterMonitor, CounterMonitorMBean.class,

                              false);

        final GaugeMonitorMBean gaugeProxy = (GaugeMonitorMBean)

            MBeanServerInvocationHandler

            .newProxyInstance(mbs, gaugeMonitor, GaugeMonitorMBean.class,

                              false);

        final StringMonitorMBean stringProxy = (StringMonitorMBean)

            MBeanServerInvocationHandler

            .newProxyInstance(mbs, stringMonitor, StringMonitorMBean.class,

                              false);

        final MonitorMBean[] proxies = new MonitorMBean[] {

            counterProxy, gaugeProxy, stringProxy,

        };

        for (int i = 0; i < 3; i++) {

            proxies[i].setGranularityPeriod(1);

            proxies[i].setObservedAttribute(attrNames[i]);

            for (int j = 0; j < N; j++)

                proxies[i].addObservedObject(mbeanNames[j]);

        }

        final CountListener[] listeners = new CountListener[] {

            new CountListener(), new CountListener(), new CountListener()

        };

        for (int i = 0; i < 3; i++) {

            mbs.addNotificationListener(monitorNames[i], listeners[i],

                                        null, null);

        }

        counterProxy.setInitThreshold(new Integer(COUNTER_THRESHOLD));

        counterProxy.setNotify(true);

        gaugeProxy.setThresholds(new Double(GAUGE_THRESHOLD), new Double(0.0));

        gaugeProxy.setNotifyHigh(true);

        stringProxy.setStringToCompare(STRING_TO_COMPARE);

        stringProxy.setNotifyDiffer(true);

        // A couple of granularity periods to detect bad behaviour

        Thread.sleep(2);

        if (!listenersAreAll(0, listeners)) {

            System.out.println("TEST FAILED: listeners not all 0");

            System.exit(1);

        }

        for (int i = 0; i < 3; i++)

            proxies[i].start();

        long startTime = System.currentTimeMillis();

        while (!listenersAreAll(N, listeners)

               && System.currentTimeMillis() < startTime + 5000)

            Thread.sleep(1);

        // More time for bad behaviour

        Thread.sleep(1000);

        if (!listenersAreAll(N, listeners)) {

            System.out.print("TEST FAILED: listener counts wrong:");

            for (int i = 0; i < listeners.length; i++)

                System.out.print(" " + listeners[i].getCount());

            System.out.println();

            System.exit(1);

        }

        for (int i = 0; i < 3; i++) {

            proxies[i].stop();

            for (int j = 0; j < N; j++)

                proxies[i].removeObservedObject(mbeanNames[j]);

            ObjectName[] observed = proxies[i].getObservedObjects();

            if (observed.length != 0) {

                System.out.println("TEST FAILED: not all observed objects " +

                                   "removed: " + Arrays.asList(observed));

                System.exit(1);

            }

        }

        System.out.println("Test passed");

    }

    public static interface MonitoredMBean {

        public int getCounterValue();

        public double getGaugeValue();

        public String getStringValue();

    }

    public static class Monitored implements MonitoredMBean {

        /* We give a small random number of normal readings (possibly

           zero) before giving a reading that provokes a

           notification.  */

        private int okCounter = randomInt(5);

        private int okGauge = randomInt(5);

        private int okString = randomInt(5);

        public synchronized int getCounterValue() {

            if (--okCounter >= 0)

                return 0;

            else

                return OVER_COUNTER_THRESHOLD;

        }

        public synchronized double getGaugeValue() {

            if (--okGauge >= 0)

                return 0.0;

            else

                return OVER_GAUGE_THRESHOLD;

        }

        public synchronized String getStringValue() {

            if (--okString >= 0)

                return STRING_TO_COMPARE;

            else

                return DIFFERENT_STRING;

        }

    }

    public static class CountListener implements NotificationListener {

        private int count;

        public synchronized void handleNotification(Notification n, Object h) {

            if (!(n instanceof MonitorNotification)) {

                System.out.println("TEST FAILED: bad notif: " +

                                   n.getClass().getName());

                System.exit(1);

            }

            if (n.getType().indexOf("error") >= 0) {

                System.out.println("TEST FAILED: error notif: " + n.getType());

                System.exit(1);

            }

            count++;

        }

        public synchronized int getCount() {

            return count;

        }

    }

    private static boolean listenersAreAll(int n, CountListener[] listeners) {

        for (int i = 0; i < listeners.length; i++) {

            if (listeners[i].getCount() != n)

                return false;

        }

        return true;

    }

    private static final Random random = new Random();

    static synchronized int randomInt(int n) {

        return random.nextInt(n);

    }

}