Examples with Statement org.neo4j.ogm.request.Statement used on opensource projects

Example 36 with Statement

use of org.neo4j.ogm.request.Statement in project neo4j-ogm by neo4j.

the class CompilerTest method addObjectToExistingCollection.

@Test
public void addObjectToExistingCollection() {
    // create
    Long wallerId = 0L;
    Long maryId = 1L;
    School waller = new School("Waller");
    waller.setId(wallerId);
    Teacher mary = new Teacher("Mary");
    mary.setId(maryId);
    // relate
    mary.setSchool(waller);
    // validate the domain model
    assertThat(mary.getSchool()).isEqualTo(waller);
    assertThat(waller.getTeachers()).contains(mary);
    assertThat(waller.getTeachers()).hasSize(1);
    // set the mapping context accordingly
    mappingContext.addNodeEntity(mary);
    mappingContext.addNodeEntity(waller);
    mappingContext.addRelationship(new MappedRelationship(maryId, "SCHOOL", wallerId, null, Teacher.class, School.class));
    mappingContext.addRelationship(new MappedRelationship(wallerId, "TEACHERS", maryId, null, School.class, Teacher.class));
    Teacher jim = new Teacher("Jim");
    jim.setSchool(waller);
    assertThat(waller.getTeachers()).contains(jim);
    assertThat(waller.getTeachers()).hasSize(2);
    assertThat(jim.getSchool()).isEqualTo(waller);
    // Save jim
    Compiler compiler = mapAndCompile(jim, -1);
    List<Statement> createNodesStatements = compiler.createNodesStatements();
    assertThat(createNodesStatements).extracting(Statement::getStatement).containsOnly("UNWIND $rows as row CREATE (n:`Teacher`) SET n=row.props RETURN row.nodeRef as ref, ID(n) as id, $type as type");
    assertThat(createNodesStatements).extracting(Statement::getParameters);
    for (Statement statement : createNodesStatements) {
        List rows = (List) statement.getParameters().get("rows");
        assertThat(rows).hasSize(1);
    }
    List<Statement> createRelsStatements = compiler.createRelationshipsStatements();
    assertThat(createRelsStatements).extracting(Statement::getStatement).containsOnly("UNWIND $rows as row MATCH (startNode) WHERE ID(startNode) = row.startNodeId WITH row,startNode MATCH (endNode) WHERE ID(endNode) = row.endNodeId MERGE (startNode)-[rel:`SCHOOL`]->(endNode) RETURN row.relRef as ref, ID(rel) as id, $type as type", "UNWIND $rows as row MATCH (startNode) WHERE ID(startNode) = row.startNodeId WITH row,startNode MATCH (endNode) WHERE ID(endNode) = row.endNodeId MERGE (startNode)-[rel:`TEACHERS`]->(endNode) RETURN row.relRef as ref, ID(rel) as id, $type as type");
    // Save waller
    compiler = mapAndCompile(waller, -1);
    createNodesStatements = compiler.createNodesStatements();
    assertThat(createNodesStatements).extracting(Statement::getStatement).containsOnly("UNWIND $rows as row CREATE (n:`Teacher`) SET n=row.props RETURN row.nodeRef as ref, ID(n) as id, $type as type");
    for (Statement statement : createNodesStatements) {
        List rows = (List) statement.getParameters().get("rows");
        assertThat(rows).hasSize(1);
    }
    createRelsStatements = compiler.createRelationshipsStatements();
    assertThat(createRelsStatements).extracting(Statement::getStatement).containsOnly("UNWIND $rows as row MATCH (startNode) WHERE ID(startNode) = row.startNodeId WITH row,startNode MATCH (endNode) WHERE ID(endNode) = row.endNodeId MERGE (startNode)-[rel:`SCHOOL`]->(endNode) RETURN row.relRef as ref, ID(rel) as id, $type as type", "UNWIND $rows as row MATCH (startNode) WHERE ID(startNode) = row.startNodeId WITH row,startNode MATCH (endNode) WHERE ID(endNode) = row.endNodeId MERGE (startNode)-[rel:`TEACHERS`]->(endNode) RETURN row.relRef as ref, ID(rel) as id, $type as type");
    // Save mary
    compiler = mapAndCompile(mary, -1);
    createNodesStatements = compiler.createNodesStatements();
    assertThat(createNodesStatements).extracting(Statement::getStatement).containsOnly("UNWIND $rows as row CREATE (n:`Teacher`) SET n=row.props RETURN row.nodeRef as ref, ID(n) as id, $type as type");
    for (Statement statement : createNodesStatements) {
        List rows = (List) statement.getParameters().get("rows");
        assertThat(rows).hasSize(1);
    }
    createRelsStatements = compiler.createRelationshipsStatements();
    assertThat(createRelsStatements).extracting(Statement::getStatement).containsOnly("UNWIND $rows as row MATCH (startNode) WHERE ID(startNode) = row.startNodeId WITH row,startNode MATCH (endNode) WHERE ID(endNode) = row.endNodeId MERGE (startNode)-[rel:`SCHOOL`]->(endNode) RETURN row.relRef as ref, ID(rel) as id, $type as type", "UNWIND $rows as row MATCH (startNode) WHERE ID(startNode) = row.startNodeId WITH row,startNode MATCH (endNode) WHERE ID(endNode) = row.endNodeId MERGE (startNode)-[rel:`TEACHERS`]->(endNode) RETURN row.relRef as ref, ID(rel) as id, $type as type");
}

