Here it is:
//---------------------------------------------------
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import junit.framework.TestCase;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.HBaseTestingUtility;
import org.apache.hadoop.hbase.HColumnDescriptor;
import org.apache.hadoop.hbase.HTableDescriptor;
import org.apache.hadoop.hbase.KeyValue;
import org.apache.hadoop.hbase.MiniHBaseCluster;
import org.apache.hadoop.hbase.client.Append;
import org.apache.hadoop.hbase.client.Get;
import org.apache.hadoop.hbase.client.HBaseAdmin;
import org.apache.hadoop.hbase.client.HTable;
import org.apache.hadoop.hbase.client.Put;
import org.apache.hadoop.hbase.client.Result;
import org.apache.log4j.ConsoleAppender;
import org.apache.log4j.Level;
import org.apache.log4j.Logger;
import org.apache.log4j.PatternLayout;
public class HBaseAppendTest extends TestCase{
static final Log LOG = LogFactory.getLog(CoprocessorBaseTest.class);
private static HBaseTestingUtility UTIL = new HBaseTestingUtility();
protected byte[] TABLE_A = "TABLE_A".getBytes();
protected byte[][] FAMILIES = new byte[][]
{"fam_a".getBytes(), "fam_b".getBytes(), "fam_c".getBytes()};
protected byte[][] COLUMNS =
{"col_a".getBytes(), "col_b".getBytes(), "col_c".getBytes()};
int VERSIONS = 10;
/** The cluster. */
MiniHBaseCluster cluster;
HTableDescriptor tableDesc;
HTable table;
/* (non-Javadoc)
* @see junit.framework.TestCase#setUp()
*/
@Override
public void setUp() throws Exception {
ConsoleAppender console = new ConsoleAppender(); // create
appender
// configure the appender
String PATTERN = "%d [%p|%c|%C{1}] %m%n";
console.setLayout(new PatternLayout(PATTERN));
console.setThreshold(Level.ERROR);
console.activateOptions();
// add appender to any Logger (here is root)
Logger.getRootLogger().removeAllAppenders();
Logger.getRootLogger().addAppender(console);
Configuration conf = UTIL.getConfiguration();
conf.set("hbase.zookeeper.useMulti", "false");
UTIL.startMiniCluster(1);
cluster = UTIL.getMiniHBaseCluster();
createHBaseTable();
}
/**
* Creates the HBase table.
*
* @throws IOException Signals that an I/O exception has occurred.
*/
protected void createHBaseTable() throws IOException {
LOG.error("Create HBase table and put data");
HColumnDescriptor famA = new HColumnDescriptor(FAMILIES[0]);
famA.setMaxVersions(VERSIONS);
HColumnDescriptor famB = new HColumnDescriptor(FAMILIES[1]);
famB.setMaxVersions(VERSIONS);
HColumnDescriptor famC = new HColumnDescriptor(FAMILIES[2]);
famC.setMaxVersions(VERSIONS);
tableDesc = new HTableDescriptor(TABLE_A);
tableDesc.addFamily(famA);
tableDesc.addFamily(famB);
tableDesc.addFamily(famC);
Configuration cfg = cluster.getConf();
HBaseAdmin admin = new HBaseAdmin(cfg);
if( admin.tableExists(tableDesc.getName()) == false){
admin.createTable(tableDesc);
LOG.error("Created table "+tableDesc);
}
table = new HTable(cfg, TABLE_A);
// Create row
List<KeyValue> rowData = generateRowData();
Put put = createPut(rowData);
// Put data
table.put(put);
LOG.error("Finished.");
}
protected Put createPut(List<KeyValue> values)
{
Put put = new Put(values.get(0).getRow());
for(KeyValue kv: values)
{
put.add(kv.getFamily(), kv.getQualifier(),
kv.getTimestamp(), kv.getValue());
}
return put;
}
List<KeyValue> generateRowData(){
byte[] row = "row".getBytes();
byte[] value = "value".getBytes();
long startTime = System.currentTimeMillis();
ArrayList<KeyValue> list = new ArrayList<KeyValue>();
int count = 0;
for(byte[] f: FAMILIES){
for(byte[] c: COLUMNS){
count = 0;
for(; count < VERSIONS; count++){
KeyValue kv = new KeyValue(row, f, c,
startTime - 1000*(count), value);
list.add(kv);
}
}
}
Collections.sort(list, KeyValue.COMPARATOR);
return list;
}
protected Append createAppend(byte[] row, List<byte[]> families,
List<byte[]> columns, byte[] value){
Append op = new Append(row);
for(byte[] f: families){
for(byte[] c: columns){
op.add(f, c, value);
}
}
return op;
}
public void testAppend() throws IOException
{
LOG.error("Test append started. Testing row: 'row'");
byte[] row ="row".getBytes();
byte[] toAppend = "_appended".getBytes();
Get get = new Get(row);
get.setMaxVersions(Integer.MAX_VALUE);
Result result = table.get(get);
assertEquals(90, result.size() );
Append append = createAppend(row, Arrays.asList(FAMILIES),
Arrays.asList(COLUMNS), toAppend);
Result r = table.append(append);
assertEquals(9, r.size());
get = new Get(row);
get.setMaxVersions(Integer.MAX_VALUE);
result = table.get(get);
assertEquals (90, result.size());
LOG.error("Test append finished.");
}
}
// -------------------------------------------------------------------------
Let me know your findings.
________________________________________
From: Ted Yu [[email protected]]
Sent: Monday, July 22, 2013 4:32 PM
To: [email protected]
Subject: Re: does append delete all cell versions except the last one?
Can you formulate the code below as a unit test so that other people can
try it out ?
I pasted the code into a unit test but there were a few unresolved method
calls such as createGet().
Cheers
Confidentiality Notice: The information contained in this message, including
any attachments hereto, may be confidential and is intended to be read only by
the individual or entity to whom this message is addressed. If the reader of
this message is not the intended recipient or an agent or designee of the
intended recipient, please note that any review, use, disclosure or
distribution of this message or its attachments, in any form, is strictly
prohibited. If you have received this message in error, please immediately
notify the sender and/or [email protected] and delete or destroy any
copy of this message and its attachments.
