Example #1
0
  /**
   * This tests writing messages and removing them in a bulk delete.
   *
   * @throws HBaseException
   */
  @Test
  public void writeAndRemoveRows() throws HBaseException {
    MessagePersister mp = new MessagePersister();
    HBaseFacade hbs = new HBaseFacade();

    mp.writeMessage(messageKey1, data1);
    mp.writeMessage(messageKey2, data2);
    mp.writeMessage(messageKey3, data2);
    mp.writeMessage(messageKey4, data2);

    Set<Object> keys = new HashSet<Object>();
    keys.add(messageKey1);
    keys.add(messageKey2);
    keys.add(messageKey3);
    keys.add(messageKey4);

    hbs.removeRows(MessagePersister.TABLE_MESSAGE, keys);

    byte[] result = mp.readMessage(messageKey1);
    assertNull(result);
    result = mp.readMessage(messageKey2);
    assertNull(result);
    result = mp.readMessage(messageKey3);
    assertNull(result);
    result = mp.readMessage(messageKey4);
    assertNull(result);
  }
Example #2
0
  /**
   * This tests writing messages, scanning the table, and removing the messages.
   *
   * @throws HBaseException
   */
  @Test
  public void writeAndScanMessageData() throws HBaseException {
    MessagePersister mp = new MessagePersister();
    HBaseFacade hbs = new HBaseFacade();

    mp.writeMessage(messageKey1, data1);
    mp.writeMessage(messageKey2, data2);
    mp.writeMessage(messageKey3, data2);
    mp.writeMessage(messageKey4, data2);
    mp.writeMessage(messageKey5, data2);
    mp.writeMessage(messageKey6, data2);
    mp.writeMessage(messageKey7, data2);
    mp.writeMessage(messageKey8, data2);
    mp.writeMessage(messageKey9, data2);
    mp.writeMessage(messageKey10, data2);

    Map<String, byte[]> dataMap =
        hbs.scan(
            MessagePersister.TABLE_MESSAGE,
            10,
            MessagePersister.COL_FAMILY_DATA,
            MessagePersister.QUANTIFIER_VALUE);

    assertTrue(Arrays.equals(dataMap.get(messageKey1), data1));
    assertTrue(Arrays.equals(dataMap.get(messageKey2), data2));
    assertTrue(Arrays.equals(dataMap.get(messageKey3), data2));
    assertTrue(Arrays.equals(dataMap.get(messageKey4), data2));
    assertTrue(Arrays.equals(dataMap.get(messageKey5), data2));
    assertTrue(Arrays.equals(dataMap.get(messageKey6), data2));
    assertTrue(Arrays.equals(dataMap.get(messageKey7), data2));
    assertTrue(Arrays.equals(dataMap.get(messageKey8), data2));
    assertTrue(Arrays.equals(dataMap.get(messageKey9), data2));
    assertTrue(Arrays.equals(dataMap.get(messageKey10), data2));

    mp.removeMessage(messageKey1);
    mp.removeMessage(messageKey2);
    mp.removeMessage(messageKey3);
    mp.removeMessage(messageKey4);
    mp.removeMessage(messageKey5);
    mp.removeMessage(messageKey6);
    mp.removeMessage(messageKey7);
    mp.removeMessage(messageKey8);
    mp.removeMessage(messageKey9);
    mp.removeMessage(messageKey10);
  }
Example #3
0
  /**
   * This tests writing messages, scanning the table for the keys, and removing the messages.
   *
   * @throws HBaseException
   */
  @Test
  public void writeAndScanMessageKeys() throws HBaseException {
    MessagePersister mp = new MessagePersister();
    HBaseFacade hbs = new HBaseFacade();

    mp.writeMessage(messageKey1, data1);
    mp.writeMessage(messageKey2, data2);
    mp.writeMessage(messageKey3, data2);
    mp.writeMessage(messageKey4, data2);
    mp.writeMessage(messageKey5, data2);
    mp.writeMessage(messageKey6, data2);
    mp.writeMessage(messageKey7, data2);
    mp.writeMessage(messageKey8, data2);
    mp.writeMessage(messageKey9, data2);
    mp.writeMessage(messageKey10, data2);

    Set<Object> keys = hbs.scanForKeys(MessagePersister.TABLE_MESSAGE);
    assertTrue("Did not return a key", keys.contains(messageKey1));
    assertTrue("Did not return a key", keys.contains(messageKey2));
    assertTrue("Did not return a key", keys.contains(messageKey3));
    assertTrue("Did not return a key", keys.contains(messageKey4));
    assertTrue("Did not return a key", keys.contains(messageKey5));
    assertTrue("Did not return a key", keys.contains(messageKey6));
    assertTrue("Did not return a key", keys.contains(messageKey7));
    assertTrue("Did not return a key", keys.contains(messageKey8));
    assertTrue("Did not return a key", keys.contains(messageKey9));
    assertTrue("Did not return a key", keys.contains(messageKey10));

    mp.removeMessage(messageKey1);
    mp.removeMessage(messageKey2);
    mp.removeMessage(messageKey3);
    mp.removeMessage(messageKey4);
    mp.removeMessage(messageKey5);
    mp.removeMessage(messageKey6);
    mp.removeMessage(messageKey7);
    mp.removeMessage(messageKey8);
    mp.removeMessage(messageKey9);
    mp.removeMessage(messageKey10);
  }
Example #4
0
  /**
   * IMPORTANT - cannot run this test two times in a row without disabling and dropping the test
   * table.
   *
   * @throws HBaseException
   */
  @Test
  public void createTable() throws HBaseException {
    String testCF = "testCF";
    String testDataVal = "This is some data.";
    String testTable = "testTable";
    String testField = "testField";
    String testKey = "testKey";

    List<String> colFamilies = new ArrayList<String>();
    colFamilies.add(testCF);

    HBaseFacade hbs = new HBaseFacade();
    try {
      hbs.createTable(testTable, colFamilies);
    } catch (HBaseException ex) {
      if (ex.getCause() instanceof TableExistsException) {
        System.err.println(
            "Cannot test createTable because the " + testTable + " table already exists.");
        return;
      } else {
        throw ex;
      }
    }

    Map<String, Map<String, byte[]>> testData = new HashMap<String, Map<String, byte[]>>();

    Map<String, byte[]> dataCells1 = new HashMap<String, byte[]>();
    dataCells1.put(testField, testDataVal.getBytes());
    testData.put(colFamilies.get(0), dataCells1);
    hbs.addRow(testTable, testKey, testData);

    testData = hbs.readRow(testTable, testKey, colFamilies);
    assertTrue(
        Arrays.equals(testData.get(colFamilies.get(0)).get(testField), testDataVal.getBytes()));

    hbs.removeRow(testTable, testKey);
  }