public ReusableRecordsAllocator( int recordSize, DynamicRecord... records ) { this.recordSize = recordSize; this.recordIterator = Iterators.iterator( records ); }
@Override public Iterator<RelationshipTypeCount> iterator() { return Iterators.iterator( typeCounts ); }
@SafeVarargs public static <T> Iterable<T> asIterable( final T... array ) { return () -> Iterators.iterator( array ); }
public static <T> Iterable<T> option( final T item ) { if ( item == null ) { return Collections.emptyList(); } return () -> Iterators.iterator( item ); }
/** * @return store files, namely the single "neostore.labelscanstore.db" store file. */ @Override public ResourceIterator<File> snapshotStoreFiles() { return asResourceIterator( iterator( storeFile ) ); }
@Override public ResourceIterator<File> snapshotFiles() { return asResourceIterator( iterator( storeFile ) ); }
private Iterable<CommonAbstractStore> instantiatedRecordStores() { Iterator<StoreType> storeTypes = new FilteringIterator<>( iterator( STORE_TYPES ), INSTANTIATED_RECORD_STORES ); return loop( new IteratorWrapper<CommonAbstractStore,StoreType>( storeTypes ) { @Override protected CommonAbstractStore underlyingObjectToObject( StoreType type ) { return (CommonAbstractStore) stores[type.ordinal()]; } } ); }
private SchemaRead schemaWithIndexes( IndexReference... indexes ) { final SchemaRead schemaRead = mock( SchemaRead.class ); when( schemaRead.indexesGetAll() ).thenReturn( Iterators.iterator( indexes ) ); return schemaRead; }
@Test void shouldHandleSingleItemIterators() { // Given ResourceIterator<Long> it1 = asResourceIterator( iterator( 1L ) ); ResourceIterator<Long> it2 = asResourceIterator( iterator( 5L, 6L, 7L ) ); CombiningResourceIterator<Long> combingIterator = new CombiningResourceIterator<>( iterator(it1, it2) ); // When I iterate through it, things come back in the right order assertThat( Iterators.asList( combingIterator ), equalTo(asList( 1L, 5L, 6L, 7L )) ); } }
@Test public void shouldMatchOnSpecialProperty() { // when List<IndexDescriptor> matched = new ArrayList<>(); NodeSchemaMatcher.onMatchingSchema( iterator( indexOnSpecialProperty ), node, new StubPropertyCursor(), specialPropId, ( schema, props ) -> matched.add( schema ) ); // then assertThat( matched, contains( indexOnSpecialProperty ) ); }
@Test public void shouldMatchOnSingleProperty() { // when List<IndexDescriptor> matched = new ArrayList<>(); NodeSchemaMatcher.onMatchingSchema( iterator( index1 ), node, new StubPropertyCursor(), unIndexedPropId, ( schema, props ) -> matched.add( schema ) ); // then assertThat( matched, contains( index1 ) ); }
@Test public void shouldNotMatchIfNodeIsMissingProperty() { // when List<IndexDescriptor> matched = new ArrayList<>(); NodeSchemaMatcher.onMatchingSchema( iterator( indexWithMissingProperty ), node, new StubPropertyCursor(), unIndexedPropId, ( schema, props ) -> matched.add( schema ) ); // then assertThat( matched, empty() ); }
@Test public void shouldNotMatchIfNodeIsMissingLabel() { // when List<IndexDescriptor> matched = new ArrayList<>(); NodeSchemaMatcher.onMatchingSchema( iterator( indexWithMissingLabel ), node, new StubPropertyCursor(), unIndexedPropId, ( schema, props ) -> matched.add( schema ) ); // then assertThat( matched, empty() ); }
private static Iterator<NumberArrayFactory> arrayFactories() { PageCache pageCache = fixture.pageCache; File dir = fixture.directory; NumberArrayFactory autoWithPageCacheFallback = auto( pageCache, dir, true, NO_MONITOR ); NumberArrayFactory pageCacheArrayFactory = new PageCachedNumberArrayFactory( pageCache, dir ); return Iterators.iterator( HEAP, OFF_HEAP, autoWithPageCacheFallback, pageCacheArrayFactory ); } }
private static Iterator<NumberArrayFactory> arrayFactories() { PageCache pageCache = fixture.pageCache; File dir = fixture.directory; NumberArrayFactory autoWithPageCacheFallback = auto( pageCache, dir, true, NO_MONITOR ); NumberArrayFactory pageCacheArrayFactory = new PageCachedNumberArrayFactory( pageCache, dir ); return Iterators.iterator( HEAP, OFF_HEAP, autoWithPageCacheFallback, pageCacheArrayFactory ); }
IndexDescriptor indexGetForSchema( SchemaDescriptor descriptor ) { IndexDescriptor indexDescriptor = storageReader.indexGetForSchema( descriptor ); Iterator<IndexDescriptor> indexes = iterator( indexDescriptor ); if ( ktx.hasTxStateWithChanges() ) { indexes = filter( SchemaDescriptor.equalTo( descriptor ), ktx.txState().indexDiffSetsBySchema( descriptor ).apply( indexes ) ); } return singleOrNull( indexes ); }
@Test public void testFilterAdded() { // GIVEN diffSets.addAll( iterator( 1L, 2L ) ); diffSets.removeAll( iterator( 3L, 4L ) ); // WHEN MutableDiffSetsImpl<Long> filtered = diffSets.filterAdded( ODD_FILTER ); // THEN assertEquals( asSet( 1L ), filtered.getAdded() ); assertEquals( asSet( 3L, 4L ), filtered.getRemoved() ); }
@Test public void shouldUpdateIndexOnLabelChange() throws Exception { // GIVEN int labelId = 1; long nodeId = 10; start(); // WHEN write( iterator( labelChanges( nodeId, NO_LABELS, new long[]{labelId} ) ) ); // THEN assertNodesForLabel( labelId, nodeId ); }
@Test public void shouldCreateDeferredSchemaIndexesInEmptyDatabase() throws Exception { // GIVEN BatchInserter inserter = newBatchInserter(); // WHEN IndexDefinition definition = inserter.createDeferredSchemaIndex( label( "Hacker" ) ).on( "handle" ).create(); // THEN assertEquals( "Hacker", single( definition.getLabels() ).name() ); assertEquals( asCollection( iterator( "handle" ) ), Iterables.asCollection( definition.getPropertyKeys() ) ); inserter.shutdown(); }