private IndexDescriptor awaitOnline(IndexDescriptor index) throws KernelException {
long start = System.currentTimeMillis();
long end = start + 20_000;
while (System.currentTimeMillis() < end) {
try (Transaction tx = db.beginTx()) {
Statement statement = bridge.instance();
switch (statement.readOperations().indexGetState(index)) {
case ONLINE:
return index;
case FAILED:
throw new IllegalStateException("Index failed instead of becoming ONLINE");
default:
break;
}
tx.success();
try {
Thread.sleep(100);
} catch (InterruptedException e) {
// ignored
}
}
}
throw new IllegalStateException("Index did not become ONLINE within reasonable time");
}
private void deleteNode(long nodeId) throws KernelException {
try (Transaction tx = db.beginTx()) {
Statement statement = bridge.instance();
statement.dataWriteOperations().nodeDelete(nodeId);
tx.success();
}
}
@Test
public void deletingNodeWithLabelsShouldHaveRemovalReflectedInLabelScans() throws Exception {
// GIVEN
Transaction tx = db.beginTx();
Label label = label("labello");
Node node = db.createNode(label);
tx.success();
tx.finish();
// AND GIVEN I DELETE IT
tx = db.beginTx();
node.delete();
tx.success();
tx.finish();
// WHEN
tx = db.beginTx();
Statement statement = statementContextProvider.instance();
int labelId = statement.readOperations().labelGetForName(label.name());
PrimitiveLongIterator nodes = statement.readOperations().nodesGetForLabel(labelId);
Set<Long> nodeSet = asSet(nodes);
tx.success();
tx.finish();
// THEN
assertThat(nodeSet, equalTo(Collections.<Long>emptySet()));
}
private void dropIndex(IndexDescriptor index) throws KernelException {
try (Transaction tx = db.beginTx()) {
Statement statement = bridge.instance();
statement.schemaWriteOperations().indexDrop(index);
tx.success();
}
}
/**
* While we transition ownership from the Beans API to the Kernel API for core database
* interactions, there will be a bit of a mess. Our first goal is an architecture like this:
*
* <p>Users / \ Beans API Cypher \ / Kernel API | Kernel Implementation
*
* <p>But our current intermediate architecture looks like this:
*
* <p>Users / \ Beans API <--- Cypher | \ / | Kernel API | | Kernel Implementation
*
* <p>Meaning Kernel API and Beans API both manipulate the underlying kernel, causing lots of
* corner cases. Most notably, those corner cases are related to Transactions, and the interplay
* between three transaction APIs: - The Beans API - The JTA Transaction Manager API - The Kernel
* TransactionContext API
*
* <p>In the long term, the goal is for JTA compliant stuff to live outside of the kernel, as an
* addon. The Kernel API will rule supreme over the land of transactions. We are a long way away
* from there, however, so as a first intermediary step, the JTA transaction manager rules
* supreme, and the Kernel API piggybacks on it.
*
* <p>This test shows us how to use both the Kernel API and the Beans API together in the same
* transaction, during the transition phase.
*/
@Test
public void mixingBeansApiWithKernelAPI() throws Exception {
// 1: Start your transactions through the Beans API
Transaction beansAPITx = db.beginTx();
// 2: Get a hold of a KernelAPI statement context for the *current* transaction this way:
Statement statement = statementContextProvider.instance();
// 3: Now you can interact through both the statement context and the kernel API to manipulate
// the
// same transaction.
Node node = db.createNode();
int labelId = statement.dataWriteOperations().labelGetOrCreateForName("labello");
statement.dataWriteOperations().nodeAddLabel(node.getId(), labelId);
// 4: Close the StatementContext
statement.close();
// 5: Commit through the beans API
beansAPITx.success();
beansAPITx.finish();
// NOTE: Transactions are still thread-bound right now, because we use JTA to "own"
// transactions,
// meaning if you use
// both the Kernel API to create transactions while a Beans API transaction is running in the
// same
// thread, the results are undefined.
// When the Kernel API implementation is done, the Kernel API transaction implementation is not
// meant
// to be bound to threads.
}
@Test
public void shouldNotBeAbleToCommitIfFailedTransactionContext() throws Exception {
Transaction tx = db.beginTx();
Statement statement = statementContextProvider.instance();
// WHEN
Node node = null;
int labelId = -1;
try {
node = db.createNode();
labelId = statement.dataWriteOperations().labelGetOrCreateForName("labello");
statement.dataWriteOperations().nodeAddLabel(node.getId(), labelId);
statement.close();
tx.failure();
tx.success();
tx.finish();
fail("Should have failed");
} catch (TransactionFailureException e) { // Expected
}
// THEN
tx = db.beginTx();
statement = statementContextProvider.instance();
try {
statement.readOperations().nodeHasLabel(node.getId(), labelId);
fail("should have thrown exception");
} catch (EntityNotFoundException e) {
// Yay!
}
tx.finish();
}
private double indexSelectivity(IndexDescriptor descriptor) throws KernelException {
try (Transaction tx = db.beginTx()) {
Statement statement = bridge.instance();
double selectivity = statement.readOperations().indexUniqueValuesSelectivity(descriptor);
tx.success();
return selectivity;
}
}
private IndexDescriptor createAnIndex(
HighlyAvailableGraphDatabase db, Label label, String propertyName) throws KernelException {
try (Transaction tx = db.beginTx()) {
Statement statement = statement(db);
int labelId = statement.tokenWriteOperations().labelGetOrCreateForName(label.name());
int propertyKeyId =
statement.tokenWriteOperations().propertyKeyGetOrCreateForName(propertyName);
IndexDescriptor index = statement.schemaWriteOperations().indexCreate(labelId, propertyKeyId);
tx.success();
return index;
}
}
private IndexDescriptor createIndex(String labelName, String propertyKeyName)
throws KernelException {
try (Transaction tx = db.beginTx()) {
Statement statement = bridge.instance();
int labelId = statement.tokenWriteOperations().labelGetOrCreateForName(labelName);
int propertyKeyId =
statement.tokenWriteOperations().propertyKeyGetOrCreateForName(propertyKeyName);
IndexDescriptor index = statement.schemaWriteOperations().indexCreate(labelId, propertyKeyId);
tx.success();
return index;
}
}
private void changeName(long nodeId, String propertyKeyName, Object newValue)
throws KernelException {
try (Transaction tx = db.beginTx()) {
Statement statement = bridge.instance();
int propertyKeyId =
statement.tokenWriteOperations().propertyKeyGetOrCreateForName(propertyKeyName);
statement
.dataWriteOperations()
.nodeSetProperty(nodeId, Property.property(propertyKeyId, newValue));
tx.success();
}
}
private void assertOnNodeCounts(
int expectedTotalNodes,
int expectedLabelledNodes,
Label label,
HighlyAvailableGraphDatabase db) {
try (Transaction ignored = db.beginTx()) {
final Statement statement = statement(db);
final int labelId = statement.readOperations().labelGetForName(label.name());
assertEquals(expectedTotalNodes, statement.readOperations().countsForNode(-1));
assertEquals(expectedLabelledNodes, statement.readOperations().countsForNode(labelId));
}
}
@Test
public void mixingBeansApiWithKernelAPIForNestedTransaction() throws Exception {
// GIVEN
Transaction outerTx = db.beginTx();
Statement statement = statementContextProvider.instance();
// WHEN
Node node = db.createNode();
int labelId = statement.dataWriteOperations().labelGetOrCreateForName("labello");
statement.dataWriteOperations().nodeAddLabel(node.getId(), labelId);
statement.close();
outerTx.finish();
}
private long createNode(
Statement statement, String labelName, String propertyKeyName, Object value)
throws KernelException {
int labelId = statement.tokenWriteOperations().labelGetOrCreateForName(labelName);
int propertyKeyId =
statement.tokenWriteOperations().propertyKeyGetOrCreateForName(propertyKeyName);
long nodeId = statement.dataWriteOperations().nodeCreate();
statement.dataWriteOperations().nodeAddLabel(nodeId, labelId);
statement
.dataWriteOperations()
.nodeSetProperty(nodeId, Property.property(propertyKeyId, value));
return nodeId;
}
@Test
public void labelShouldBeRemovedAfterCommit() throws Exception {
// GIVEN
Transaction tx = db.beginTx();
Statement statement = statementContextProvider.instance();
Node node = db.createNode();
int labelId1 = statement.dataWriteOperations().labelGetOrCreateForName("labello1");
statement.dataWriteOperations().nodeAddLabel(node.getId(), labelId1);
statement.close();
tx.success();
tx.finish();
// WHEN
tx = db.beginTx();
statement = statementContextProvider.instance();
statement.dataWriteOperations().nodeRemoveLabel(node.getId(), labelId1);
statement.close();
tx.success();
tx.finish();
// THEN
tx = db.beginTx();
statement = statementContextProvider.instance();
PrimitiveIntIterator labels = statement.readOperations().nodeGetLabels(node.getId());
statement.close();
tx.success();
tx.finish();
assertThat(asSet(labels), equalTo(Collections.<Integer>emptySet()));
}
@Test
public void transactionStateShouldReflectRemovingLabelImmediately() throws Exception {
// GIVEN
Transaction tx = db.beginTx();
Statement statement = statementContextProvider.instance();
Node node = db.createNode();
int labelId1 = statement.dataWriteOperations().labelGetOrCreateForName("labello1");
int labelId2 = statement.dataWriteOperations().labelGetOrCreateForName("labello2");
statement.dataWriteOperations().nodeAddLabel(node.getId(), labelId1);
statement.dataWriteOperations().nodeAddLabel(node.getId(), labelId2);
statement.close();
tx.success();
tx.finish();
tx = db.beginTx();
statement = statementContextProvider.instance();
// WHEN
statement.dataWriteOperations().nodeRemoveLabel(node.getId(), labelId2);
// THEN
PrimitiveIntIterator labelsIterator = statement.readOperations().nodeGetLabels(node.getId());
Set<Integer> labels = asSet(labelsIterator);
assertFalse(statement.readOperations().nodeHasLabel(node.getId(), labelId2));
assertEquals(asSet(labelId1), labels);
statement.close();
tx.success();
tx.finish();
}
@Test
public void deletingNodeWithLabelsShouldHaveThoseLabelRemovalsReflectedInTransaction()
throws Exception {
// GIVEN
Transaction tx = db.beginTx();
Label label = label("labello");
Node node = db.createNode(label);
tx.success();
tx.finish();
tx = db.beginTx();
Statement statement = statementContextProvider.instance();
// WHEN
statement.dataWriteOperations().nodeDelete(node.getId());
// Then
int labelId = statement.readOperations().labelGetForName(label.name());
Set<Integer> labels = asSet(statement.readOperations().nodeGetLabels(node.getId()));
boolean labelIsSet = statement.readOperations().nodeHasLabel(node.getId(), labelId);
Set<Long> nodes = asSet(statement.readOperations().nodesGetForLabel(labelId));
statement.close();
tx.success();
tx.finish();
assertEquals(emptySetOf(Long.class), nodes);
assertEquals(emptySetOf(Integer.class), labels);
assertFalse("Label should not be set on node here", labelIsSet);
}
private String getOrCreateSchemaState(String key, final String maybeSetThisState) {
Transaction tx = db.beginTx();
Statement statement = statementContextProvider.instance();
String state =
statement
.readOperations()
.schemaStateGetOrCreate(
key,
new Function<String, String>() {
@Override
public String apply(String s) {
return maybeSetThisState;
}
});
tx.success();
tx.finish();
return state;
}
private boolean schemaStateContains(String key) {
Transaction tx = db.beginTx();
Statement statement = statementContextProvider.instance();
final AtomicBoolean result = new AtomicBoolean(true);
statement
.readOperations()
.schemaStateGetOrCreate(
key,
new Function<String, Object>() {
@Override
public Object apply(String s) {
result.set(false);
return null;
}
});
tx.success();
tx.finish();
return result.get();
}
@Test
public void removingNonExistentLabelFromNodeShouldRespondFalse() throws Exception {
// GIVEN
Transaction tx = db.beginTx();
Node node = db.createNode();
Statement statement = statementContextProvider.instance();
int labelId = statement.dataWriteOperations().labelGetOrCreateForName("mylabel");
statement.close();
tx.success();
tx.finish();
// WHEN
tx = db.beginTx();
statement = statementContextProvider.instance();
boolean removed = statement.dataWriteOperations().nodeRemoveLabel(node.getId(), labelId);
// THEN
assertFalse("Shouldn't have been removed now", removed);
tx.finish();
}
@Test
public void addingExistingLabelToNodeShouldRespondFalse() throws Exception {
// GIVEN
Transaction tx = db.beginTx();
Node node = db.createNode();
Statement statement = statementContextProvider.instance();
int labelId = statement.dataWriteOperations().labelGetOrCreateForName("mylabel");
statement.close();
tx.success();
tx.finish();
// WHEN
tx = db.beginTx();
statement = statementContextProvider.instance();
boolean added = statement.dataWriteOperations().nodeAddLabel(node.getId(), labelId);
// THEN
assertTrue("Should have been added now", added);
tx.finish();
}
private void showStructure(Statement statement, DbStructureVisitor visitor) {
ReadOperations read = statement.readOperations();
try {
showTokens(visitor, read);
showSchema(visitor, read);
showStatistics(visitor, read);
} catch (KernelException e) {
throw new IllegalStateException(
"Kernel exception when traversing database schema structure and statistics. This is not expected to happen.",
e);
}
}
@Test
public void changesInTransactionContextShouldBeRolledBackWhenTxIsRolledBack() throws Exception {
// GIVEN
Transaction tx = db.beginTx();
Statement statement = statementContextProvider.instance();
// WHEN
Node node = db.createNode();
int labelId = statement.dataWriteOperations().labelGetOrCreateForName("labello");
statement.dataWriteOperations().nodeAddLabel(node.getId(), labelId);
statement.close();
tx.finish();
// THEN
tx = db.beginTx();
statement = statementContextProvider.instance();
try {
statement.readOperations().nodeHasLabel(node.getId(), labelId);
fail("should have thrown exception");
} catch (EntityNotFoundException e) {
// Yay!
}
tx.finish();
}
@Test
public void transactionStateShouldRemovePreviouslyAddedLabel() throws Exception {
Transaction tx = db.beginTx();
Statement statement = statementContextProvider.instance();
// WHEN
Node node = db.createNode();
int labelId1 = statement.dataWriteOperations().labelGetOrCreateForName("labello1");
int labelId2 = statement.dataWriteOperations().labelGetOrCreateForName("labello2");
statement.dataWriteOperations().nodeAddLabel(node.getId(), labelId1);
statement.dataWriteOperations().nodeAddLabel(node.getId(), labelId2);
statement.dataWriteOperations().nodeRemoveLabel(node.getId(), labelId2);
statement.close();
tx.success();
tx.finish();
// THEN
tx = db.beginTx();
statement = statementContextProvider.instance();
assertEquals(asSet(labelId1), asSet(statement.readOperations().nodeGetLabels(node.getId())));
tx.finish();
}