rendezvousload.java

jxta_src_2.41b jxta 2.41b 最新版源码 from www.jxta.org

/*
 *  $Id: RendezvousLoad.java,v 1.8 2004/10/15 19:20:52 hamada Exp $
 *
 *  Copyright (c) 2002 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.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
     */
    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);
    }
}