Page MenuHomec4science
Files F67240470

CommandLine.java
No OneTemporary
Actions

File Metadata

Created
Fri, Jun 21, 03:00

CommandLine.java
View Options

/**
* Copyright (c) 2010 Yahoo! Inc. All rights reserved.
*
* 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. See accompanying
* LICENSE file.
*/
package com.yahoo.ycsb;
import java.util.Properties;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Enumeration;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.HashSet;
import java.util.Vector;
import com.yahoo.ycsb.Client;
import com.yahoo.ycsb.workloads.*;
/**
* A simple command line client to a database, using the appropriate com.yahoo.ycsb.DB implementation.
*/
public class CommandLine
{
public static final String DEFAULT_DB="com.yahoo.ycsb.BasicDB";
public static void usageMessage()
{
System.out.println("YCSB Command Line Client");
System.out.println("Usage: java com.yahoo.ycsb.CommandLine [options]");
System.out.println("Options:");
System.out.println(" -P filename: Specify a property file");
System.out.println(" -p name=value: Specify a property value");
System.out.println(" -db classname: Use a specified DB class (can also set the \"db\" property)");
System.out.println(" -table tablename: Use the table name instead of the default \""+CoreWorkload.TABLENAME_PROPERTY_DEFAULT+"\"");
System.out.println();
}
public static void help()
{
System.out.println("Commands:");
System.out.println(" read key [field1 field2 ...] - Read a record");
System.out.println(" scan key recordcount [field1 field2 ...] - Scan starting at key");
System.out.println(" insert key name1=value1 [name2=value2 ...] - Insert a new record");
System.out.println(" update key name1=value1 [name2=value2 ...] - Update a record");
System.out.println(" delete key - Delete a record");
System.out.println(" table [tablename] - Get or [set] the name of the table");
System.out.println(" quit - Quit");
}
public static void main(String[] args)
{
int argindex=0;
Properties props=new Properties();
Properties fileprops=new Properties();
String table=CoreWorkload.TABLENAME_PROPERTY_DEFAULT;
while ( (argindex<args.length) && (args[argindex].startsWith("-")) )
{
if ( (args[argindex].compareTo("-help")==0) ||
(args[argindex].compareTo("--help")==0) ||
(args[argindex].compareTo("-?")==0) ||
(args[argindex].compareTo("--?")==0) )
{
usageMessage();
System.exit(0);
}
if (args[argindex].compareTo("-db")==0)
{
argindex++;
if (argindex>=args.length)
{
usageMessage();
System.exit(0);
}
props.setProperty(Client.DB_PROPERTY, args[argindex]);
argindex++;
}
else if (args[argindex].compareTo("-P")==0)
{
argindex++;
if (argindex>=args.length)
{
usageMessage();
System.exit(0);
}
String propfile=args[argindex];
argindex++;
Properties myfileprops=new Properties();
try
{
myfileprops.load(new FileInputStream(propfile));
}
catch (IOException e)
{
System.out.println(e.getMessage());
System.exit(0);
}
for (Enumeration e=myfileprops.propertyNames(); e.hasMoreElements(); )
{
String prop=(String)e.nextElement();
fileprops.setProperty(prop,myfileprops.getProperty(prop));
}
}
else if (args[argindex].compareTo("-p")==0)
{
argindex++;
if (argindex>=args.length)
{
usageMessage();
System.exit(0);
}
int eq=args[argindex].indexOf('=');
if (eq<0)
{
usageMessage();
System.exit(0);
}
String name=args[argindex].substring(0,eq);
String value=args[argindex].substring(eq+1);
props.put(name,value);
//System.out.println("["+name+"]=["+value+"]");
argindex++;
}
else if (args[argindex].compareTo("-table")==0)
{
argindex++;
if (argindex>=args.length)
{
usageMessage();
System.exit(0);
}
table=args[argindex];
argindex++;
}
else
{
System.out.println("Unknown option "+args[argindex]);
usageMessage();
System.exit(0);
}
if (argindex>=args.length)
{
break;
}
}
if (argindex!=args.length)
{
usageMessage();
System.exit(0);
}
for (Enumeration e=props.propertyNames(); e.hasMoreElements(); )
{
String prop=(String)e.nextElement();
fileprops.setProperty(prop,props.getProperty(prop));
}
props=fileprops;
System.out.println("YCSB Command Line client");
System.out.println("Type \"help\" for command line help");
System.out.println("Start with \"-help\" for usage info");
//create a DB
String dbname=props.getProperty(Client.DB_PROPERTY, DEFAULT_DB);
ClassLoader classLoader = CommandLine.class.getClassLoader();
DB db=null;
try
{
Class dbclass = classLoader.loadClass(dbname);
db=(DB)dbclass.newInstance();
}
catch (Exception e)
{
e.printStackTrace();
System.exit(0);
}
db.setProperties(props);
try
{
db.init();
}
catch (DBException e)
{
e.printStackTrace();
System.exit(0);
}
System.out.println("Connected.");
//main loop
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
for (;;)
{
//get user input
System.out.print("> ");
String input=null;
try
{
input=br.readLine();
}
catch (IOException e)
{
e.printStackTrace();
System.exit(1);
}
if (input.compareTo("")==0)
{
continue;
}
if (input.compareTo("help")==0)
{
help();
continue;
}
if (input.compareTo("quit")==0)
{
break;
}
String[] tokens=input.split(" ");
long st=System.currentTimeMillis();
//handle commands
if (tokens[0].compareTo("table")==0)
{
if (tokens.length==1)
{
System.out.println("Using table \""+table+"\"");
}
else if (tokens.length==2)
{
table=tokens[1];
System.out.println("Using table \""+table+"\"");
}
else
{
System.out.println("Error: syntax is \"table tablename\"");
}
}
else if (tokens[0].compareTo("read")==0)
{
if (tokens.length==1)
{
System.out.println("Error: syntax is \"read keyname [field1 field2 ...]\"");
}
else
{
Set<String> fields=null;
if (tokens.length>2)
{
fields=new HashSet<String>();
for (int i=2; i<tokens.length; i++)
{
fields.add(tokens[i]);
}
}
HashMap<String,ByteIterator> result=new HashMap<String,ByteIterator>();
Status ret=db.read(table,tokens[1],fields,result);
System.out.println("Return code: "+ret.getName());
for (Map.Entry<String,ByteIterator> ent : result.entrySet())
{
System.out.println(ent.getKey()+"="+ent.getValue());
}
}
}
else if (tokens[0].compareTo("scan")==0)
{
if (tokens.length<3)
{
System.out.println("Error: syntax is \"scan keyname scanlength [field1 field2 ...]\"");
}
else
{
Set<String> fields=null;
if (tokens.length>3)
{
fields=new HashSet<String>();
for (int i=3; i<tokens.length; i++)
{
fields.add(tokens[i]);
}
}
Vector<HashMap<String,ByteIterator>> results=new Vector<HashMap<String,ByteIterator>>();
Status ret=db.scan(table,tokens[1],Integer.parseInt(tokens[2]),fields,results);
System.out.println("Result: "+ret.getName());
int record=0;
if (results.isEmpty())
{
System.out.println("0 records");
}
else
{
System.out.println("--------------------------------");
}
for (HashMap<String,ByteIterator> result : results)
{
System.out.println("Record "+(record++));
for (Map.Entry<String,ByteIterator> ent : result.entrySet())
{
System.out.println(ent.getKey()+"="+ent.getValue());
}
System.out.println("--------------------------------");
}
}
}
else if (tokens[0].compareTo("update")==0)
{
if (tokens.length<3)
{
System.out.println("Error: syntax is \"update keyname name1=value1 [name2=value2 ...]\"");
}
else
{
HashMap<String,ByteIterator> values=new HashMap<String,ByteIterator>();
for (int i=2; i<tokens.length; i++)
{
String[] nv=tokens[i].split("=");
values.put(nv[0],new StringByteIterator(nv[1]));
}
Status ret=db.update(table,tokens[1],values);
System.out.println("Result: "+ret.getName());
}
}
else if (tokens[0].compareTo("insert")==0)
{
if (tokens.length<3)
{
System.out.println("Error: syntax is \"insert keyname name1=value1 [name2=value2 ...]\"");
}
else
{
HashMap<String,ByteIterator> values=new HashMap<String,ByteIterator>();
for (int i=2; i<tokens.length; i++)
{
String[] nv=tokens[i].split("=");
values.put(nv[0],new StringByteIterator(nv[1]));
}
Status ret=db.insert(table,tokens[1],values);
System.out.println("Result: "+ret.getName());
}
}
else if (tokens[0].compareTo("delete")==0)
{
if (tokens.length!=2)
{
System.out.println("Error: syntax is \"delete keyname\"");
}
else
{
Status ret=db.delete(table,tokens[1]);
System.out.println("Return result: "+ret.getName());
}
}
else
{
System.out.println("Error: unknown command \""+tokens[0]+"\"");
}
System.out.println((System.currentTimeMillis()-st)+" ms");
}
}
}

Event Timeline

c4science · Help