rendezvousload.java

JXTA&#8482 is a set of open, generalized peer-to-peer (P2P) protocols that allow any networked devi

/*
 * Copyright (c) 2002-2007 Sun Microsystems, Inc.  All rights reserved.
 *  
 *  The Sun Project JXTA(TM) Software License
 *  
 *  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 Sun Microsystems, Inc. for JXTA(TM) technology." 
 *     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.
 *  
 *  JXTA is a registered trademark of Sun Microsystems, Inc. in the United 
 *  States and other countries.
 *  
 *  Please see the license information page at :
 *  <http://www.jxta.org/project/www/license.html> for instructions on use of 
 *  the license in source files.
 *  
 *  ====================================================================
 *  
 *  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.rendezvous;


import java.net.InetAddress;
import java.util.Iterator;
import java.util.Hashtable;
import junit.framework.Test;
import junit.framework.TestCase;
import junit.framework.TestSuite;
import junit.textui.TestRunner;
import net.jxta.endpoint.EndpointAddress;
import net.jxta.endpoint.Message;
import net.jxta.impl.endpoint.IPUtils;
import net.jxta.peer.PeerID;
import net.jxta.peergroup.PeerGroup;
import net.jxta.peergroup.PeerGroupID;
import net.jxta.protocol.PeerAdvertisement;
import net.jxta.test.util.AdvUtil;
import net.jxta.test.util.JxtaSink;
import net.jxta.test.util.MessageUtil;
import net.jxta.test.util.TcpConnection;


/**
 * Rendezvous load test is intended to load a rendezvous peer with connection
 * requests, and test it's limits.
 *
 * Running Requirements:
 * Run a local rdv peer on the same host running the test
 * The peer must be configured with a tcp transport on port 9701
 * In addition $JXTA_HOME/jxta.properties must define RdvManager.MaxClients=#clients simulated
 */
public class RendezvousLoad extends TestCase {

    String incarnationTagName = "RdvIncarnjxta-NetGroup" + System.currentTimeMillis();
    private String myAddress = null;
    private int PORT = 8000;
    // 200 is the default max number of edge peers, modify this
    // after modifying $JXTA_HOME/jxta.properties RdvManager.MaxClients with the corresponding value
    private int ITERATIONS = 1000;
    private long NAP = 100;
    private String serviceParm = PeerGroupID.defaultNetPeerGroupID.getUniqueValue().toString();
    private String service = PeerGroup.rendezvousClassID.toString();

    /**
     *  Constructor for the RendezvousLoad
     *
     *@param  testName
     *@exception  Exception
     */
    public RendezvousLoad(String testName) throws Exception {
        super(testName);
    }

    /**
     *  Description of the Method
     *
     *@exception  Exception
     */
    @Override
    protected void finalize() throws Exception {}

    /**
     *  The main program for the RendezvousLoad class
     *
     *@param  args  The command line arguments
     */
    public static void main(java.lang.String[] args) {
        TestRunner.run(suite());
        System.err.flush();
        System.out.flush();
    }

    /**
     *  A unit test suite for JUnit
     *
     *@return    The test suite
     */
    public static Test suite() {
        TestSuite suite = new TestSuite(RendezvousLoad.class);

        return suite;
    }

    /**
     *  Load test rdv peer using a tcp configuration
     */
    public void testLoad() {
        TcpConnection connection = null;
        PeerID destPeerID = null;
        Hashtable sinkTBL = new Hashtable(ITERATIONS);
        Hashtable connectionTBL = new Hashtable(ITERATIONS);

        for (int i = 0; i < ITERATIONS; i++) {
            PeerAdvertisement padv = AdvUtil.newPeerAdv("Fakey" + i, getMyAddress(), PORT + i, false);
            EndpointAddress epa = new EndpointAddress("tcp", getMyAddress() + ":9701", null, null);

            try {
                connection = new TcpConnection(epa, getMyInetAddress(), padv.getPeerID(), null);
                destPeerID = (PeerID) connection.getDestinationPeerID();
                JxtaSink sink = new JxtaSink(getMyInetAddress(), PORT + i, connection.getWM());
                // sinkTBL.put(padv.getPeerID(), sink);
                // connectionTBL.put(padv.getPeerID(), connection);
            } catch (Exception e) {
                e.printStackTrace();
                fail("Could not create a connection to " + getMyInetAddress() + " rdv on port 9701");
            }
            String incNum = newIncarnation();
            Message connectMsg = MessageUtil.rdvConnectMessage(padv, incNum);

            MessageUtil.addServiceParam(connectMsg, padv, "tcp://" + getMyAddress() + ":" + (PORT + i), destPeerID, service
                    ,
                    serviceParm);
            // MessageUtil.printMessageStats(connectMsg, true);
            try {
                connection.sendMessage(connectMsg);
                System.out.println("Connection request : " + i + " sent");
                Thread.sleep(NAP);
            } catch (Exception e) {
                e.printStackTrace();
                fail("Could not send Message to rdv");
            }
        }

    }

    private String getMyAddress() {
        if (myAddress == null) {
            myAddress = getMyInetAddress().getHostAddress();
        }
        return myAddress;
    }

    private InetAddress getMyInetAddress() {
        Iterator en = IPUtils.getAllLocalAddresses();

        while (en.hasNext()) {
            return (InetAddress) en.next();
        }
        return null;
    }

    public String newIncarnation() {

        return Long.toString(System.currentTimeMillis());
    }

    public static void fail(String message) {
        junit.framework.TestCase.fail(message);
    }
}