/*
 * Copyright (c) 2012 Continuuity Inc. All rights reserved.
 */
package com.continuuity.data.engine.leveldb;

import com.continuuity.api.common.Bytes;
import com.continuuity.api.data.OperationException;
import com.continuuity.data.operation.StatusCode;
import com.continuuity.data.table.OrderedVersionedColumnarTable;
import com.continuuity.data.table.SimpleOVCTableHandle;
import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import com.google.inject.Inject;
import com.google.inject.name.Named;

import java.util.concurrent.ExecutionException;

/**
 * This class implements the table handle for LevelDB.
 */
public class LevelDBOVCTableHandle extends SimpleOVCTableHandle {
  @Inject
  @Named("LevelDBOVCTableHandleBasePath")
  private String basePath;

  @Inject
  @Named("LevelDBOVCTableHandleBlockSize")
  private Integer blockSize;

  @Inject
  @Named("LevelDBOVCTableHandleCacheSize")
  private Long cacheSize;

  private final LoadingCache<String, LevelDBOVCTable> tableCache;

  /**
   * This class is a singleton.
   * We have to guard against creating multiple instances because level db supports only one active client
   */
  private static final LevelDBOVCTableHandle INSTANCE = new LevelDBOVCTableHandle();

  private LevelDBOVCTableHandle() {
    tableCache = CacheBuilder.newBuilder().build(new CacheLoader<String, LevelDBOVCTable>() {
      @Override
      public LevelDBOVCTable load(String tableName) throws Exception {
        return openOrCreateTable(tableName);
      }
    });
  }

  public static LevelDBOVCTableHandle getInstance() {
    return INSTANCE;
  }

  @Override
  protected OrderedVersionedColumnarTable createNewTable(byte[] tableName) throws OperationException {
    // Do the same action as openTable, which gets the table through the LoadingCache.
    return openTable(tableName);
  }

  @Override
  protected OrderedVersionedColumnarTable openTable(byte[] tableName) throws OperationException {
    try {
      return tableCache.get(Bytes.toString(tableName));
    } catch (ExecutionException e) {
      Throwable cause = e.getCause();
      if (cause instanceof OperationException) {
        throw (OperationException) cause;
      }
      throw new OperationException(StatusCode.INTERNAL_ERROR, cause.getMessage(), cause);
    }
  }

  /**
   * Opens the table if it already exists or creates a new one if it doesn't.
   * @param tableName The name of the table.
   * @return A LevelDBOVCTable.
   * @throws OperationException If there is any error when try to open or create the table.
   */
  private LevelDBOVCTable openOrCreateTable(String tableName) throws OperationException {
    LevelDBOVCTable table = new LevelDBOVCTable(basePath, tableName, blockSize, cacheSize);

    if (table.openTable()) {
      return table;
    }
    table.initializeTable();
    return table;
  }

  @Override
  public String getName() {
    return "leveldb";
  }
}

/*
 * Copyright (c) 2012 Continuuity Inc. All rights reserved.
 */
package com.continuuity.data.engine.leveldb;

import com.continuuity.api.common.Bytes;
import com.continuuity.api.data.OperationException;
import com.continuuity.data.operation.StatusCode;
import com.continuuity.data.table.OrderedVersionedColumnarTable;
import com.continuuity.data.table.SimpleOVCTableHandle;
import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import com.google.inject.Inject;
import com.google.inject.name.Named;

import java.util.concurrent.ExecutionException;

/**
 * This class implements the table handle for LevelDB.
 */
public class LevelDBOVCTableHandle extends SimpleOVCTableHandle {
  @Inject
  @Named("LevelDBOVCTableHandleBasePath")
  private String basePath;

  @Inject
  @Named("LevelDBOVCTableHandleBlockSize")
  private Integer blockSize;

  @Inject
  @Named("LevelDBOVCTableHandleCacheSize")
  private Long cacheSize;

  private final LoadingCache<String, LevelDBOVCTable> tableCache;

  /**
   * This class is a singleton.
   * We have to guard against creating multiple instances because level db supports only one active client
   */
  private static final LevelDBOVCTableHandle INSTANCE = new LevelDBOVCTableHandle();

  private LevelDBOVCTableHandle() {
    tableCache = CacheBuilder.newBuilder().build(new CacheLoader<String, LevelDBOVCTable>() {
      @Override
      public LevelDBOVCTable load(String tableName) throws Exception {
        return openOrCreateTable(tableName);
      }
    });
  }

  public static LevelDBOVCTableHandle getInstance() {
    return INSTANCE;
  }

  @Override
  protected OrderedVersionedColumnarTable createNewTable(byte[] tableName) throws OperationException {
    try {
      return tableCache.get(Bytes.toString(tableName));
    } catch (ExecutionException e) {
      Throwable cause = e.getCause();
      if (cause instanceof OperationException) {
        throw (OperationException) cause;
      }
      throw new OperationException(StatusCode.INTERNAL_ERROR, cause.getMessage(), cause);
    }
  }

  @Override
  protected OrderedVersionedColumnarTable openTable(byte[] tableName) throws OperationException {
    return tableCache.getIfPresent(Bytes.toString(tableName));
  }

  /**
   * Opens the table if it already exists or creates a new one if it doesn't.
   * @param tableName The name of the table.
   * @return A LevelDBOVCTable.
   * @throws OperationException If there is any error when try to open or create the table.
   */
  private LevelDBOVCTable openOrCreateTable(String tableName) throws OperationException {
    LevelDBOVCTable table = new LevelDBOVCTable(basePath, tableName, blockSize, cacheSize);

    if (table.openTable()) {
      return table;
    }
    table.initializeTable();
    return table;
  }

  @Override
  public String getName() {
    return "leveldb";
  }
}

Make it preserve old behavior.