Java 类org.apache.hadoop.hbase.thrift2.generated.TColumnValue 实例源码

项目:ditb    文件:ThriftUtilities.java   
/**
 * Creates a {@link TResult} (Thrift) from a {@link Result} (HBase).
 *
 * @param in the <code>Result</code> to convert
 *
 * @return converted result, returns an empty result if the input is <code>null</code>
 */
public static TResult resultFromHBase(Result in) {
  Cell[] raw = in.rawCells();
  TResult out = new TResult();
  byte[] row = in.getRow();
  if (row != null) {
    out.setRow(in.getRow());
  }
  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  for (Cell kv : raw) {
    TColumnValue col = new TColumnValue();
    col.setFamily(CellUtil.cloneFamily(kv));
    col.setQualifier(CellUtil.cloneQualifier(kv));
    col.setTimestamp(kv.getTimestamp());
    col.setValue(CellUtil.cloneValue(kv));
    if (kv.getTagsLength() > 0) {
      col.setTags(CellUtil.getTagArray(kv));
    }
    columnValues.add(col);
  }
  out.setColumnValues(columnValues);
  return out;
}
项目:ditb    文件:ThriftUtilities.java   
public static Append appendFromThrift(TAppend append) throws IOException {
  Append out = new Append(append.getRow());
  for (TColumnValue column : append.getColumns()) {
    out.add(column.getFamily(), column.getQualifier(), column.getValue());
  }

  if (append.isSetAttributes()) {
    addAttributes(out, append.getAttributes());
  }

  if (append.isSetDurability()) {
    out.setDurability(durabilityFromThrift(append.getDurability()));
  }

  if(append.getCellVisibility() != null) {
    out.setCellVisibility(new CellVisibility(append.getCellVisibility().getExpression()));
  }

  return out;
}
项目:ditb    文件:TestThriftHBaseServiceHandlerWithLabels.java   
public void assertTColumnValuesEqual(List<TColumnValue> columnValuesA,
    List<TColumnValue> columnValuesB) {
  assertEquals(columnValuesA.size(), columnValuesB.size());
  Comparator<TColumnValue> comparator = new Comparator<TColumnValue>() {
    @Override
    public int compare(TColumnValue o1, TColumnValue o2) {
      return Bytes.compareTo(Bytes.add(o1.getFamily(), o1.getQualifier()),
          Bytes.add(o2.getFamily(), o2.getQualifier()));
    }
  };
  Collections.sort(columnValuesA, comparator);
  Collections.sort(columnValuesB, comparator);

  for (int i = 0; i < columnValuesA.size(); i++) {
    TColumnValue a = columnValuesA.get(i);
    TColumnValue b = columnValuesB.get(i);
    assertArrayEquals(a.getFamily(), b.getFamily());
    assertArrayEquals(a.getQualifier(), b.getQualifier());
    assertArrayEquals(a.getValue(), b.getValue());
  }
}
项目:ditb    文件:TestThriftHBaseServiceHandler.java   
public void assertTColumnValuesEqual(List<TColumnValue> columnValuesA,
    List<TColumnValue> columnValuesB) {
  assertEquals(columnValuesA.size(), columnValuesB.size());
  Comparator<TColumnValue> comparator = new Comparator<TColumnValue>() {
    @Override
    public int compare(TColumnValue o1, TColumnValue o2) {
      return Bytes.compareTo(Bytes.add(o1.getFamily(), o1.getQualifier()),
          Bytes.add(o2.getFamily(), o2.getQualifier()));
    }
  };
  Collections.sort(columnValuesA, comparator);
  Collections.sort(columnValuesB, comparator);

  for (int i = 0; i < columnValuesA.size(); i++) {
    TColumnValue a = columnValuesA.get(i);
    TColumnValue b = columnValuesB.get(i);
    assertArrayEquals(a.getFamily(), b.getFamily());
    assertArrayEquals(a.getQualifier(), b.getQualifier());
    assertArrayEquals(a.getValue(), b.getValue());
  }
}
项目:ditb    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testExists() throws TIOError, TException {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testExists".getBytes();
  ByteBuffer table = wrap(tableAname);

  TGet get = new TGet(wrap(rowName));
  assertFalse(handler.exists(table, get));

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname), wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);

  handler.put(table, put);

  assertTrue(handler.exists(table, get));
}
项目:ditb    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testPutGet() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testPutGet".getBytes();
  ByteBuffer table = wrap(tableAname);

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname), wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);

  put.setColumnValues(columnValues);

  handler.put(table, put);

  TGet get = new TGet(wrap(rowName));

  TResult result = handler.get(table, get);
  assertArrayEquals(rowName, result.getRow());
  List<TColumnValue> returnedColumnValues = result.getColumnValues();
  assertTColumnValuesEqual(columnValues, returnedColumnValues);
}
项目:ditb    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testIncrement() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testIncrement".getBytes();
  ByteBuffer table = wrap(tableAname);

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname),
    wrap(Bytes.toBytes(1L))));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);
  handler.put(table, put);

  List<TColumnIncrement> incrementColumns = new ArrayList<TColumnIncrement>();
  incrementColumns.add(new TColumnIncrement(wrap(familyAname), wrap(qualifierAname)));
  TIncrement increment = new TIncrement(wrap(rowName), incrementColumns);
  handler.increment(table, increment);

  TGet get = new TGet(wrap(rowName));
  TResult result = handler.get(table, get);

  assertArrayEquals(rowName, result.getRow());
  assertEquals(1, result.getColumnValuesSize());
  TColumnValue columnValue = result.getColumnValues().get(0);
  assertArrayEquals(Bytes.toBytes(2L), columnValue.getValue());
}
项目:ditb    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testAppend() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testAppend".getBytes();
  ByteBuffer table = wrap(tableAname);
  byte[] v1 = Bytes.toBytes("42");
  byte[] v2 = Bytes.toBytes("23");
  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(v1)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);
  handler.put(table, put);

  List<TColumnValue> appendColumns = new ArrayList<TColumnValue>();
  appendColumns.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(v2)));
  TAppend append = new TAppend(wrap(rowName), appendColumns);
  handler.append(table, append);

  TGet get = new TGet(wrap(rowName));
  TResult result = handler.get(table, get);

  assertArrayEquals(rowName, result.getRow());
  assertEquals(1, result.getColumnValuesSize());
  TColumnValue columnValue = result.getColumnValues().get(0);
  assertArrayEquals(Bytes.add(v1, v2), columnValue.getValue());
}
项目:ditb    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testMetrics() throws Exception {
  Configuration conf = UTIL.getConfiguration();
  ThriftMetrics metrics = getMetrics(conf);
  ThriftHBaseServiceHandler hbaseHandler = createHandler();
  THBaseService.Iface handler =
      ThriftHBaseServiceHandler.newInstance(hbaseHandler, metrics);
  byte[] rowName = "testMetrics".getBytes();
  ByteBuffer table = wrap(tableAname);

  TGet get = new TGet(wrap(rowName));
  assertFalse(handler.exists(table, get));

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname),  wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);

  handler.put(table, put);

  assertTrue(handler.exists(table, get));
  metricsHelper.assertCounter("put_num_ops", 1, metrics.getSource());
  metricsHelper.assertCounter( "exists_num_ops", 2, metrics.getSource());
}
项目:LCIndex-HBase-0.94.16    文件:TestThriftHBaseServiceHandler.java   
public void assertTColumnValuesEqual(List<TColumnValue> columnValuesA,
    List<TColumnValue> columnValuesB) {
  assertEquals(columnValuesA.size(), columnValuesB.size());
  Comparator<TColumnValue> comparator = new Comparator<TColumnValue>() {
    @Override
    public int compare(TColumnValue o1, TColumnValue o2) {
      return Bytes.compareTo(Bytes.add(o1.getFamily(), o1.getQualifier()),
          Bytes.add(o2.getFamily(), o2.getQualifier()));
    }
  };
  Collections.sort(columnValuesA, comparator);
  Collections.sort(columnValuesB, comparator);

  for (int i = 0; i < columnValuesA.size(); i++) {
    TColumnValue a = columnValuesA.get(i);
    TColumnValue b = columnValuesB.get(i);
    assertArrayEquals(a.getFamily(), b.getFamily());
    assertArrayEquals(a.getQualifier(), b.getQualifier());
    assertArrayEquals(a.getValue(), b.getValue());
  }
}
项目:LCIndex-HBase-0.94.16    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testExists() throws TIOError, TException {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testExists".getBytes();
  ByteBuffer table = wrap(tableAname);

  TGet get = new TGet(wrap(rowName));
  assertFalse(handler.exists(table, get));

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname), wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);

  handler.put(table, put);

  assertTrue(handler.exists(table, get));
}
项目:LCIndex-HBase-0.94.16    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testPutGet() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testPutGet".getBytes();
  ByteBuffer table = wrap(tableAname);

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname), wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);

  put.setColumnValues(columnValues);

  handler.put(table, put);

  TGet get = new TGet(wrap(rowName));

  TResult result = handler.get(table, get);
  assertArrayEquals(rowName, result.getRow());
  List<TColumnValue> returnedColumnValues = result.getColumnValues();
  assertTColumnValuesEqual(columnValues, returnedColumnValues);
}
项目:LCIndex-HBase-0.94.16    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testIncrement() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testIncrement".getBytes();
  ByteBuffer table = wrap(tableAname);

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname),
    wrap(Bytes.toBytes(1L))));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);
  handler.put(table, put);

  List<TColumnIncrement> incrementColumns = new ArrayList<TColumnIncrement>();
  incrementColumns.add(new TColumnIncrement(wrap(familyAname), wrap(qualifierAname)));
  TIncrement increment = new TIncrement(wrap(rowName), incrementColumns);
  handler.increment(table, increment);

  TGet get = new TGet(wrap(rowName));
  TResult result = handler.get(table, get);

  assertArrayEquals(rowName, result.getRow());
  assertEquals(1, result.getColumnValuesSize());
  TColumnValue columnValue = result.getColumnValues().get(0);
  assertArrayEquals(Bytes.toBytes(2L), columnValue.getValue());
}
项目:LCIndex-HBase-0.94.16    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testMetrics() throws Exception {
  Configuration conf = UTIL.getConfiguration();
  ThriftMetrics metrics = getMetrics(conf);
  THBaseService.Iface handler =
      ThriftHBaseServiceHandler.newInstance(conf, metrics);
  byte[] rowName = "testMetrics".getBytes();
  ByteBuffer table = wrap(tableAname);

  TGet get = new TGet(wrap(rowName));
  assertFalse(handler.exists(table, get));

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname),  wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);

  handler.put(table, put);

  assertTrue(handler.exists(table, get));
  logMetrics(metrics);
  verifyMetrics(metrics, "put_num_ops", 1);
  verifyMetrics(metrics, "exists_num_ops", 2);
}
项目:HIndex    文件:ThriftUtilities.java   
/**
 * Creates a {@link TResult} (Thrift) from a {@link Result} (HBase).
 *
 * @param in the <code>Result</code> to convert
 *
 * @return converted result, returns an empty result if the input is <code>null</code>
 */
