rdvpeerrdvservice.java

jxta平台的开发包

/*
 *  $Id: RdvPeerRdvService.java,v 1.81 2005/09/14 16:09:05 bondolo Exp $
 *
 *  Copyright (c) 2001 Sun Microsystems, Inc.  All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions
 *  are met:
 *
 *  1. Redistributions of source code must retain the above copyright
 *  notice, this list of conditions and the following disclaimer.
 *
 *  2. Redistributions in binary form must reproduce the above copyright
 *  notice, this list of conditions and the following disclaimer in
 *  the documentation and/or other materials provided with the
 *  distribution.
 *
 *  3. The end-user documentation included with the redistribution,
 *  if any, must include the following acknowledgment:
 *  "This product includes software developed by the
 *  Sun Microsystems, Inc. for Project JXTA."
 *  Alternately, this acknowledgment may appear in the software itself,
 *  if and wherever such third-party acknowledgments normally appear.
 *
 *  4. The names "Sun", "Sun Microsystems, Inc.", "JXTA" and "Project JXTA"
 *  must not be used to endorse or promote products derived from this
 *  software without prior written permission. For written
 *  permission, please contact Project JXTA at http://www.jxta.org.
 *
 *  5. Products derived from this software may not be called "JXTA",
 *  nor may "JXTA" appear in their name, without prior written
 *  permission of Sun.
 *
 *  THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
 *  WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 *  OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 *  DISCLAIMED.  IN NO EVENT SHALL SUN MICROSYSTEMS OR
 *  ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
 *  USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 *  ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 *  OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 *  OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 *  SUCH DAMAGE.
 *
 *  =========================================================
 *
 *  This software consists of voluntary contributions made by many
 *  individuals on behalf of Project JXTA.  For more
 *  information on Project JXTA, please see
 *  <http://www.jxta.org/>.
 *
 *  This license is based on the BSD license adopted by the Apache Foundation.
 */
package net.jxta.impl.rendezvous.rdv;


import java.util.Arrays;
import java.util.Collections;
import java.util.Enumeration;
import java.util.Map;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Vector;
import java.util.TimerTask;
import java.util.List;

import java.io.IOException;

import org.apache.log4j.Level;
import org.apache.log4j.Logger;

import net.jxta.discovery.DiscoveryService;
import net.jxta.document.Advertisement;
import net.jxta.document.AdvertisementFactory;
import net.jxta.document.Element;
import net.jxta.document.MimeMediaType;
import net.jxta.document.StructuredDocumentFactory;
import net.jxta.document.XMLDocument;
import net.jxta.endpoint.EndpointAddress;
import net.jxta.endpoint.EndpointListener;
import net.jxta.endpoint.Message;
import net.jxta.endpoint.MessageElement;
import net.jxta.endpoint.StringMessageElement;
import net.jxta.endpoint.TextDocumentMessageElement;
import net.jxta.id.ID;
import net.jxta.peer.PeerID;
import net.jxta.peergroup.PeerGroup;
import net.jxta.protocol.ConfigParams;
import net.jxta.protocol.PeerAdvertisement;

import net.jxta.rendezvous.RendezvousEvent;

import net.jxta.impl.protocol.RdvConfigAdv;
import net.jxta.impl.rendezvous.PeerConnection;
import net.jxta.impl.rendezvous.RdvGreeter;
import net.jxta.impl.rendezvous.RdvWalk;
import net.jxta.impl.rendezvous.RdvWalker;
import net.jxta.impl.rendezvous.RendezVousPropagateMessage;
import net.jxta.impl.rendezvous.RendezVousServiceImpl;
import net.jxta.impl.rendezvous.StdRendezVousService;
import net.jxta.impl.rendezvous.limited.LimitedRangeWalk;
import net.jxta.impl.rendezvous.rendezvousMeter.ClientConnectionMeter;
import net.jxta.impl.rendezvous.rendezvousMeter.RendezvousMeterBuildSettings;
import net.jxta.impl.util.TimeUtils;


/**
 * A JXTA {@link net.jxta.rendezvous.RendezvousService} implementation which
 * implements the rendezvous server portion of the standard JXTA Rendezvous
 * Protocol (RVP).
 *
 * @see net.jxta.rendezvous.RendezvousService
 * @see <a href="http://spec.jxta.org/nonav/v1.0/docbook/JXTAProtocols.html#proto-rvp" target="_blank">JXTA Protocols Specification : Rendezvous Protocol</a>
 **/
public class RdvPeerRdvService extends StdRendezVousService {
    
    /**
     *  Log4J Logger
     **/
    private final static Logger LOG = Logger.getLogger(RdvPeerRdvService.class.getName());
    
    public static final String RDV_SVC_NAME = "RdvWalkSvcName";
    public static final String RDV_SVC_PARAM = "RdvWalkSvcParam";
    
    public final static long   GC_INTERVAL = 2 * TimeUtils.AMINUTE;
    
    private final Map clients = Collections.synchronizedMap(new HashMap());
    
    private long leaseDuration = 20L * TimeUtils.AMINUTE;
    private long maxNbOfClients = 200;
    
    private RdvWalk walk = null;
    private RdvGreeter greeter = null;
    private RdvWalker walker = null;
    private WalkListener walkListener = null;
    
