237 lines
8.1 KiB
Java
237 lines
8.1 KiB
Java
/*
|
|
* Copyright (c) 2003, 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 7654321
|
|
* @summary Tests to receive notifications for opened and closed connections
|
|
* @author sjiang
|
|
*
|
|
* @run clean RMINotifTest
|
|
* @run build RMINotifTest
|
|
* @run main 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 =
|
|
JMXConnectorServerFactory.newJMXConnectorServer(url, null,
|
|
null);
|
|
|
|
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];
|
|
}
|