Examples with LoadPlan org.hibernate.loader.plan.spi.LoadPlan used on opensource projects

Example 1 with LoadPlan

use of org.hibernate.loader.plan.spi.LoadPlan in project hibernate-orm by hibernate.

the class EntityGraphLoadPlanBuilderTest method testBasicLoadLoadPlanBuilding.

/**
	 * EntityGraph(1):
	 *
	 * Cat
	 *
	 * LoadPlan:
	 *
	 * Cat
	 *
	 * ---------------------
	 *
	 * EntityGraph(2):
	 *
	 * Cat
	 * owner -- Person
	 *
	 * LoadPlan:
	 *
	 * Cat
	 * owner -- Person
	 * address --- Address
	 * country -- Country
	 */
@Test
public void testBasicLoadLoadPlanBuilding() {
    EntityManager em = getOrCreateEntityManager();
    EntityGraph eg = em.createEntityGraph(Cat.class);
    LoadPlan plan = buildLoadPlan(eg, Mode.LOAD, Cat.class);
    LoadPlanTreePrinter.INSTANCE.logTree(plan, new AliasResolutionContextImpl(sfi()));
    QuerySpace rootQuerySpace = plan.getQuerySpaces().getRootQuerySpaces().get(0);
    assertFalse("With fetchgraph property and an empty EntityGraph, there should be no join at all", rootQuerySpace.getJoins().iterator().hasNext());
    // -------------------------------------------------- another a little more complicated case
    eg = em.createEntityGraph(Cat.class);
    eg.addSubgraph("owner", Person.class);
    plan = buildLoadPlan(eg, Mode.LOAD, Cat.class);
    LoadPlanTreePrinter.INSTANCE.logTree(plan, new AliasResolutionContextImpl(sfi()));
    rootQuerySpace = plan.getQuerySpaces().getRootQuerySpaces().get(0);
    Iterator<Join> iterator = rootQuerySpace.getJoins().iterator();
    assertTrue("With fetchgraph property and an empty EntityGraph, there should be no join at all", iterator.hasNext());
    Join personJoin = iterator.next();
    assertNotNull(personJoin);
    QuerySpace.Disposition disposition = personJoin.getRightHandSide().getDisposition();
    assertEquals("This should be an entity join which fetches Person", QuerySpace.Disposition.ENTITY, disposition);
    iterator = personJoin.getRightHandSide().getJoins().iterator();
    assertTrue("The composite address should be fetched", iterator.hasNext());
    Join addressJoin = iterator.next();
    assertNotNull(addressJoin);
    disposition = addressJoin.getRightHandSide().getDisposition();
    assertEquals(QuerySpace.Disposition.COMPOSITE, disposition);
    iterator = addressJoin.getRightHandSide().getJoins().iterator();
    assertTrue(iterator.hasNext());
    Join countryJoin = iterator.next();
    assertNotNull(countryJoin);
    disposition = countryJoin.getRightHandSide().getDisposition();
    assertEquals(QuerySpace.Disposition.ENTITY, disposition);
    assertFalse("The ManyToOne attribute in composite should not be fetched", countryJoin.getRightHandSide().getJoins().iterator().hasNext());
    em.close();
}

Example 2 with LoadPlan

use of org.hibernate.loader.plan.spi.LoadPlan in project hibernate-orm by hibernate.

the class EntityGraphLoadPlanBuilderTest method testEmbeddedCollection.