    /**
     * Constructor for the RdvPeerRdvService object
     */
    public RdvPeerRdvService(PeerGroup g, RendezVousServiceImpl rdvService) {
        
        super(g, rdvService);
        
        ConfigParams confAdv = (ConfigParams) g.getConfigAdvertisement();
        
        // Get the config. If we do not have a config, we're done; we just keep
        // the defaults (edge peer/no auto-rdv)
        if (confAdv != null) {
            Advertisement adv = null;
            
            try {
                XMLDocument configDoc = (XMLDocument) confAdv.getServiceParam(rdvService.getAssignedID());
                
                if (null != configDoc) {
                    // XXX 20041027 backwards compatibility
                    configDoc.addAttribute( "type", RdvConfigAdv.getAdvertisementType() );

                    adv = AdvertisementFactory.newAdvertisement(configDoc);
                }
            } catch (java.util.NoSuchElementException failed) {
                ;
            }
            
            if (adv instanceof RdvConfigAdv) {
                RdvConfigAdv rdvConfigAdv = (RdvConfigAdv) adv;
                
                if (rdvConfigAdv.getMaxTTL() > 0) {
                    MAX_TTL = rdvConfigAdv.getMaxTTL();
                }
                
                if (rdvConfigAdv.getMaxClients() > 0) {
                    maxNbOfClients = rdvConfigAdv.getMaxClients();
                }
                
                if (rdvConfigAdv.getLeaseDuration() > 0) {
                    leaseDuration = rdvConfigAdv.getLeaseDuration();
                }
            }
        }
        
        // Update the peeradv with that information:
        try {
            XMLDocument params = (XMLDocument)
            StructuredDocumentFactory.newStructuredDocument(MimeMediaType.XMLUTF8, "Parm");
            Element e = params.createElement("Rdv", Boolean.TRUE.toString());
            
            params.appendChild(e);
            group.getPeerAdvertisement().putServiceParam(rdvService.getAssignedID(), params);
        } catch (Exception ohwell) {
            // don't worry about it for now. It'll still work.
            if (LOG.isEnabledFor(Level.WARN)) {
                LOG.warn("Failed adding service params", ohwell);
            }
        }
        
        if (LOG.isEnabledFor(Level.INFO)) {
            LOG.info("RendezVous Service is initialized for " + group.getPeerGroupID() + " as a Rendezvous peer.");
        }
    }
    
    /**
     *  Listener for
     *
     *  &lt;assignedID>/&lt;group-unique>
     **/
    private class StdRdvRdvProtocolListener implements StdRendezVousService.StdRdvProtocolListener {
        
        /**
         *  {@inheritDoc}
         **/
        public void processIncomingMessage(Message msg, EndpointAddress srcAddr, EndpointAddress dstAddr) {
            
            if (LOG.isEnabledFor(Level.DEBUG)) {
                LOG.debug("[" + group.getPeerGroupID() + "] processing " + msg);
            }
            
            if (msg.getMessageElement("jxta", ConnectRequest) != null) {
                processLeaseRequest(msg);
            }
            
            if (msg.getMessageElement("jxta", DisconnectRequest) != null) {
                processDisconnectRequest(msg);
            }
            
            if (msg.getMessageElement("jxta", RdvAdvReply) != null) {
                processRdvAdvReply(msg);
            }
        }
    }
    
    /**
     *  {@inheritDoc}
     **/
    protected int startApp(String[] argv) {
        
        super.startApp(argv, new StdRdvRdvProtocolListener());
        
        // The other services may not be fully functional but they're there
        // so we can start our subsystems.
        // As for us, it does not matter if our methods are called between init
        // and startApp().
        
        // Start the Greeter / Walker protcol
        walkInit();
        
        timer.scheduleAtFixedRate(new GCTask(), GC_INTERVAL, GC_INTERVAL);
        
        if (RendezvousMeterBuildSettings.RENDEZVOUS_METERING && (rendezvousMeter != null)) {
            rendezvousMeter.startRendezvous();
        }
        
        rdvService.generateEvent(RendezvousEvent.BECAMERDV, group.getPeerID());
        
        if (LOG.isEnabledFor(Level.INFO)) {
            LOG.info("RdvPeerRdvService is started");
        }
        
        return 0;
    }
    
    /**
     *  {@inheritDoc}
     **/
    protected synchronized void stopApp() {
        
        if (closed) {
            return;
        }
        
        closed = true;
        
        if (walkListener != null) {
            greeter.setEndpointListener(null);
            walkListener = null;
            greeter.stop();
            greeter = null;
        }
        
        if (walker != null) {
            walker.stop();
            walker = null;
        }
        
        if (walk != null) {
            walk.stop();
            walk = null;
        }
        
        // Tell all our clientss that we are going down
        disconnectAllClients();
        
        clients.clear();
        
        super.stopApp();
        
        if (RendezvousMeterBuildSettings.RENDEZVOUS_METERING && (rendezvousMeter != null)) {
            rendezvousMeter.stopRendezvous();
        }
    }
    
    /**
     *  {@inheritDoc}
     **/
    public void challengeRendezVous(ID peer, long delay) {
        throw new UnsupportedOperationException("Not supported by rendezvous");
    }
    
    /**
     *  {@inheritDoc}