mbclient.py

#!/usr/bin/env python
import os
from argparse import ArgumentParser
from threaded_ssh import ThreadedClients
from ServerConfig import Storage
from ServerConfig import TellStore
from ServerConfig import Kudu
from ServerConfig import Cassandra
from ServerConfig import Microbench

def startMBClient(populate = False, uoutFile = None):
    default_out = ""
    if Storage.storage == TellStore:
        default_out = "mbench_{0}".format(TellStore.approach)
    elif Storage.storage == Kudu:
        default_out = "mbench_kudu"
    elif Storage.storage == Cassandra:
        default_out = "mbench_cassandra"
    
    default_out = '{0}/{1}_sf{2}_N{3}'.format(Microbench.result_dir, default_out, Microbench.scaling, Microbench.numColumns)
    
    if (uoutFile):
        outfile = uoutFile
    else:
        outfile = default_out
    appendFile = 0
    while os.path.isfile(outfile + ".db"):
        appendFile = appendFile + 1
        outfile = "{0}_{1}".format(outfile, appendFile)

    probabilities = "-i {0} -d {1} -u {2}".format(Microbench.insertProb, Microbench.deleteProb, Microbench.updateProb)
    
    cmd = '{0}/watch/microbench/mbclient -H "{1}" -s {2} -c {3} -t {4} -a {5} -o {6} -b {7} -w {8} {9}'.format(TellStore.builddir, Microbench.getServerList(), Microbench.scaling, Microbench.clients, Microbench.clientThreads, Microbench.analyticalClients, outfile + ".db", Microbench.txBatch, Microbench.oltpWaitTime, probabilities)
    if Microbench.onlyQ1:
        cmd += ' -q'
    if Microbench.noWarmUp:
        cmd += " --no-warmup"
    
    if (populate):
        cmd += ' -P'
    
    print "Execute {0}".format(cmd)
    return os.system(cmd)

if __name__ == "__main__":
    default_out = ''
    parser = ArgumentParser()
    parser.add_argument("-P", dest='populate', help="Populate data", action="store_true")
    parser.add_argument("outfile", help="Result database", default=default_out, nargs='?')
    args = parser.parse_args()
    if (default_out != ''):
        exit(startMBClient(args.populate, default_out))
    else:
        exit(startMBClient(args.populate))