BFTMapClientCloudFIT.java

/**
Copyright (c) 2007-2013 Alysson Bessani, Eduardo Alchieri, Paulo Sousa, and the authors indicated in the @author tags

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
 */
package bftsmart.demo.bftmap;

import java.io.IOException;
import java.util.TreeMap;
import java.util.Random;
import java.util.logging.FileHandler;
import java.util.logging.Logger;
import java.util.logging.SimpleFormatter;

public class BFTMapClientCloudFIT {

	public static Logger logger;


	public static void main(String[] args) throws NumberFormatException	{
		if(args.length < 3)
		{
			System.out.println("Usage: java KVClients <number clients> <process id base> <time for running (sec)>");
			System.exit(-1);
		}

		//get arguments 
		int numberClients = Integer.parseInt(args[0]);
		int idBase = Integer.parseInt(args[1]);
		int timeSec = Integer.parseInt(args[2]);

		//init log
		initLog();

		//create the BFTMapInteractiveClient threads
		KVClientInstance [] list = new KVClientInstance[numberClients]; 

		for(int i = 0 ; i < list.length ; i ++)
		{
			list[i] = new KVClientInstance(idBase + i);
			list[i].start();
		}

		//leave the threads processing for a while
		try
		{
			Thread.sleep(timeSec * 1000);			
		} catch (InterruptedException e) {}

		System.out.println("Stop running...");


		//stop and wait for the threads
		for(int i = 0 ; i < list.length ; i ++)
		{
			list[i].stopRun();
			try {
				list[i].join();
			} catch (InterruptedException e) {}
		}

		System.out.println("Test ended...exit!");
		System.exit(0);
	}	

	private static void initLog()
	{
		try {
			boolean append = true;
			FileHandler fh = new FileHandler(BFTMapClientCloudFIT.class.getName()+".log", append);
			fh.setFormatter(new SimpleFormatter());

			logger = Logger.getLogger(BFTMapClientCloudFIT.class.getName());
			logger.addHandler(fh);
		}
		catch (IOException e) {
			System.out.println("PROBLEMS]: "+e.getMessage());
			System.exit(-1);
		}
	}
}

class KVClientInstance extends Thread {

	private int inc;
	private int id;
	private boolean run;
	private Random rand;

	public KVClientInstance(int id)
	{
		this.id = id;
		this.inc = 0;
		this.run = true;
		this.rand = new Random(id);
	}

	public void run()
	{


		BFTMap bftMap = new BFTMap(id);
		String tableName = "table-"+id;

		try {
			createTable(bftMap,tableName);
		} catch (Exception e1) {
			System.out.println("Problems: Inserting a new value into the table("+tableName+"): "+e1.getLocalizedMessage());
			System.exit(1);	
		}

		while(run)
		{
			try {

				boolean result = insertValue(bftMap,tableName);
				if(!result)
				{
					System.out.println("Problems: Inserting a new value into the table("+tableName+")");
					System.exit(1);	
				}

			} catch (InterruptedException e) {
				System.out.println("Client id["+id+"]: it was interrupted");
				run = false;
			} catch (Exception e) {
				bftMap = new BFTMap(id);
				try {
					createTable(bftMap,tableName);
				} catch (Exception e1) {
					System.out.println("Client id["+id+"]: problems");
				}
			}
		}

		BFTMapClientCloudFIT.logger.info("Client id["+id+"] operations: "+inc);
		return;
	}

	public void stopRun()
	{
		this.run = false;
	}

	private boolean createTable(BFTMap bftMap, String nameTable) throws Exception
	{
		boolean tableExists;

		tableExists = bftMap.containsKey(nameTable);
		if (tableExists == false)
			bftMap.put(nameTable, new TreeMap<String,byte[]>());

		return tableExists;
	}

	private boolean insertValue(BFTMap bftMap, String nameTable) throws Exception
	{
		String key = "Key"+rand.nextInt();
		String value = Integer.toString(new Random().nextInt());
		byte[] valueBytes = value.getBytes();

		byte[] resultBytes = bftMap.putEntry(nameTable, key, valueBytes);
		if(resultBytes== null)
			throw new Exception();
		else
			inc++;

		return true;
	}
}