Examples with FilteringIterator org.neo4j.helpers.collection.FilteringIterator used on opensource projects

Example 1 with FilteringIterator

use of org.neo4j.helpers.collection.FilteringIterator in project neo4j by neo4j.

the class IndexCreationTest method verifyThatIndexCreationTransactionIsTheFirstOne.

private void verifyThatIndexCreationTransactionIsTheFirstOne() throws Exception {
    PhysicalLogFile pLogFile = db.getDependencyResolver().resolveDependency(PhysicalLogFile.class);
    long version = db.getDependencyResolver().resolveDependency(LogVersionRepository.class).getCurrentLogVersion();
    db.getDependencyResolver().resolveDependency(LogRotation.class).rotateLogFile();
    db.getDependencyResolver().resolveDependency(CheckPointer.class).forceCheckPoint(new SimpleTriggerInfo("test"));
    ReadableLogChannel logChannel = pLogFile.getReader(LogPosition.start(version));
    final AtomicBoolean success = new AtomicBoolean(false);
    try (IOCursor<LogEntry> cursor = new LogEntryCursor(new VersionAwareLogEntryReader<>(), logChannel)) {
        List<StorageCommand> commandsInFirstEntry = new ArrayList<>();
        boolean startFound = false;
        while (cursor.next()) {
            LogEntry entry = cursor.get();
            if (entry instanceof LogEntryStart) {
                if (startFound) {
                    throw new IllegalArgumentException("More than one start entry");
                }
                startFound = true;
            }
            if (startFound && entry instanceof LogEntryCommand) {
                commandsInFirstEntry.add(entry.<LogEntryCommand>as().getXaCommand());
            }
            if (entry instanceof LogEntryCommit) {
                // The first COMMIT
                assertTrue(startFound);
                assertFalse("Index creation transaction wasn't the first one", commandsInFirstEntry.isEmpty());
                List<StorageCommand> createCommands = Iterators.asList(new FilteringIterator<>(commandsInFirstEntry.iterator(), item -> item instanceof IndexDefineCommand));
                assertEquals(1, createCommands.size());
                success.set(true);
                break;
            }
        }
    }
    assertTrue("Didn't find any commit record in log " + version, success.get());
}

Example 2 with FilteringIterator

use of org.neo4j.helpers.collection.FilteringIterator in project neo4j by neo4j.

the class Ls method wrapInLimitingIterator.

private Iterator<Relationship> wrapInLimitingIterator(AppCommandParser parser, Iterator<Relationship> iterator, Map<String, Object> filterMap, boolean caseInsensitiveFilters, boolean looseFilters) throws ShellException {
    final AtomicBoolean handBreak = new AtomicBoolean();
    int maxRelsPerType = parser.optionAsNumber("m", DEFAULT_MAX_RELS_PER_TYPE_LIMIT).intValue();
    Map<String, Direction> types = filterMapToTypes(getServer().getDb(), Direction.BOTH, filterMap, caseInsensitiveFilters, looseFilters);
    return new FilteringIterator<Relationship>(iterator, new LimitPerTypeFilter(maxRelsPerType, types, handBreak)) {

        @Override
        protected Relationship fetchNextOrNull() {
            return handBreak.get() ? null : super.fetchNextOrNull();
        }
    };
}