public static TResult resultFromHBase(Result in) {
  Cell[] raw = in.rawCells();
  TResult out = new TResult();
  byte[] row = in.getRow();
  if (row != null) {
    out.setRow(in.getRow());
  }
  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  for (Cell kv : raw) {
    TColumnValue col = new TColumnValue();
    col.setFamily(CellUtil.cloneFamily(kv));
    col.setQualifier(CellUtil.cloneQualifier(kv));
    col.setTimestamp(kv.getTimestamp());
    col.setValue(CellUtil.cloneValue(kv));
    if (kv.getTagsLength() > 0) {
      col.setTags(CellUtil.getTagArray(kv));
    }
    columnValues.add(col);
  }
  out.setColumnValues(columnValues);
  return out;
}
项目:HIndex    文件:ThriftUtilities.java   
public static Append appendFromThrift(TAppend append) throws IOException {
  Append out = new Append(append.getRow());
  for (TColumnValue column : append.getColumns()) {
    out.add(column.getFamily(), column.getQualifier(), column.getValue());
  }

  if (append.isSetAttributes()) {
    addAttributes(out, append.getAttributes());
  }

  if (append.isSetDurability()) {
    out.setDurability(durabilityFromThrift(append.getDurability()));
  }

  if(append.getCellVisibility() != null) {
    out.setCellVisibility(new CellVisibility(append.getCellVisibility().getExpression()));
  }

  return out;
}
项目:HIndex    文件:TestThriftHBaseServiceHandlerWithLabels.java   
public void assertTColumnValuesEqual(List<TColumnValue> columnValuesA,
    List<TColumnValue> columnValuesB) {
  assertEquals(columnValuesA.size(), columnValuesB.size());
  Comparator<TColumnValue> comparator = new Comparator<TColumnValue>() {
    @Override
    public int compare(TColumnValue o1, TColumnValue o2) {
      return Bytes.compareTo(Bytes.add(o1.getFamily(), o1.getQualifier()),
          Bytes.add(o2.getFamily(), o2.getQualifier()));
    }
  };
  Collections.sort(columnValuesA, comparator);
  Collections.sort(columnValuesB, comparator);

  for (int i = 0; i < columnValuesA.size(); i++) {
    TColumnValue a = columnValuesA.get(i);
    TColumnValue b = columnValuesB.get(i);
    assertArrayEquals(a.getFamily(), b.getFamily());
    assertArrayEquals(a.getQualifier(), b.getQualifier());
    assertArrayEquals(a.getValue(), b.getValue());
  }
}
项目:HIndex    文件:TestThriftHBaseServiceHandler.java   
public void assertTColumnValuesEqual(List<TColumnValue> columnValuesA,
    List<TColumnValue> columnValuesB) {
  assertEquals(columnValuesA.size(), columnValuesB.size());
  Comparator<TColumnValue> comparator = new Comparator<TColumnValue>() {
    @Override
    public int compare(TColumnValue o1, TColumnValue o2) {
      return Bytes.compareTo(Bytes.add(o1.getFamily(), o1.getQualifier()),
          Bytes.add(o2.getFamily(), o2.getQualifier()));
    }
  };
  Collections.sort(columnValuesA, comparator);
  Collections.sort(columnValuesB, comparator);

  for (int i = 0; i < columnValuesA.size(); i++) {
    TColumnValue a = columnValuesA.get(i);
    TColumnValue b = columnValuesB.get(i);
    assertArrayEquals(a.getFamily(), b.getFamily());
    assertArrayEquals(a.getQualifier(), b.getQualifier());
    assertArrayEquals(a.getValue(), b.getValue());
  }
}
项目:HIndex    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testExists() throws TIOError, TException {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testExists".getBytes();
  ByteBuffer table = wrap(tableAname);

  TGet get = new TGet(wrap(rowName));
  assertFalse(handler.exists(table, get));

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname), wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);

  handler.put(table, put);

  assertTrue(handler.exists(table, get));
}
项目:HIndex    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testPutGet() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testPutGet".getBytes();
  ByteBuffer table = wrap(tableAname);

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname), wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);

  put.setColumnValues(columnValues);

  handler.put(table, put);

  TGet get = new TGet(wrap(rowName));

  TResult result = handler.get(table, get);
  assertArrayEquals(rowName, result.getRow());
  List<TColumnValue> returnedColumnValues = result.getColumnValues();
  assertTColumnValuesEqual(columnValues, returnedColumnValues);
}
项目:HIndex    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testIncrement() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testIncrement".getBytes();
  ByteBuffer table = wrap(tableAname);

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname),
    wrap(Bytes.toBytes(1L))));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);
  handler.put(table, put);

  List<TColumnIncrement> incrementColumns = new ArrayList<TColumnIncrement>();
  incrementColumns.add(new TColumnIncrement(wrap(familyAname), wrap(qualifierAname)));
  TIncrement increment = new TIncrement(wrap(rowName), incrementColumns);
  handler.increment(table, increment);

  TGet get = new TGet(wrap(rowName));
  TResult result = handler.get(table, get);

  assertArrayEquals(rowName, result.getRow());
  assertEquals(1, result.getColumnValuesSize());
  TColumnValue columnValue = result.getColumnValues().get(0);
  assertArrayEquals(Bytes.toBytes(2L), columnValue.getValue());
}
项目:HIndex    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testAppend() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testAppend".getBytes();
  ByteBuffer table = wrap(tableAname);
  byte[] v1 = Bytes.toBytes("42");
  byte[] v2 = Bytes.toBytes("23");
  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(v1)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);
  handler.put(table, put);

  List<TColumnValue> appendColumns = new ArrayList<TColumnValue>();
  appendColumns.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(v2)));
  TAppend append = new TAppend(wrap(rowName), appendColumns);
  handler.append(table, append);

  TGet get = new TGet(wrap(rowName));
  TResult result = handler.get(table, get);

  assertArrayEquals(rowName, result.getRow());
  assertEquals(1, result.getColumnValuesSize());
  TColumnValue columnValue = result.getColumnValues().get(0);
  assertArrayEquals(Bytes.add(v1, v2), columnValue.getValue());
}
项目:HIndex    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testMetrics() throws Exception {
  Configuration conf = UTIL.getConfiguration();
  ThriftMetrics metrics = getMetrics(conf);
  THBaseService.Iface handler =
      ThriftHBaseServiceHandler.newInstance(conf, metrics);
  byte[] rowName = "testMetrics".getBytes();
  ByteBuffer table = wrap(tableAname);

  TGet get = new TGet(wrap(rowName));
  assertFalse(handler.exists(table, get));

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname),  wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);

  handler.put(table, put);

  assertTrue(handler.exists(table, get));
  metricsHelper.assertCounter("put_num_ops", 1, metrics.getSource());
  metricsHelper.assertCounter( "exists_num_ops", 2, metrics.getSource());
}
项目:IRIndex    文件:TestThriftHBaseServiceHandler.java   
public void assertTColumnValuesEqual(List<TColumnValue> columnValuesA,
    List<TColumnValue> columnValuesB) {
  assertEquals(columnValuesA.size(), columnValuesB.size());
  Comparator<TColumnValue> comparator = new Comparator<TColumnValue>() {
    @Override
    public int compare(TColumnValue o1, TColumnValue o2) {
      return Bytes.compareTo(Bytes.add(o1.getFamily(), o1.getQualifier()),
          Bytes.add(o2.getFamily(), o2.getQualifier()));
    }
  };
  Collections.sort(columnValuesA, comparator);
  Collections.sort(columnValuesB, comparator);

  for (int i = 0; i < columnValuesA.size(); i++) {
    TColumnValue a = columnValuesA.get(i);
    TColumnValue b = columnValuesB.get(i);
    assertArrayEquals(a.getFamily(), b.getFamily());
    assertArrayEquals(a.getQualifier(), b.getQualifier());
    assertArrayEquals(a.getValue(), b.getValue());
  }
}
项目:IRIndex    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testExists() throws TIOError, TException {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testExists".getBytes();
  ByteBuffer table = wrap(tableAname);

  TGet get = new TGet(wrap(rowName));
  assertFalse(handler.exists(table, get));

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname), wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);

  handler.put(table, put);

  assertTrue(handler.exists(table, get));
}
项目:IRIndex    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testPutGet() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testPutGet".getBytes();
  ByteBuffer table = wrap(tableAname);

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname), wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);

  put.setColumnValues(columnValues);

  handler.put(table, put);

  TGet get = new TGet(wrap(rowName));

  TResult result = handler.get(table, get);
  assertArrayEquals(rowName, result.getRow());
  List<TColumnValue> returnedColumnValues = result.getColumnValues();
  assertTColumnValuesEqual(columnValues, returnedColumnValues);
}
项目:IRIndex    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testIncrement() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = "testIncrement".getBytes();
  ByteBuffer table = wrap(tableAname);

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname),
    wrap(Bytes.toBytes(1L))));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);
  handler.put(table, put);

  List<TColumnIncrement> incrementColumns = new ArrayList<TColumnIncrement>();
  incrementColumns.add(new TColumnIncrement(wrap(familyAname), wrap(qualifierAname)));
  TIncrement increment = new TIncrement(wrap(rowName), incrementColumns);
  handler.increment(table, increment);

  TGet get = new TGet(wrap(rowName));
  TResult result = handler.get(table, get);

  assertArrayEquals(rowName, result.getRow());
  assertEquals(1, result.getColumnValuesSize());
  TColumnValue columnValue = result.getColumnValues().get(0);
  assertArrayEquals(Bytes.toBytes(2L), columnValue.getValue());
}
项目:IRIndex    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testMetrics() throws Exception {
  Configuration conf = UTIL.getConfiguration();
  ThriftMetrics metrics = getMetrics(conf);
  THBaseService.Iface handler =
      ThriftHBaseServiceHandler.newInstance(conf, metrics);
  byte[] rowName = "testMetrics".getBytes();
  ByteBuffer table = wrap(tableAname);

  TGet get = new TGet(wrap(rowName));
  assertFalse(handler.exists(table, get));

  List<TColumnValue> columnValues = new ArrayList<TColumnValue>();
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname),  wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);

  handler.put(table, put);

  assertTrue(handler.exists(table, get));
  logMetrics(metrics);
  verifyMetrics(metrics, "put_num_ops", 1);
  verifyMetrics(metrics, "exists_num_ops", 2);
}
项目:hbase    文件:ThriftUtilities.java   
/**
 * Creates a {@link TResult} (Thrift) from a {@link Result} (HBase).
 *
 * @param in the <code>Result</code> to convert
 *
 * @return converted result, returns an empty result if the input is <code>null</code>
 */
