Examples with RowModel org.neo4j.ogm.model.RowModel used on opensource projects

Example 6 with RowModel

use of org.neo4j.ogm.model.RowModel in project neo4j-ogm by neo4j.

the class RequestExecutor method executeStatements.

private void executeStatements(CompileContext context, List<ReferenceMapping> entityReferenceMappings, List<ReferenceMapping> relReferenceMappings, List<Statement> statements) {
    if (statements.size() > 0) {
        List<Statement> noCheckStatements = new ArrayList<>();
        for (Statement statement : statements) {
            if (statement.optimisticLockingConfig().isPresent()) {
                DefaultRequest request = new DefaultRequest(statement);
                try (Response<RowModel> response = session.requestHandler().execute(request)) {
                    List<RowModel> rowModels = response.toList();
                    session.optimisticLockingChecker().checkResultsCount(rowModels, statement);
                    registerEntityIds(context, rowModels, entityReferenceMappings, relReferenceMappings);
                }
            } else {
                noCheckStatements.add(statement);
            }
        }
        DefaultRequest defaultRequest = new DefaultRequest();
        defaultRequest.setStatements(noCheckStatements);
        try (Response<RowModel> response = session.requestHandler().execute(defaultRequest)) {
            registerEntityIds(context, response.toList(), entityReferenceMappings, relReferenceMappings);
        }
    }
}

Example 7 with RowModel

use of org.neo4j.ogm.model.RowModel in project neo4j-ogm by neo4j.

the class EmbeddedRequest method execute.

@Override
public Response<RowModel> execute(DefaultRequest query) {
    // TODO this is a hack to get the embedded driver to work with executing multiple statements
    final List<RowModel> rowmodels = new ArrayList<>();
    String[] columns = null;
    for (Statement statement : query.getStatements()) {
        Result result = executeRequest(statement);
        if (columns == null) {
            columns = result.columns().toArray(new String[result.columns().size()]);
        }
        RowModelResponse rowModelResponse = new RowModelResponse(result, entityAdapter);
        RowModel model;
        while ((model = rowModelResponse.next()) != null) {
            rowmodels.add(model);
        }
        result.close();
    }
    final String[] finalColumns = columns;
    return new Response<>() {

        int currentRow = 0;

        @Override
        public RowModel next() {
            if (currentRow < rowmodels.size()) {
                return rowmodels.get(currentRow++);
            }
            return null;
        }

        @Override
        public void close() {
        }

        @Override
        public String[] columns() {
            return finalColumns;
        }
    };
}

Example 8 with RowModel

use of org.neo4j.ogm.model.RowModel in project neo4j-ogm by neo4j.

the class OptimisticLockingChecker method checkResultsCount.

public void checkResultsCount(List<RowModel> models, Statement request) {
    OptimisticLockingConfig olConfig = request.optimisticLockingConfig().orElseThrow(() -> new IllegalArgumentException("Statement " + request + " doesn't require results count check"));
    StringBuilder message = new StringBuilder();
    if (olConfig.getExpectedResultsCount() != models.size()) {
        message.append("Optimistic locking exception failed. ");
        Map<String, Object> parameters = request.getParameters();
        Object type = parameters.get("type");
        List<Map<String, Object>> rows = (List<Map<String, Object>>) parameters.get("rows");
        if (rows != null) {
            Map<Long, Long> nodeIds = new HashMap<>();
            Map<Long, Long> relIds = new HashMap<>();
            for (Map<String, Object> row : rows) {
                if (type.equals("node")) {
                    nodeIds.put((Long) row.get("nodeId"), (Long) row.get(olConfig.getVersionProperty()));
                } else if (type.equals("rel")) {
                    relIds.put((Long) row.get("relId"), (Long) row.get(olConfig.getVersionProperty()));
                }
            }
            if (!models.isEmpty()) {
                int idPosition = indexOf(models.get(0).variables(), "id");
                for (RowModel model : models) {
                    Object id = model.getValues()[idPosition];
                    if (type.equals("node")) {
                        nodeIds.remove(id);
                    } else if (type.equals("rel")) {
                        relIds.remove(id);
                    }
                }
            }
            for (Map.Entry<Long, Long> node : nodeIds.entrySet()) {
                Long nodeId = node.getKey();
                session.context().detachNodeEntity(nodeId);
                message.append("Entity with type='").append(Arrays.toString(olConfig.getTypes())).append("' and id='").append(nodeId).append("' had incorrect version ").append(node.getValue());
            }
            for (Map.Entry<Long, Long> rel : relIds.entrySet()) {
                Long relId = rel.getKey();
                session.context().detachRelationshipEntity(relId);
                message.append("Relationship entity with id ").append(relId).append(" had incorrect version ").append(rel.getValue());
            }
        } else {
            Object id = parameters.get("id");
            if (id != null && models.isEmpty()) {
                if (type.equals("node")) {
                    session.context().detachNodeEntity((Long) id);
                } else if (type.equals("rel")) {
                    session.context().detachRelationshipEntity((Long) id);
                }
            }
        }
        throw new OptimisticLockingException(message.toString());
    }
}

