use of com.palantir.atlasdb.keyvalue.api.Namespace in project atlasdb by palantir.
the class StreamingTable method getTableRef.
public TableReference getTableRef() {
Namespace namespace = Namespace.create("default", Namespace.UNCHECKED_NAME);
String tableName = "blobs";
return TableReference.create(namespace, tableName);
}
use of com.palantir.atlasdb.keyvalue.api.Namespace in project atlasdb by palantir.
the class TableMigratorTest method testMigrationToDifferentKvs.
// Table/IndexDefinition syntax
@SuppressWarnings({ "checkstyle:Indentation", "checkstyle:RightCurly" })
@Test
public void testMigrationToDifferentKvs() throws TableMappingNotFoundException {
final TableReference tableRef = TableReference.create(Namespace.DEFAULT_NAMESPACE, "table");
final TableReference namespacedTableRef = TableReference.createFromFullyQualifiedName("namespace." + tableRef.getTablename());
TableDefinition definition = new TableDefinition() {
{
rowName();
rowComponent("r", ValueType.BLOB);
columns();
column("c", "c", ValueType.BLOB);
}
};
keyValueService.createTable(tableRef, definition.toTableMetadata().persistToBytes());
keyValueService.createTable(namespacedTableRef, definition.toTableMetadata().persistToBytes());
keyValueService.putMetadataForTable(namespacedTableRef, definition.toTableMetadata().persistToBytes());
final Cell theCell = Cell.create(PtBytes.toBytes("r1"), PtBytes.toBytes("c"));
final byte[] theValue = PtBytes.toBytes("v1");
txManager.runTaskWithRetry((TransactionTask<Void, RuntimeException>) txn -> {
Map<Cell, byte[]> values = ImmutableMap.of(theCell, theValue);
txn.put(tableRef, values);
txn.put(namespacedTableRef, values);
return null;
});
final InMemoryKeyValueService kvs2 = new InMemoryKeyValueService(false);
final ConflictDetectionManager cdm2 = ConflictDetectionManagers.createWithNoConflictDetection();
final SweepStrategyManager ssm2 = SweepStrategyManagers.completelyConservative(kvs2);
final TestTransactionManagerImpl txManager2 = new TestTransactionManagerImpl(kvs2, timestampService, lockClient, lockService, transactionService, cdm2, ssm2, MultiTableSweepQueueWriter.NO_OP);
kvs2.createTable(tableRef, definition.toTableMetadata().persistToBytes());
kvs2.createTable(namespacedTableRef, definition.toTableMetadata().persistToBytes());
TableReference checkpointTable = TableReference.create(Namespace.DEFAULT_NAMESPACE, "checkpoint");
GeneralTaskCheckpointer checkpointer = new GeneralTaskCheckpointer(checkpointTable, kvs2, txManager2);
for (final TableReference name : Lists.newArrayList(tableRef, namespacedTableRef)) {
TransactionRangeMigrator rangeMigrator = new TransactionRangeMigratorBuilder().srcTable(name).readTxManager(txManager).txManager(txManager2).checkpointer(checkpointer).build();
TableMigratorBuilder builder = new TableMigratorBuilder().srcTable(name).partitions(1).executor(PTExecutors.newSingleThreadExecutor()).checkpointer(checkpointer).rangeMigrator(rangeMigrator);
TableMigrator migrator = builder.build();
migrator.migrate();
}
checkpointer.deleteCheckpoints();
final ConflictDetectionManager verifyCdm = ConflictDetectionManagers.createWithNoConflictDetection();
final SweepStrategyManager verifySsm = SweepStrategyManagers.completelyConservative(kvs2);
final TestTransactionManagerImpl verifyTxManager = new TestTransactionManagerImpl(kvs2, timestampService, lockClient, lockService, transactionService, verifyCdm, verifySsm, MultiTableSweepQueueWriter.NO_OP);
final MutableLong count = new MutableLong();
for (final TableReference name : Lists.newArrayList(tableRef, namespacedTableRef)) {
verifyTxManager.runTaskReadOnly((TransactionTask<Void, RuntimeException>) txn -> {
BatchingVisitable<RowResult<byte[]>> bv = txn.getRange(name, RangeRequest.all());
bv.batchAccept(1000, AbortingVisitors.batching(new AbortingVisitor<RowResult<byte[]>, RuntimeException>() {
@Override
public boolean visit(RowResult<byte[]> item) throws RuntimeException {
Iterable<Entry<Cell, byte[]>> cells = item.getCells();
Entry<Cell, byte[]> entry = Iterables.getOnlyElement(cells);
Assert.assertEquals(theCell, entry.getKey());
Assert.assertArrayEquals(theValue, entry.getValue());
count.increment();
return true;
}
}));
return null;
});
}
Assert.assertEquals(2L, count.longValue());
}
use of com.palantir.atlasdb.keyvalue.api.Namespace in project atlasdb by palantir.
the class TableFactoryRenderer method getFields.
private List<FieldSpec> getFields() {
List<FieldSpec> results = new ArrayList<>();
TypeName functionOfTransactionAndTriggersType = ParameterizedTypeName.get(ClassName.get(Function.class), WildcardTypeName.supertypeOf(Transaction.class), sharedTriggersType);
results.add(getDefaultNamespaceField());
results.add(FieldSpec.builder(ParameterizedTypeName.get(ClassName.get(List.class), functionOfTransactionAndTriggersType), "sharedTriggers").addModifiers(Modifier.PRIVATE, Modifier.FINAL).build());
results.add(FieldSpec.builder(Namespace.class, "namespace").addModifiers(Modifier.PRIVATE, Modifier.FINAL).build());
return results;
}
use of com.palantir.atlasdb.keyvalue.api.Namespace in project atlasdb by palantir.
the class TableFactoryRenderer method getConstructors.
private List<MethodSpec> getConstructors() {
List<MethodSpec> results = new ArrayList<>();
TypeName functionOfTransactionAndTriggersType = ParameterizedTypeName.get(ClassName.get(Function.class), WildcardTypeName.supertypeOf(Transaction.class), sharedTriggersType);
TypeName sharedTriggersListType = ParameterizedTypeName.get(ClassName.get(List.class), functionOfTransactionAndTriggersType);
results.add(factoryBaseBuilder().addParameter(ParameterizedTypeName.get(ClassName.get(List.class), functionOfTransactionAndTriggersType), "sharedTriggers").addParameter(Namespace.class, "namespace").addStatement("return new $T($L, $L)", tableFactoryType, "sharedTriggers", "namespace").build());
results.add(factoryBaseBuilder().addParameter(sharedTriggersListType, "sharedTriggers").addStatement("return new $T($L, $L)", tableFactoryType, "sharedTriggers", "defaultNamespace").build());
results.add(factoryBaseBuilder().addParameter(Namespace.class, "namespace").addStatement("return of($T.<$T>of(), $L)", ImmutableList.class, functionOfTransactionAndTriggersType, "namespace").build());
results.add(factoryBaseBuilder().addStatement("return of($T.<$T>of(), $L)", ImmutableList.class, functionOfTransactionAndTriggersType, "defaultNamespace").build());
results.add(MethodSpec.constructorBuilder().addModifiers(Modifier.PRIVATE).addParameter(sharedTriggersListType, "sharedTriggers").addParameter(Namespace.class, "namespace").addStatement("this.$L = $L", "sharedTriggers", "sharedTriggers").addStatement("this.$L = $L", "namespace", "namespace").build());
return results;
}
Aggregations