Example 37 with Statement

use of org.neo4j.ogm.request.Statement in project neo4j-ogm by neo4j.

the class CompilerTest method shouldCorrectlyRemoveRelationshipWhenItemIsRemovedFromCollection.

@Test
public void shouldCorrectlyRemoveRelationshipWhenItemIsRemovedFromCollection() {
    // simple music course with three students
    Long mid = 0L;
    Long xid = 1L;
    Long yid = 2L;
    Long zid = 3L;
    Course music = new Course("GCSE Music");
    music.setId(mid);
    Student xavier = new Student("xavier");
    xavier.setId(xid);
    Student yvonne = new Student("Yvonne");
    yvonne.setId(yid);
    Student zack = new Student("Zack");
    zack.setId(zid);
    music.setStudents(Arrays.asList(yvonne, xavier, zack));
    mappingContext.addRelationship(new MappedRelationship(mid, "STUDENTS", xid, null, Course.class, Student.class));
    mappingContext.addRelationship(new MappedRelationship(mid, "STUDENTS", yid, null, Course.class, Student.class));
    mappingContext.addRelationship(new MappedRelationship(mid, "STUDENTS", zid, null, Course.class, Student.class));
    mappingContext.addNodeEntity(xavier);
    mappingContext.addNodeEntity(yvonne);
    mappingContext.addNodeEntity(zack);
    mappingContext.addNodeEntity(music);
    // now, update the domain model, setting yvonne as the only music student (i.e remove zack and xavier)
    music.setStudents(Arrays.asList(yvonne));
    // Save music
    Compiler compiler = mapAndCompile(music, -1);
    assertThat(compiler.createNodesStatements()).isEmpty();
    assertThat(compiler.createRelationshipsStatements()).isEmpty();
    List<Statement> deleteRelsStatement = compiler.deleteRelationshipStatements();
    assertThat(deleteRelsStatement).extracting(Statement::getStatement).containsOnly("UNWIND $rows as row MATCH (startNode) WHERE ID(startNode) = row.startNodeId WITH row,startNode MATCH (endNode) WHERE ID(endNode) = row.endNodeId MATCH (startNode)-[rel:`STUDENTS`]->(endNode) DELETE rel");
    assertThat(((List) deleteRelsStatement.get(0).getParameters().get("rows"))).hasSize(2);
}

Example 38 with Statement

use of org.neo4j.ogm.request.Statement in project neo4j-ogm by neo4j.

the class HttpRequest method cypherRequest.

// we use the OBJECT_MAPPER to create the request string from the statement.
// this driver is the only one that needs to do this, because the request format
// is different for each type of request - GraphModelRequest/RowModelRequest, etc
private String cypherRequest(Statement statement) {
    List<Statement> statementList = new ArrayList<>();
    statementList.add(statement);
    try {
        return OBJECT_MAPPER.writeValueAsString(new Statements(statementList));
    } catch (JsonProcessingException jpe) {
        throw new ResultProcessingException("Could not create JSON due to " + jpe.getLocalizedMessage(), jpe);
    }
}