AFAP_RR_LoadGenerator.java

/*******************************************************************************
 * gMix open source project - https://svs.informatik.uni-hamburg.de/gmix/
 * Copyright (C) 2014  SVS
 *
 * This program is free software: you can redistribute it and/or modify 
 * it under the terms of the GNU General Public License as published by 
 * the Free Software Foundation, either version 3 of the License, or 
 * (at your option) any later version.
 *  
 * This program 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 for more details.
 *
 * You should have received a copy of the GNU General Public License 
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *******************************************************************************/
package staticContent.evaluation.loadGenerator.asFastAsPossible;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Random;

import staticContent.evaluation.loadGenerator.LoadGenerator;
import staticContent.evaluation.loadGenerator.LoadGenerator.InsertLevel;
import staticContent.framework.AnonNode;
import staticContent.framework.launcher.ToolName;
import staticContent.framework.routing.RoutingMode;
import staticContent.framework.socket.socketInterfaces.StreamAnonSocket;
import staticContent.framework.socket.socketInterfaces.AnonSocketOptions.CommunicationDirection;
import staticContent.framework.util.Util;


public class AFAP_RR_LoadGenerator extends AFAP_LoadGenerator {
	
	private AnonNode client;

	
	
	protected AFAP_RR_LoadGenerator(LoadGenerator owner) {
		super(owner);
		System.out.println("LOAD_GENERATOR (AFAP): start at " +System.currentTimeMillis());
		// create client
		owner.commandLineParameters.gMixTool = ToolName.CLIENT;
		this.client = new AnonNode(owner.commandLineParameters);
		int numberOfConnections;
		if (owner.INSERT_LEVEL == InsertLevel.APPLICATION_LEVEL)
			numberOfConnections = settings.getPropertyAsInt("AL-AFAP-NUMBER_OF_CLIENTS");
		else if (owner.INSERT_LEVEL == InsertLevel.MIX_PACKET_LEVEL)
			numberOfConnections = settings.getPropertyAsInt("MPL-AFAP-NUMBER_OF_CLIENTS");
		else
			throw new RuntimeException("unknown mode: " +owner.INSERT_LEVEL); 
		
		int connectionsPerThread = numberOfConnections / client.NUMBER_OF_THREADS;
		for (int i=0; i<client.NUMBER_OF_THREADS; i++)
			new RequestThread(connectionsPerThread).start();
	}

	
	private class RequestThread extends Thread {

		private StreamAnonSocket[] sockets;
		private OutputStream[] outputStreams;
		private InputStream[] inputStreams;
		private Random random = new Random();
		
		
		public RequestThread(int numberOfConnections) {
			this.sockets = new StreamAnonSocket[numberOfConnections];
			this.outputStreams = new OutputStream[numberOfConnections];
			if (client.IS_DUPLEX)
				this.inputStreams = new InputStream[numberOfConnections];
		}
		

		@Override
		public void run() {
			try {
				// create and connect sockets:
				System.out.println("LOAD_GENERATOR (AFAP): creating " +sockets.length +" connections...");
				CommunicationDirection cm = client.IS_DUPLEX ? CommunicationDirection.DUPLEX : CommunicationDirection.SIMPLEX_SENDER;
				for (int i=0; i<sockets.length; i++) {
					sockets[i] = client.createStreamSocket(cm, client.ROUTING_MODE != RoutingMode.GLOBAL_ROUTING);
					sockets[i].connect(settings.getPropertyAsInt("SERVICE_PORT1"));
					outputStreams[i] = sockets[i].getOutputStream();
					if (client.IS_DUPLEX)
						inputStreams[i] = sockets[i].getInputStream();	
				} 
				System.out.println("LOAD_GENERATOR (AFAP): creating " +sockets.length +" connections done: start sending");
				if (client.IS_DUPLEX)
					new ReplyThread(sockets, inputStreams).start();
				// start sending
				while (true) {
					for (int i=0; i<sockets.length; i++) {
						byte[] data = new byte[sockets[i].getMTU()];
						random.nextBytes(data);
						outputStreams[i].write(data);
						outputStreams[i].flush();
					} 
				}
			} catch (IOException e) {
				e.printStackTrace();
			}

		}
	}
	
	
	private class ReplyThread extends Thread {
		
		private StreamAnonSocket[] sockets;
		private InputStream[] inputStreams;
		
		
		public ReplyThread(StreamAnonSocket[] sockets, InputStream[] inputStreams) {
			this.sockets = sockets;
			this.inputStreams = inputStreams;
		}

		
		@Override
		public void run() {
			while (true) {
				try {
					for (int i=0; i<sockets.length; i++) {
						byte[] data = new byte[sockets[i].getMTU()];
						Util.forceRead(inputStreams[i], data);
					} 
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
		
	}
	
}