@Test
public void testEmbeddedCollection() {
    EntityManager em = getOrCreateEntityManager();
    EntityGraph eg = em.createEntityGraph(ExpressCompany.class);
    eg.addAttributeNodes("shipAddresses");
    //WTF name!!!
    LoadPlan loadLoadPlan = buildLoadPlan(eg, Mode.LOAD, ExpressCompany.class);
    LoadPlanTreePrinter.INSTANCE.logTree(loadLoadPlan, new AliasResolutionContextImpl(sfi()));
    QuerySpace querySpace = loadLoadPlan.getQuerySpaces().getRootQuerySpaces().iterator().next();
    Iterator<Join> iterator = querySpace.getJoins().iterator();
    assertTrue(iterator.hasNext());
    Join collectionJoin = iterator.next();
    assertEquals(QuerySpace.Disposition.COLLECTION, collectionJoin.getRightHandSide().getDisposition());
    assertFalse(iterator.hasNext());
    iterator = collectionJoin.getRightHandSide().getJoins().iterator();
    assertTrue(iterator.hasNext());
    Join collectionElementJoin = iterator.next();
    assertFalse(iterator.hasNext());
    assertEquals(QuerySpace.Disposition.COMPOSITE, collectionElementJoin.getRightHandSide().getDisposition());
    iterator = collectionElementJoin.getRightHandSide().getJoins().iterator();
    assertTrue(iterator.hasNext());
    Join countryJoin = iterator.next();
    assertFalse(iterator.hasNext());
    assertEquals(QuerySpace.Disposition.ENTITY, countryJoin.getRightHandSide().getDisposition());
    //----------------------------------------------------------------
    LoadPlan fetchLoadPlan = buildLoadPlan(eg, Mode.FETCH, ExpressCompany.class);
    LoadPlanTreePrinter.INSTANCE.logTree(fetchLoadPlan, new AliasResolutionContextImpl(sfi()));
    querySpace = fetchLoadPlan.getQuerySpaces().getRootQuerySpaces().iterator().next();
    iterator = querySpace.getJoins().iterator();
    assertTrue(iterator.hasNext());
    collectionJoin = iterator.next();
    assertEquals(QuerySpace.Disposition.COLLECTION, collectionJoin.getRightHandSide().getDisposition());
    assertFalse(iterator.hasNext());
    iterator = collectionJoin.getRightHandSide().getJoins().iterator();
    assertTrue(iterator.hasNext());
    collectionElementJoin = iterator.next();
    assertFalse(iterator.hasNext());
    assertEquals(QuerySpace.Disposition.COMPOSITE, collectionElementJoin.getRightHandSide().getDisposition());
    iterator = collectionElementJoin.getRightHandSide().getJoins().iterator();
    assertFalse(iterator.hasNext());
    //----------------------------------------------------------------
    em.close();
}

Example 3 with LoadPlan

use of org.hibernate.loader.plan.spi.LoadPlan in project hibernate-orm by hibernate.

the class EntityGraphLoadPlanBuilderTest method testBasicFetchLoadPlanBuilding.

/**
	 * EntityGraph(1):
	 *
	 * Cat
	 *
	 * LoadPlan:
	 *
	 * Cat
	 *
	 * ---------------------
	 *
	 * EntityGraph(2):
	 *
	 * Cat
	 * owner -- Person
	 *
	 * LoadPlan:
	 *
	 * Cat
	 * owner -- Person
	 * address --- Address
	 */
@Test
public void testBasicFetchLoadPlanBuilding() {
    EntityManager em = getOrCreateEntityManager();
    EntityGraph eg = em.createEntityGraph(Cat.class);
    LoadPlan plan = buildLoadPlan(eg, Mode.FETCH, Cat.class);
    LoadPlanTreePrinter.INSTANCE.logTree(plan, new AliasResolutionContextImpl(sfi()));
    QuerySpace rootQuerySpace = plan.getQuerySpaces().getRootQuerySpaces().get(0);
    assertFalse("With fetchgraph property and an empty EntityGraph, there should be no join at all", rootQuerySpace.getJoins().iterator().hasNext());
    // -------------------------------------------------- another a little more complicated case
    eg = em.createEntityGraph(Cat.class);
    eg.addSubgraph("owner", Person.class);
    plan = buildLoadPlan(eg, Mode.FETCH, Cat.class);
    LoadPlanTreePrinter.INSTANCE.logTree(plan, new AliasResolutionContextImpl(sfi()));
    rootQuerySpace = plan.getQuerySpaces().getRootQuerySpaces().get(0);
    Iterator<Join> iterator = rootQuerySpace.getJoins().iterator();
    assertTrue("With fetchgraph property and an empty EntityGraph, there should be no join at all", iterator.hasNext());
    Join personJoin = iterator.next();
    assertNotNull(personJoin);
    QuerySpace.Disposition disposition = personJoin.getRightHandSide().getDisposition();
    assertEquals("This should be an entity join which fetches Person", QuerySpace.Disposition.ENTITY, disposition);
    iterator = personJoin.getRightHandSide().getJoins().iterator();
    assertTrue("The composite address should be fetched", iterator.hasNext());
    Join addressJoin = iterator.next();
    assertNotNull(addressJoin);
    disposition = addressJoin.getRightHandSide().getDisposition();
    assertEquals(QuerySpace.Disposition.COMPOSITE, disposition);
    assertFalse(iterator.hasNext());
    assertFalse("The ManyToOne attribute in composite should not be fetched", addressJoin.getRightHandSide().getJoins().iterator().hasNext());
    em.close();
}

