DiscoveryManager.java

package org.objectweb.jonas.discovery;

import javax.management.JMException;
import javax.management.MBeanRegistration;
import javax.management.MBeanServer;
import javax.management.MBeanServerNotification;
import javax.management.Notification;
import javax.management.NotificationListener;
import javax.management.ObjectName;
import javax.management.remote.JMXServiceURL;

import org.objectweb.jonas.jmx.J2eeObjectName;

public class DiscoveryManager implements
    DiscoveryManagerMBean,
    MBeanRegistration,
    NotificationListener {

    private int           listeningPort;
    private String        listeningIp;
    private DiscoveryComm dc;
    private Thread        commDaemon;
    private MBeanServer   mbeanServer;
    private ObjectName    myOn;
    private int           ttl = 1;

    private String jonasName = null;
    private String domainName = null;
    private String[] urls = null;

    public DiscoveryManager(int listeningPort, String listeningIp) {
      this.listeningIp = listeningIp;
      this.listeningPort = listeningPort;
    }

    /*
     * (non-Javadoc)
     *
     * @see org.objectweb.jonas.server.discovery.DiscoveryManagerMBean#start()
     */

    public void start() {

        // creates an instance of the Discovery communication class
        dc = new DiscoveryComm(this);
        dc.setJonasName(jonasName);
        dc.setDomainName(domainName);
        dc.setUrls(urls);

        if (commDaemon == null)
            commDaemon = new Thread(dc, "commDaemon");
        commDaemon.start();
    }

    public int getListeningPort() {
        return listeningPort;
    }

    public void setListeningPort(int listeningPort) {
        this.listeningPort = listeningPort;

    }

    public String getListeningIp() {
        return listeningIp;
    }

    public void setListeningIp(String listeningIp) {
        this.listeningIp = listeningIp;

    }

    public void setTimeToLive(int ttl) {
        this.ttl = ttl;
    }

    public int getTimeToLive() {
        return this.ttl;
    }

    public ObjectName preRegister(MBeanServer mbeanServer, ObjectName on) throws Exception {
        // set mbeanServer
        this.mbeanServer = mbeanServer;
        this.myOn = on;

        return myOn;
    }

    public void postRegister(Boolean arg0) {
      // Add discovery manager as a listener of the delegate object to receive unregistration notification of the j2EEServer.
      ObjectName delegate;
      try {
        delegate = new ObjectName("JMImplementation:type=MBeanServerDelegate");
        mbeanServer.addNotificationListener(delegate, this, null, null);
      } catch (JMException e) {
        e.printStackTrace();
      }
    }

    public void preDeregister() throws Exception {
        // stop the communication Daemon
        dc.stop();
    }

    public void postDeregister() {

    }

    public void handleNotification(Notification notification, Object handback) {
        if (notification.getType().equals(MBeanServerNotification.UNREGISTRATION_NOTIFICATION)) {
            ObjectName notificationSender = ((MBeanServerNotification) notification).getMBeanName();
            ObjectName j2eeServerName = J2eeObjectName.J2EEServer(domainName, jonasName);
            if (notificationSender.equals(j2eeServerName)) {
                dc.stop();
            }
        }
    }

    protected void setJonasName(String jonasName) {
        this.jonasName = jonasName;
    }
    protected void setDomainName(String domainNamle) {
        this.domainName = domainNamle;
    }
    protected void setUrls(String[] urls) {
        this.urls = urls;
    }

    public void stop() {
        dc.stop();
    }
}

---