ClientInsertWorkload.java

package org.fi;

import java.io.*;
import java.nio.channels.*;

import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import java.util.SortedMap;
import java.util.TreeMap;



public class ClientInsertWorkload implements Workload {

  private Driver     driver;
  private Cass       cass;
  private Utility    u;
  private Experiment exp;

  // workload specific
  private String KEYSPACE = "Keyspace1";
  private String COLUMNFAMILY = "Standard1";
  private String COLUMNPATH = "experimentValue";
  private String VALUE = "berkeley";
  private static final String ENCODING = "UTF8";

  // *******************************************
  public ClientInsertWorkload(Driver driver, Experiment exp) {
    this.driver = driver;
    this.cass = driver.getCass();
    this.u = driver.getUtility();
    this.exp = exp;
   }


  // *******************************************
  // the algorithm
  // *******************************************
  public void run() {


    // 1. setup for this specific workload
    preSetup();


    // 3. the exact workload where we want to run with failure
    runWithFailure();

    // 5. run post setup
    postSetup();
  }


  // *******************************************
  public void runWithFailure() {
  	//TODO remove this
  	//temporary measure
  	u.createNewFile(Driver.EXPERIMENT_RUN_FLAG);


    // prepare the new file we want to put to hdfs, and put it
    // if there is exception it's caught inside hdfs and
    // put it in exp
    String key = String.format("key-%03d", exp.getExpNum());
    cass.insertEntry(key, VALUE , exp);

    u.deleteFile(Driver.EXPERIMENT_RUN_FLAG);

  }


  // *******************************************
  public void preSetup() {

    cass.setEncoding(ENCODING);
    cass.setKeyspace(KEYSPACE);
    cass.setColumnFamily(COLUMNFAMILY);
    cass.setColumnPath(COLUMNPATH);



    // 2. enable failures (and optimizer)
    Driver.enableFailureManager();
    Driver.enableClientOptimizer();
    Driver.enableFrog();
    Driver.enableCoverage();

    // 3. let's make sure, we setup the connection before we go
    // into the run with failure
    cass.assertConnection();
  }

  // *******************************************
  public void postSetup() {

    // 4. stop the failure
    Driver.disableCoverage();
    Driver.disableFrog();
    Driver.disableFailureManager();
    Driver.disableClientOptimizer();

    // get entry
    String key = String.format("key-%03d", exp.getExpNum());
    cass.getEntry(key, exp);

    // then delete the file
    cass.delete(key, exp);


    //Doesn't work when the experiment fails because the folder name changes to wiped-exp-..s.dfa
    //REMOVE THIS LATER
    		//JINSU: I want to see the out file for successful this experiment.
    		String from = Driver.CASS_LOGS_DIR;
    		String to = exp.getExpNumDir().getAbsolutePath();
    		if(!(new File(to).exists())) {
    			String temp = String.format("%s/wiped-%05d-%s",
				      Driver.EXP_RESULT_DIR,
				      Driver.getWipedOutNum(),
				      to);
				  to = temp;
    		}
    		u.copyFile(from+"fi.out", to+"/fi.out");
				u.copyFile(from+"node0.out", to+"/node0.out");
				u.copyFile(from+"node1.out", to+"/node1.out");
				u.copyFile(from+"node2.out", to+"/node2.out");
				u.copyFile(from+"node3.out", to+"/node3.out");

  }


}