Example 9 with RowModel

use of org.neo4j.ogm.model.RowModel in project neo4j-ogm by neo4j.

the class DeleteDelegate method deleteAll.

/**
 * Deletes all nodes of a given type. They will get discovered by using the matching label for that type.
 * To avoid a delete of every node in the database the method will abort the delete operation if no label
 * can be determined.
 *
 * @param type  The type of the nodes/objects to be deleted.
 * @param <T>   The type to work with
 */
public <T> void deleteAll(Class<T> type) {
    ClassInfo classInfo = session.metaData().classInfo(type.getName());
    if (classInfo != null) {
        String entityLabel = classInfo.neo4jName();
        if (entityLabel == null) {
            session.warn("Unable to find database label for entity " + type.getName() + " : no results will be returned. Make sure the class is registered, " + "and not abstract without @NodeEntity annotation");
            return;
        }
        Statement request = getDeleteStatementsBasedOnType(type).delete(entityLabel);
        RowModelRequest query = new DefaultRowModelRequest(request.getStatement(), request.getParameters());
        session.notifyListeners(new PersistenceEvent(type, Event.TYPE.PRE_DELETE));
        session.doInTransaction(() -> {
            try (Response<RowModel> response = session.requestHandler().execute(query)) {
                session.context().removeType(type);
                if (session.eventsEnabled()) {
                    session.notifyListeners(new PersistenceEvent(type, Event.TYPE.POST_DELETE));
                }
            }
        }, Transaction.Type.READ_WRITE);
    } else {
        session.warn(type.getName() + " is not a persistable class");
    }
}

Example 10 with RowModel

use of org.neo4j.ogm.model.RowModel in project neo4j-ogm by neo4j.

the class DeleteDelegate method deleteOneOrMoreObjects.

// TODO : this is being done in multiple requests at the moment, one per object. Why not put them in a single request?
private void deleteOneOrMoreObjects(List<?> objects, Set<Object> neighbours) {
    Set<Object> notified = new HashSet<>();
    if (session.eventsEnabled()) {
        for (Object affectedObject : neighbours) {
            if (!notified.contains(affectedObject)) {
                session.notifyListeners(new PersistenceEvent(affectedObject, Event.TYPE.PRE_SAVE));
                notified.add(affectedObject);
            }
        }
    }
    for (Object object : objects) {
        MetaData metaData = session.metaData();
        ClassInfo classInfo = metaData.classInfo(object);
        if (classInfo == null) {
            session.warn(object.getClass().getName() + " is not an instance of a persistable class");
        } else {
            MappingContext mappingContext = session.context();
            Long id = mappingContext.optionalNativeId(object).filter(possibleId -> possibleId >= 0).orElseGet(() -> {
                session.warn(String.format("Instance of class %s has to be reloaded to be deleted. This can happen if the session has " + "been cleared between loading and deleting or using an object from a different transaction.", object.getClass()));
                return classInfo.getPrimaryIndexOrIdReader().apply(object).map(primaryIndexOrId -> session.load(object.getClass(), (Serializable) primaryIndexOrId)).flatMap(reloadedObject -> mappingContext.optionalNativeId(reloadedObject)).orElse(-1L);
            });
            if (id >= 0) {
                Statement request = getDeleteStatement(object, id, classInfo);
                if (session.eventsEnabled() && !notified.contains(object)) {
                    session.notifyListeners(new PersistenceEvent(object, Event.TYPE.PRE_DELETE));
                    notified.add(object);
                }
                RowModelRequest query = new DefaultRowModelRequest(request.getStatement(), request.getParameters());
                session.doInTransaction(() -> {
                    try (Response<RowModel> response = session.requestHandler().execute(query)) {
                        if (request.optimisticLockingConfig().isPresent()) {
                            List<RowModel> rowModels = response.toList();
                            session.optimisticLockingChecker().checkResultsCount(rowModels, request);
                        }
                        if (metaData.isRelationshipEntity(classInfo.name())) {
                            session.detachRelationshipEntity(id);
                        } else {
                            session.detachNodeEntity(id);
                        }
                        // enabled in the first place.
                        if (notified.contains(object)) {
                            session.notifyListeners(new PersistenceEvent(object, Event.TYPE.POST_DELETE));
                        }
                    }
                }, Transaction.Type.READ_WRITE);
            }
        }
    }
    if (session.eventsEnabled()) {
        for (Object affectedObject : neighbours) {
            if (notified.contains(affectedObject)) {
                session.notifyListeners(new PersistenceEvent(affectedObject, Event.TYPE.POST_SAVE));
            }
        }
    }
}