/*

 * Copyright 2003-2008 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

 * CA 95054 USA or visit www.sun.com if you need additional information or

 * have any questions.

 */

/*

 * @test

 * @bug 7654321

 * @summary Tests to receive notifications for opened and closed connections

 * @author sjiang

 * @run clean RMINotifTest

 * @run build RMINotifTest

 * @run main RMINotifTest event

 */

// java imports

//

import java.rmi.RemoteException;

import java.rmi.registry.LocateRegistry;

import java.rmi.registry.Registry;

import java.util.Random;

import javax.management.MBeanNotificationInfo;

import javax.management.MBeanServer;

import javax.management.MBeanServerConnection;

import javax.management.MBeanServerFactory;

import javax.management.Notification;

import javax.management.NotificationBroadcasterSupport;

import javax.management.NotificationFilterSupport;

import javax.management.NotificationListener;

import javax.management.ObjectInstance;

import javax.management.ObjectName;

import javax.management.remote.JMXConnector;

import javax.management.remote.JMXConnectorFactory;

import javax.management.remote.JMXConnectorServer;

import javax.management.remote.JMXConnectorServerFactory;

import javax.management.remote.JMXServiceURL;

public class RMINotifTest {

    public static void main(String[] args) {

        try {

            // create a rmi registry

            Registry reg = null;

            int port = 6666;

            final Random r = new Random();

            while(port++<7000) {

                try {

                    reg = LocateRegistry.createRegistry(++port);

                    System.out.println("Creation of rmi registry succeeded. Running on port " + port);

                    break;

                } catch (RemoteException re) {

                // no problem

                }

            }

            if (reg == null) {

                System.out.println("Failed to create a RMI registry, "+

                                   "the ports from 6666 to 6999 are all occupied.");

                System.exit(1);

            }

            // create a MBeanServer

            MBeanServer server = MBeanServerFactory.createMBeanServer();

            // create a notif emitter mbean

            ObjectName mbean = new ObjectName ("Default:name=NotificationEmitter");

            server.registerMBean(new NotificationEmitter(), mbean);

            // create a rmi server

            JMXServiceURL url =

                new JMXServiceURL("rmi", null, port,

                                  "/jndi/rmi://:" + port + "/server" + port);

            System.out.println("RMIConnectorServer address " + url);

            JMXConnectorServer sServer =

            ObjectInstance ss = server.registerMBean(sServer, new ObjectName("Default:name=RmiConnectorServer"));

            sServer.start();

            // create a rmi client

            JMXConnector rmiConnection =

                JMXConnectorFactory.newJMXConnector(url, null);

            rmiConnection.connect(null);

            MBeanServerConnection client = rmiConnection.getMBeanServerConnection();

            // add listener at the client side

            client.addNotificationListener(mbean, listener, null, null);

            //ask to send notifs

            Object[] params = new Object[1];

            String[] signatures = new String[1];

            params[0] = new Integer(nb);

            signatures[0] = "java.lang.Integer";

            client.invoke(mbean, "sendNotifications", params, signatures);

            // waiting ...

            synchronized (lock) {

                if (receivedNotifs != nb) {

                    lock.wait(10000);

                    System.out.println(">>> Received notifications..."+receivedNotifs);

                }

            }

            // check

            if (receivedNotifs != nb) {

                System.exit(1);

            } else {

                System.out.println("The client received all notifications.");

            }

            // remove listener

            client.removeNotificationListener(mbean, listener);

            // more test

            NotificationFilterSupport filter = new NotificationFilterSupport();

            Object o = new Object();

            client.addNotificationListener(mbean, listener, filter, o);

            client.removeNotificationListener(mbean, listener, filter, o);

            sServer.stop();

//          // clean

//          client.unregisterMBean(mbean);

//          rmiConnection.close();

//          Thread.sleep(2000);

        } catch (Exception e) {

            e.printStackTrace();

            System.exit(1);

        }

    }

//--------------------------

// private classes

//--------------------------

    private static class Listener implements NotificationListener {

        public void handleNotification(Notification notif, Object handback) {

            if(++receivedNotifs == nb) {

                synchronized(lock) {

                    lock.notifyAll();

                }

            }

        }

    }

    public static class NotificationEmitter extends NotificationBroadcasterSupport implements NotificationEmitterMBean {

//      public NotificationEmitter() {

//              super();

// System.out.println("===NotificationEmitter: new instance.");

//      }

        /**

         * Returns a NotificationInfo object containing the name of the Java class of the notification

         * and the notification types sent by this notification broadcaster.

         */

        public MBeanNotificationInfo[] getNotificationInfo() {

            MBeanNotificationInfo[] ntfInfoArray  = new MBeanNotificationInfo[1];

            String[] ntfTypes = new String[1];

            ntfTypes[0] = myType;

            ntfInfoArray[0] = new MBeanNotificationInfo(ntfTypes,

                                                        "javax.management.Notification",

                                                        "Notifications sent by the NotificationEmitter");

            return ntfInfoArray;

        }

//      public void addNotificationListener(NotificationListener listener, NotificationFilter filter, Object handback) {

//          super.addNotificationListener(listener, filter, handback);

//          System.out.println("============NotificationEmitter: add new listener");

//      }

        /**

         * Send a Notification object with the specified times.

         * The sequence number will be from zero to times-1.

         *

         * @param nb The number of notifications to send

         */

        public void sendNotifications(Integer nb) {

            System.out.println("===NotificationEmitter: be asked to send notifications: "+nb);

            Notification notif;

            for (int i=1; i<=nb.intValue(); i++) {

                notif = new Notification(myType, this, i);

                sendNotification(notif);

            }

        }

        private String myType = "notification.my_notification";

    }

    public interface NotificationEmitterMBean {

        public void sendNotifications(Integer nb);

    }

    private static NotificationListener listener = new Listener();

    private static int nb = 10;

    private static int receivedNotifs = 0;

    private static int[] lock = new int[0];

}