@Override public boolean hasNext() { if ( source == null ) { return false; } if ( currentIterator != null && currentIterator.hasNext() ) { return true; } while ( source.hasNext() ) { EntityRef ref = source.next(); Results r = getResultsFor( ref ); if ( r.size() > 0 ) { currentIterator = new PagingResultsIterator( r, query.getResultsLevel(), null); return currentIterator.hasNext(); } } currentIterator = null; source = null; return false; }
@Override public boolean hasNext() { if (source == null) { return false; } if (currentIterator != null && currentIterator.hasNext()) { return true; } while (source.hasNext()) { Object next = source.next(); Results r; EntityRef ref = (EntityRef) next; r = getResultsFor(ref); if (r.size() > 0) { if(ref.getType().equals(Group.ENTITY_TYPE)) { currentIterator = new PagingResultsIterator(r, query.getResultsLevel(), Query.Level.REFS); }else{ currentIterator = new PagingResultsIterator(r, query.getResultsLevel(), null); } return currentIterator.hasNext(); } } currentIterator = null; source = null; return false; }
public Iterator<E> graphIterator( EntityManager em ) { try { if ( uuid != null && type != null ) { return new PagingResultsIterator( getHeadResults( em ), query.getResultsLevel(), null); }else { return new NotificationGraphIterator(em, source.refIterator(em, true), query); } } catch ( Exception e ) { throw new RuntimeException( e ); } }
public Iterator<E> iterator( EntityManager em ) { try { if ( uuid != null && type != null ) { return new PagingResultsIterator( getHeadResults( em ), query.getResultsLevel(), null); } else { return new MultiQueryIterator( em, source.refIterator( em, false), query ); } } catch ( Exception e ) { throw new RuntimeException( e ); } }
PagingResultsIterator itr = new PagingResultsIterator( entities ); for ( Object e : itr ) { starting_time = checkTimeDelta( starting_time, jobExecution );
PagingResultsIterator itr = new PagingResultsIterator(results);
@Override public BiMap<UUID, String> getApplicationsForOrganization( UUID organizationGroupId ) throws Exception { if ( organizationGroupId == null ) { return null; } final BiMap<UUID, String> applications = HashBiMap.create(); final EntityManager em = emf.getEntityManager(smf.getManagementAppId()); // query for application_info entities final Results results = em.getTargetEntities( new SimpleEntityRef(Group.ENTITY_TYPE, organizationGroupId), ORG_APP_RELATIONSHIP, CpNamingUtils.APPLICATION_INFO, Level.ALL_PROPERTIES); final PagingResultsIterator itr = new PagingResultsIterator( results ); String entityName; while ( itr.hasNext() ) { final Entity entity = ( Entity ) itr.next(); entityName = entity.getName(); if ( entityName != null ) { entityName = entityName.toLowerCase(); } // make sure we return applicationId and not the application_info UUID UUID applicationId = entity.getUuid(); applications.put( applicationId, entityName ); } return applications; }
protected Iterator refIterator(EntityManager em, boolean useGraph) throws Exception { if ( query.getQl() == null && query.getSingleNameOrEmailIdentifier() != null){ return new PagingResultsIterator( getHeadResults( em ), Level.REFS, null); } if ( type != null && uuid != null) { return new PagingResultsIterator( getHeadResults( em ), Level.REFS, null); } else { Query q = query; if ( query.getResultsLevel() != Level.REFS ) { // ensure REFS level q = new Query( q ); q.setResultsLevel( Level.REFS ); } if( useGraph){ return new NotificationGraphIterator( em, source.refIterator( em, true), q ); }else{ return new MultiQueryIterator( em, source.refIterator( em, false ), q ); } } }
PagingResultsIterator itr = new PagingResultsIterator(entities);
private int getConnectionCountViaGet( final Import importRoot ) { try { EntityManager emMgmtApp = setup.getEmf() .getEntityManager(setup.getEmf().getManagementAppId() ); Results entities = emMgmtApp.getTargetEntities( importRoot, "includes", null, Query.Level.ALL_PROPERTIES); PagingResultsIterator itr = new PagingResultsIterator( entities ); int count = 0; while ( itr.hasNext() ) { itr.next(); count++; } return count; } catch ( Exception e ) { logger.error( "application doesn't exist within the current context" ); throw new RuntimeException( e ); } }
PagingResultsIterator itr = new PagingResultsIterator(em.getSourceEntities( returned, "following", "user", Level.ALL_PROPERTIES, 1000));
PagingResultsIterator pri = new PagingResultsIterator( results ); int count = 13; while ( pri.hasNext() ) {
@Test public void testConnectionsIterable() throws Exception { EntityManager em = app.getEntityManager(); assertNotNull( em ); User first = new User(); first.setUsername( "first" ); first.setEmail( "first@usergrid.com" ); Entity firstUserEntity = em.create( first ); assertNotNull( firstUserEntity ); final int connectionCount = 100; final Map<UUID, Entity> things = new HashMap<>(); for(int i = 0; i < connectionCount; i ++){ Map<String, Object> data = new HashMap<String, Object>(); data.put( "ordinal", i ); Entity entity = em.create( "thing", data ); em.createConnection( firstUserEntity, "likes", entity ); things.put( entity.getUuid(), entity ); } app.waitForQueueDrainAndRefreshIndex(); Results r = em.getTargetEntities(firstUserEntity, "likes", null, Level.ALL_PROPERTIES) ; PagingResultsIterator itr = new PagingResultsIterator( r ); int checkedIndex = 0; for(; checkedIndex < connectionCount && itr.hasNext(); checkedIndex ++){ final Entity returned = ( Entity ) itr.next(); final Entity expected = things.get( returned.getUuid() ); assertEquals("Entity expected", expected, returned); } assertEquals("Checked all entities", connectionCount, checkedIndex ); }