msuresh.raftdistdb.AtomixDB.java Source code

Introduction

Here is the source code for msuresh.raftdistdb.AtomixDB.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package msuresh.raftdistdb;

import java.io.File;
import java.io.FileNotFoundException;
import java.util.concurrent.ExecutionException;
import static msuresh.raftdistdb.Constants.STATE_LOCATION;
import org.apache.commons.cli.BasicParser;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.Option;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;

/**
 * The main driver class for AtomixDDB.
 * @author muthukumarsuresh
 */
public class AtomixDB {
    /**
     * Main Method which runs a Apache CLI which takes 3 different options for the 3 operations <br>
     * @param args <br>
     * Options : <br>
     * -setup numberOfReplicas numberOfPartitions -- which sets the system given thenumber of partitions number of replicas per partition<br>
     * -set key Value -- adds a key value pair to the DB<br>
     * -get key -- returns a value for the key if it exists<br>
     * @throws InterruptedException 
     */
    public static void main(String[] args) throws InterruptedException, ExecutionException, FileNotFoundException {
        Options options = new Options();
        Option opt = new Option("setup", true, "Sets up the replica that run the Raft Consensus Algorithm.");
        opt.setArgs(3);
        options.addOption(opt);
        opt = new Option("set", true, " Add a key-value pair into the Distributed Database.");
        opt.setArgs(3);
        options.addOption(opt);
        opt = new Option("get", true, "Given a key gets the value from the DB");
        opt.setArgs(2);
        options.addOption(opt);
        opt = new Option("clean", false, "Cleans the state information.");
        options.addOption(opt);
        opt = new Option("test", true, "Cleans the state information.");
        opt.setArgs(2);
        options.addOption(opt);
        try {
            CommandLineParser parser = new BasicParser();
            CommandLine line = null;
            line = parser.parse(options, args);
            if (line.hasOption("setup")) {
                String[] vals = line.getOptionValues("setup");
                System.out.println(vals[0]);
                RaftCluster.createCluster(vals[0], Integer.parseInt(vals[1]), Integer.parseInt(vals[2]));
            } else if (line.hasOption("set")) {
                String[] vals = line.getOptionValues("set");
                addKey(vals[0], vals[1], vals[2]);
            } else if (line.hasOption("get")) {
                String[] vals = line.getOptionValues("get");
                getKey(vals[0], vals[1]);
            } else if (line.hasOption("clean")) {
                cleanState();
            } else if (line.hasOption("test")) {
                String[] vals = line.getOptionValues("test");
                TestAtomix.createCluster(vals[0], Integer.parseInt(vals[1]));
            }
        } catch (ParseException exp) {
            System.out.println("Unexpected exception:" + exp.getMessage());
        }
    }

    /**
     * Method to setup server. Internally calls RaftCluster to setup the cluster with replicas and partition managers
     * @param noOfServers
     * @param noOfReplicas
     * @throws InterruptedException 
     */
    /**
     * adds key to DB
     * @param key
     * @param value 
     */
    private static void addKey(String name, String key, String value) throws FileNotFoundException {
        RaftClient.SetValue(name, key, value);
    }

    private static void getKey(String name, String key) throws FileNotFoundException {
        RaftClient.GetValue(name, key);
    }

    public static void delete(File path) throws FileNotFoundException {
        if (!path.exists())
            throw new FileNotFoundException(path.getAbsolutePath());
        for (File f : path.listFiles()) {
            f.delete();
        }

    }

    private static void cleanState() throws FileNotFoundException {
        delete(new File(Constants.STATE_LOCATION));
    }
    /**
     * returns a value for the key if it exists
     * @param parseInt
     * @return String
     */

}