Search in sources :

Example 6 with IDBI

use of org.skife.jdbi.v2.IDBI in project presto by prestodb.

the class TestOrcStorageManager method setup.

@BeforeMethod
public void setup() throws Exception {
    temporary = createTempDir();
    File directory = new File(temporary, "data");
    storageService = new FileStorageService(directory);
    storageService.start();
    File backupDirectory = new File(temporary, "backup");
    fileBackupStore = new FileBackupStore(backupDirectory);
    fileBackupStore.start();
    backupStore = Optional.of(fileBackupStore);
    IDBI dbi = new DBI("jdbc:h2:mem:test" + System.nanoTime());
    dummyHandle = dbi.open();
    createTablesWithRetry(dbi);
    ShardManager shardManager = createShardManager(dbi);
    Duration discoveryInterval = new Duration(5, TimeUnit.MINUTES);
    recoveryManager = new ShardRecoveryManager(storageService, backupStore, nodeManager, shardManager, discoveryInterval, 10);
    shardRecorder = new InMemoryShardRecorder();
}
Also used : IDBI(org.skife.jdbi.v2.IDBI) FileBackupStore(com.facebook.presto.raptor.backup.FileBackupStore) IDBI(org.skife.jdbi.v2.IDBI) DBI(org.skife.jdbi.v2.DBI) ShardManager(com.facebook.presto.raptor.metadata.ShardManager) TestDatabaseShardManager.createShardManager(com.facebook.presto.raptor.metadata.TestDatabaseShardManager.createShardManager) Duration(io.airlift.units.Duration) FileAssert.assertFile(org.testng.FileAssert.assertFile) File(java.io.File) BeforeMethod(org.testng.annotations.BeforeMethod)

Example 7 with IDBI

use of org.skife.jdbi.v2.IDBI in project presto by prestodb.

the class TestShardRecovery method setup.

@BeforeMethod
public void setup() throws Exception {
    temporary = createTempDir();
    File directory = new File(temporary, "data");
    File backupDirectory = new File(temporary, "backup");
    backupStore = new FileBackupStore(backupDirectory);
    backupStore.start();
    storageService = new FileStorageService(directory);
    storageService.start();
    IDBI dbi = new DBI("jdbc:h2:mem:test" + System.nanoTime());
    dummyHandle = dbi.open();
    createTablesWithRetry(dbi);
    ShardManager shardManager = createShardManager(dbi);
    recoveryManager = createShardRecoveryManager(storageService, Optional.of(backupStore), shardManager);
}
Also used : IDBI(org.skife.jdbi.v2.IDBI) FileBackupStore(com.facebook.presto.raptor.backup.FileBackupStore) IDBI(org.skife.jdbi.v2.IDBI) DBI(org.skife.jdbi.v2.DBI) ShardManager(com.facebook.presto.raptor.metadata.ShardManager) TestDatabaseShardManager.createShardManager(com.facebook.presto.raptor.metadata.TestDatabaseShardManager.createShardManager) File.createTempFile(java.io.File.createTempFile) File(java.io.File) BeforeMethod(org.testng.annotations.BeforeMethod)

Example 8 with IDBI

use of org.skife.jdbi.v2.IDBI in project presto by prestodb.

the class TestShardCompactor method setup.

@BeforeMethod
public void setup() throws Exception {
    temporary = createTempDir();
    IDBI dbi = new DBI("jdbc:h2:mem:test" + System.nanoTime());
    dummyHandle = dbi.open();
    storageManager = createOrcStorageManager(dbi, temporary, MAX_SHARD_ROWS);
    compactor = new ShardCompactor(storageManager, READER_ATTRIBUTES);
}
Also used : IDBI(org.skife.jdbi.v2.IDBI) IDBI(org.skife.jdbi.v2.IDBI) DBI(org.skife.jdbi.v2.DBI) BeforeMethod(org.testng.annotations.BeforeMethod)

Aggregations

IDBI (org.skife.jdbi.v2.IDBI)7 DBI (org.skife.jdbi.v2.DBI)6 BeforeMethod (org.testng.annotations.BeforeMethod)4 FileBackupStore (com.facebook.presto.raptor.backup.FileBackupStore)2 ShardManager (com.facebook.presto.raptor.metadata.ShardManager)2 TestDatabaseShardManager.createShardManager (com.facebook.presto.raptor.metadata.TestDatabaseShardManager.createShardManager)2 File (java.io.File)2 SQLException (java.sql.SQLException)2 Distribution (com.facebook.presto.raptor.metadata.Distribution)1 ForMetadata (com.facebook.presto.raptor.metadata.ForMetadata)1 TableColumn (com.facebook.presto.raptor.metadata.TableColumn)1 Domain (com.facebook.presto.spi.predicate.Domain)1 TupleDomain (com.facebook.presto.spi.predicate.TupleDomain)1 VisibleForTesting (com.google.common.annotations.VisibleForTesting)1 ImmutableList (com.google.common.collect.ImmutableList)1 Provides (com.google.inject.Provides)1 Duration (io.airlift.units.Duration)1 DataSegment (io.druid.timeline.DataSegment)1 VersionedIntervalTimeline (io.druid.timeline.VersionedIntervalTimeline)1 File.createTempFile (java.io.File.createTempFile)1