TestTHLog.java

/**
 * Copyright 2009 The Apache Software Foundation
 *
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */
package org.apache.hadoop.hbase.regionserver.transactional;

import java.io.IOException;
import java.util.List;
import java.util.Map;

import org.apache.hadoop.fs.Path;
import org.apache.hadoop.hbase.HBaseTestCase;
import org.apache.hadoop.hbase.HConstants;
import org.apache.hadoop.hbase.HRegionInfo;
import org.apache.hadoop.hbase.HTableDescriptor;
import org.apache.hadoop.hbase.KeyValue;
import org.apache.hadoop.hbase.client.Put;
import org.apache.hadoop.hbase.util.Bytes;
import org.apache.hadoop.hdfs.MiniDFSCluster;

/** JUnit test case for HLog */
public class TestTHLog extends HBaseTestCase implements
    HConstants {
  private Path dir;
  private MiniDFSCluster cluster;

  final byte[] tableName = Bytes.toBytes("tablename");
  final HTableDescriptor tableDesc = new HTableDescriptor(tableName);
  final HRegionInfo regionInfo = new HRegionInfo(tableDesc,
      HConstants.EMPTY_START_ROW, HConstants.EMPTY_END_ROW);
  final byte[] row1 = Bytes.toBytes("row1");
  final byte[] val1 = Bytes.toBytes("val1");
  final byte[] row2 = Bytes.toBytes("row2");
  final byte[] val2 = Bytes.toBytes("val2");
  final byte[] row3 = Bytes.toBytes("row3");
  final byte[] val3 = Bytes.toBytes("val3");
  final byte[] family = Bytes.toBytes("family");
  final byte[] column = Bytes.toBytes("a");

  @Override
  public void setUp() throws Exception {
    cluster = new MiniDFSCluster(conf, 2, true, (String[]) null);
    // Set the hbase.rootdir to be the home directory in mini dfs.
    this.conf.set(HConstants.HBASE_DIR, this.cluster.getFileSystem()
        .getHomeDirectory().toString());
    super.setUp();
    this.dir = new Path("/hbase", getName());
    if (fs.exists(dir)) {
      fs.delete(dir, true);
    }
  }

  @Override
  public void tearDown() throws Exception {
    if (this.fs.exists(this.dir)) {
      this.fs.delete(this.dir, true);
    }
    shutdownDfs(cluster);
    super.tearDown();
  }

  /**
   * @throws IOException
   */
  public void testSingleCommit() throws IOException {

    THLog log = new THLog(fs, dir, this.conf, null);
    THLogRecoveryManager logRecoveryMangaer = new THLogRecoveryManager(fs,
        regionInfo, conf);

    // Write columns named 1, 2, 3, etc. and then values of single byte
    // 1, 2, 3...
    long transactionId = 1;
    log.writeUpdateToLog(regionInfo, transactionId, new Put(row1).add(family,
        column, val1));
    log.writeUpdateToLog(regionInfo, transactionId, new Put(row2).add(family,
        column, val2));
    log.writeUpdateToLog(regionInfo, transactionId, new Put(row3).add(family,
        column, val3));

    log.writeCommitToLog(regionInfo, transactionId);

    // log.completeCacheFlush(regionName, tableName, logSeqId);

    log.close();
    Path filename = log.computeFilename(log.getFilenum());

    Map<Long, List<KeyValue>> commits = logRecoveryMangaer.getCommitsFromLog(
        filename, -1, null);

    assertNull(commits);

  }

  /**
   * @throws IOException
   */
  public void testSingleAbort() throws IOException {

    THLog log = new THLog(fs, dir, this.conf, null);
    THLogRecoveryManager logRecoveryMangaer = new THLogRecoveryManager(fs,
        regionInfo, conf);

    long transactionId = 1;
    log.writeUpdateToLog(regionInfo, transactionId, new Put(row1).add(family,
        column, val1));
    log.writeUpdateToLog(regionInfo, transactionId, new Put(row2).add(family,
        column, val2));
    log.writeUpdateToLog(regionInfo, transactionId, new Put(row3).add(family,
        column, val3));

    log.writeAbortToLog(regionInfo, transactionId);
    // log.completeCacheFlush(regionName, tableName, logSeqId);

    log.close();
    Path filename = log.computeFilename(log.getFilenum());

    Map<Long, List<KeyValue>> commits = logRecoveryMangaer.getCommitsFromLog(
        filename, -1, null);

    assertNull(commits);
  }

  /**
   * @throws IOException
   */
  public void testInterlievedCommits() throws IOException {

    THLog log = new THLog(fs, dir, this.conf, null);
    THLogRecoveryManager logMangaer = new THLogRecoveryManager(fs, regionInfo,
        conf);

    long transaction1Id = 1;
    long transaction2Id = 2;

    log.writeUpdateToLog(regionInfo, transaction1Id, new Put(row1).add(family,
        column, val1));

    log.writeUpdateToLog(regionInfo, transaction2Id, new Put(row2).add(family,
        column, val2));

    log.writeUpdateToLog(regionInfo, transaction1Id, new Put(row3).add(family,
        column, val3));

    log.writeCommitToLog(regionInfo, transaction1Id);
    log.writeCommitToLog(regionInfo, transaction2Id);

    // log.completeCacheFlush(regionName, tableName, logSeqId);

    log.close();
    Path filename = log.computeFilename(log.getFilenum());

    Map<Long, List<KeyValue>> commits = logMangaer.getCommitsFromLog(filename,
        -1, null);

    assertNull(commits);
  }

  /**
   * @throws IOException
   */
  public void testInterlievedAbortCommit() throws IOException {

    THLog log = new THLog(fs, dir, this.conf, null);
    THLogRecoveryManager logMangaer = new THLogRecoveryManager(fs, regionInfo,
        conf);

    long transaction1Id = 1;
    long transaction2Id = 2;

    log.writeUpdateToLog(regionInfo, transaction1Id, new Put(row1).add(family,
        column, val1));

    log.writeUpdateToLog(regionInfo, transaction2Id, new Put(row2).add(family,
        column, val2));
    log.writeAbortToLog(regionInfo, transaction2Id);

    log.writeUpdateToLog(regionInfo, transaction1Id, new Put(row3).add(family,
        column, val3));

    log.writeCommitToLog(regionInfo, transaction1Id);

    // log.completeCacheFlush(regionName, tableName, logSeqId);

    log.close();
    Path filename = log.computeFilename(log.getFilenum());

    Map<Long, List<KeyValue>> commits = logMangaer.getCommitsFromLog(filename,
        -1, null);

    assertNull(commits);
  }

  /**
   * @throws IOException
   */
  public void testInterlievedCommitAbort() throws IOException {

    THLog log = new THLog(fs, dir, this.conf, null);
    THLogRecoveryManager logMangaer = new THLogRecoveryManager(fs, regionInfo,
        conf);

    long transaction1Id = 1;
    long transaction2Id = 2;

    log.writeUpdateToLog(regionInfo, transaction1Id, new Put(row1).add(family,
        column, val1));

    log.writeUpdateToLog(regionInfo, transaction2Id, new Put(row2).add(family,
        column, val2));
    log.writeCommitToLog(regionInfo, transaction2Id);

    log.writeUpdateToLog(regionInfo, transaction1Id, new Put(row3).add(family,
        column, val3));

    log.writeAbortToLog(regionInfo, transaction1Id);

    // log.completeCacheFlush(regionName, tableName, logSeqId);

    log.close();
    Path filename = log.computeFilename(log.getFilenum());

    Map<Long, List<KeyValue>> commits = logMangaer.getCommitsFromLog(filename,
        -1, null);

    assertNull(commits);
  }

  // FIXME Cannot do this test without a global transacton manager
  // public void testMissingCommit() {
  // fail();
  // }

  // FIXME Cannot do this test without a global transacton manager
  // public void testMissingAbort() {
  // fail();
  // }

}