public static TResult resultFromHBase(Result in) {
  Cell[] raw = in.rawCells();
  TResult out = new TResult();
  byte[] row = in.getRow();
  if (row != null) {
    out.setRow(in.getRow());
  }
  List<TColumnValue> columnValues = new ArrayList<>(raw.length);
  for (Cell kv : raw) {
    TColumnValue col = new TColumnValue();
    col.setFamily(CellUtil.cloneFamily(kv));
    col.setQualifier(CellUtil.cloneQualifier(kv));
    col.setTimestamp(kv.getTimestamp());
    col.setValue(CellUtil.cloneValue(kv));
    if (kv.getTagsLength() > 0) {
      col.setTags(PrivateCellUtil.cloneTags(kv));
    }
    columnValues.add(col);
  }
  out.setColumnValues(columnValues);
  return out;
}
项目:hbase    文件:ThriftUtilities.java   
public static Append appendFromThrift(TAppend append) throws IOException {
  Append out = new Append(append.getRow());
  for (TColumnValue column : append.getColumns()) {
    out.addColumn(column.getFamily(), column.getQualifier(), column.getValue());
  }

  if (append.isSetAttributes()) {
    addAttributes(out, append.getAttributes());
  }

  if (append.isSetDurability()) {
    out.setDurability(durabilityFromThrift(append.getDurability()));
  }

  if(append.getCellVisibility() != null) {
    out.setCellVisibility(new CellVisibility(append.getCellVisibility().getExpression()));
  }

  return out;
}
项目:hbase    文件:TestThriftHBaseServiceHandlerWithLabels.java   
public void assertTColumnValuesEqual(List<TColumnValue> columnValuesA,
    List<TColumnValue> columnValuesB) {
  assertEquals(columnValuesA.size(), columnValuesB.size());
  Comparator<TColumnValue> comparator = new Comparator<TColumnValue>() {
    @Override
    public int compare(TColumnValue o1, TColumnValue o2) {
      return Bytes.compareTo(Bytes.add(o1.getFamily(), o1.getQualifier()),
          Bytes.add(o2.getFamily(), o2.getQualifier()));
    }
  };
  Collections.sort(columnValuesA, comparator);
  Collections.sort(columnValuesB, comparator);

  for (int i = 0; i < columnValuesA.size(); i++) {
    TColumnValue a = columnValuesA.get(i);
    TColumnValue b = columnValuesB.get(i);
    assertArrayEquals(a.getFamily(), b.getFamily());
    assertArrayEquals(a.getQualifier(), b.getQualifier());
    assertArrayEquals(a.getValue(), b.getValue());
  }
}
项目:hbase    文件:TestThriftHBaseServiceHandlerWithReadOnly.java   
@Test
public void testPutWithReadOnly() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  ByteBuffer table = wrap(tableAname);
  byte[] rowName = Bytes.toBytes("testPut");

  List<TColumnValue> columnValues = new ArrayList<>(2);
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname), wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);

  boolean exceptionCaught = false;
  try {
    handler.put(table, put);
  } catch (TIOError e) {
    exceptionCaught = true;
    assertTrue(e.getCause() instanceof DoNotRetryIOException);
    assertEquals("Thrift Server is in Read-only mode.", e.getMessage());
  } finally {
    assertTrue(exceptionCaught);
  }
}
项目:hbase    文件:TestThriftHBaseServiceHandlerWithReadOnly.java   
@Test
public void testPutMultipleWithReadOnly() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  ByteBuffer table = wrap(tableAname);
  byte[] rowName1 = Bytes.toBytes("testPutMultiple1");
  byte[] rowName2 = Bytes.toBytes("testPutMultiple2");

  List<TColumnValue> columnValues = new ArrayList<>(2);
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname), wrap(valueBname)));
  List<TPut> puts = new ArrayList<>(2);
  puts.add(new TPut(wrap(rowName1), columnValues));
  puts.add(new TPut(wrap(rowName2), columnValues));

  boolean exceptionCaught = false;
  try {
    handler.putMultiple(table, puts);
  } catch (TIOError e) {
    exceptionCaught = true;
    assertTrue(e.getCause() instanceof DoNotRetryIOException);
    assertEquals("Thrift Server is in Read-only mode.", e.getMessage());
  } finally {
    assertTrue(exceptionCaught);
  }
}
项目:hbase    文件:TestThriftHBaseServiceHandlerWithReadOnly.java   
@Test
public void testAppendWithReadOnly() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = Bytes.toBytes("testAppend");
  ByteBuffer table = wrap(tableAname);
  byte[] v1 = Bytes.toBytes("42");

  List<TColumnValue> appendColumns = new ArrayList<>(1);
  appendColumns.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(v1)));
  TAppend append = new TAppend(wrap(rowName), appendColumns);

  boolean exceptionCaught = false;
  try {
    handler.append(table, append);
  } catch (TIOError e) {
    exceptionCaught = true;
    assertTrue(e.getCause() instanceof DoNotRetryIOException);
    assertEquals("Thrift Server is in Read-only mode.", e.getMessage());
  } finally {
    assertTrue(exceptionCaught);
  }
}
项目:hbase    文件:TestThriftHBaseServiceHandler.java   
public void assertTColumnValuesEqual(List<TColumnValue> columnValuesA,
    List<TColumnValue> columnValuesB) {
  assertEquals(columnValuesA.size(), columnValuesB.size());
  Comparator<TColumnValue> comparator = new Comparator<TColumnValue>() {
    @Override
    public int compare(TColumnValue o1, TColumnValue o2) {
      return Bytes.compareTo(Bytes.add(o1.getFamily(), o1.getQualifier()),
          Bytes.add(o2.getFamily(), o2.getQualifier()));
    }
  };
  Collections.sort(columnValuesA, comparator);
  Collections.sort(columnValuesB, comparator);

  for (int i = 0; i < columnValuesA.size(); i++) {
    TColumnValue a = columnValuesA.get(i);
    TColumnValue b = columnValuesB.get(i);
    assertTColumnValueEqual(a, b);
  }
}
项目:hbase    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testExists() throws TIOError, TException {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = Bytes.toBytes("testExists");
  ByteBuffer table = wrap(tableAname);

  TGet get = new TGet(wrap(rowName));
  assertFalse(handler.exists(table, get));

  List<TColumnValue> columnValues = new ArrayList<>(2);
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname), wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);

  handler.put(table, put);

  assertTrue(handler.exists(table, get));
}
项目:hbase    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testPutGet() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = Bytes.toBytes("testPutGet");
  ByteBuffer table = wrap(tableAname);

  List<TColumnValue> columnValues = new ArrayList<>(2);
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname), wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);

  put.setColumnValues(columnValues);

  handler.put(table, put);

  TGet get = new TGet(wrap(rowName));

  TResult result = handler.get(table, get);
  assertArrayEquals(rowName, result.getRow());
  List<TColumnValue> returnedColumnValues = result.getColumnValues();
  assertTColumnValuesEqual(columnValues, returnedColumnValues);
}
项目:hbase    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testIncrement() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = Bytes.toBytes("testIncrement");
  ByteBuffer table = wrap(tableAname);

  List<TColumnValue> columnValues = new ArrayList<>(1);
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname),
    wrap(Bytes.toBytes(1L))));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);
  handler.put(table, put);

  List<TColumnIncrement> incrementColumns = new ArrayList<>(1);
  incrementColumns.add(new TColumnIncrement(wrap(familyAname), wrap(qualifierAname)));
  TIncrement increment = new TIncrement(wrap(rowName), incrementColumns);
  handler.increment(table, increment);

  TGet get = new TGet(wrap(rowName));
  TResult result = handler.get(table, get);

  assertArrayEquals(rowName, result.getRow());
  assertEquals(1, result.getColumnValuesSize());
  TColumnValue columnValue = result.getColumnValues().get(0);
  assertArrayEquals(Bytes.toBytes(2L), columnValue.getValue());
}
项目:hbase    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testAppend() throws Exception {
  ThriftHBaseServiceHandler handler = createHandler();
  byte[] rowName = Bytes.toBytes("testAppend");
  ByteBuffer table = wrap(tableAname);
  byte[] v1 = Bytes.toBytes("42");
  byte[] v2 = Bytes.toBytes("23");
  List<TColumnValue> columnValues = new ArrayList<>(1);
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(v1)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);
  handler.put(table, put);

  List<TColumnValue> appendColumns = new ArrayList<>(1);
  appendColumns.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(v2)));
  TAppend append = new TAppend(wrap(rowName), appendColumns);
  handler.append(table, append);

  TGet get = new TGet(wrap(rowName));
  TResult result = handler.get(table, get);

  assertArrayEquals(rowName, result.getRow());
  assertEquals(1, result.getColumnValuesSize());
  TColumnValue columnValue = result.getColumnValues().get(0);
  assertArrayEquals(Bytes.add(v1, v2), columnValue.getValue());
}
项目:hbase    文件:TestThriftHBaseServiceHandler.java   
@Test
public void testMetrics() throws Exception {
  Configuration conf = UTIL.getConfiguration();
  ThriftMetrics metrics = getMetrics(conf);
  ThriftHBaseServiceHandler hbaseHandler = createHandler();
  THBaseService.Iface handler =
      ThriftHBaseServiceHandler.newInstance(hbaseHandler, metrics);
  byte[] rowName = Bytes.toBytes("testMetrics");
  ByteBuffer table = wrap(tableAname);

  TGet get = new TGet(wrap(rowName));
  assertFalse(handler.exists(table, get));

  List<TColumnValue> columnValues = new ArrayList<>(2);
  columnValues.add(new TColumnValue(wrap(familyAname), wrap(qualifierAname), wrap(valueAname)));
  columnValues.add(new TColumnValue(wrap(familyBname), wrap(qualifierBname),  wrap(valueBname)));
  TPut put = new TPut(wrap(rowName), columnValues);
  put.setColumnValues(columnValues);

  handler.put(table, put);

  assertTrue(handler.exists(table, get));
  metricsHelper.assertCounter("put_num_ops", 1, metrics.getSource());
  metricsHelper.assertCounter("exists_num_ops", 2, metrics.getSource());
}