Example 4 with LoadPlan

use of org.hibernate.loader.plan.spi.LoadPlan in project hibernate-orm by hibernate.

the class EntityGraphLoadPlanBuilderTest method testBasicElementCollections.

@Test
public void testBasicElementCollections() {
    EntityManager em = getOrCreateEntityManager();
    EntityGraph eg = em.createEntityGraph(Dog.class);
    eg.addAttributeNodes("favorites");
    //WTF name!!!
    LoadPlan loadLoadPlan = buildLoadPlan(eg, Mode.LOAD, Dog.class);
    LoadPlanTreePrinter.INSTANCE.logTree(loadLoadPlan, new AliasResolutionContextImpl(sfi()));
    QuerySpace querySpace = loadLoadPlan.getQuerySpaces().getRootQuerySpaces().iterator().next();
    Iterator<Join> iterator = querySpace.getJoins().iterator();
    assertTrue(iterator.hasNext());
    Join collectionJoin = iterator.next();
    assertEquals(QuerySpace.Disposition.COLLECTION, collectionJoin.getRightHandSide().getDisposition());
    assertFalse(iterator.hasNext());
    //----------------------------------------------------------------
    LoadPlan fetchLoadPlan = buildLoadPlan(eg, Mode.FETCH, Dog.class);
    LoadPlanTreePrinter.INSTANCE.logTree(fetchLoadPlan, new AliasResolutionContextImpl(sfi()));
    querySpace = fetchLoadPlan.getQuerySpaces().getRootQuerySpaces().iterator().next();
    iterator = querySpace.getJoins().iterator();
    assertTrue(iterator.hasNext());
    collectionJoin = iterator.next();
    assertEquals(QuerySpace.Disposition.COLLECTION, collectionJoin.getRightHandSide().getDisposition());
    assertFalse(iterator.hasNext());
    em.close();
}

Example 5 with LoadPlan

use of org.hibernate.loader.plan.spi.LoadPlan in project hibernate-orm by hibernate.

the class EncapsulatedCompositeAttributeResultSetProcessorTest method getResults.

private List<?> getResults(EntityPersister entityPersister) {
    final LoadPlan plan = Helper.INSTANCE.buildLoadPlan(sessionFactory(), entityPersister);
    final LoadQueryDetails queryDetails = Helper.INSTANCE.buildLoadQueryDetails(plan, sessionFactory());
    final String sql = queryDetails.getSqlStatement();
    final ResultSetProcessor resultSetProcessor = queryDetails.getResultSetProcessor();
    final List results = new ArrayList();
    final Session workSession = openSession();
    workSession.beginTransaction();
    workSession.doWork(new Work() {

        @Override
        public void execute(Connection connection) throws SQLException {
            PreparedStatement ps = connection.prepareStatement(sql);
            ps.setInt(1, 1);
            ResultSet resultSet = ps.executeQuery();
            results.addAll(resultSetProcessor.extractResults(resultSet, (SessionImplementor) workSession, new QueryParameters(), new NamedParameterContext() {

                @Override
                public int[] getNamedParameterLocations(String name) {
                    return new int[0];
                }
            }, true, false, null, null));
            resultSet.close();
            ps.close();
        }
    });
    workSession.getTransaction().commit();
    workSession.close();
    return results;
}