Search in sources :

Example 1 with QueryTreeTransformer

use of com.haulmont.cuba.core.sys.jpql.transform.QueryTreeTransformer in project cuba by cuba-platform.

the class QueryAnalyzerTest method replaceOrderBy.

@Test
public void replaceOrderBy() throws RecognitionException {
    DomainModel model = prepareDomainModel();
    QueryTreeTransformer qa = new QueryTreeTransformer();
    qa.prepare(model, "select c from Car c order by c.model");
    CommonTree tree = qa.getTree();
    CommonTree orderByNode = (CommonTree) tree.getFirstChildWithType(JPA2Lexer.T_ORDER_BY);
    Tree orderByField = orderByNode.getFirstChildWithType(JPA2Lexer.T_ORDER_BY_FIELD);
    assertEquals(1, orderByField.getChildCount());
    PathNode pathNode = (PathNode) orderByField.getChild(0);
    assertEquals("c", pathNode.getEntityVariableName());
    assertEquals("model", pathNode.getChild(0).getText());
    pathNode = new PathNode(JPA2Lexer.T_SELECTED_FIELD, "c");
    pathNode.addDefaultChild("regNumber");
    qa.replaceOrderBy(true, new PathEntityReference(pathNode, "Car"));
    orderByNode = (CommonTree) tree.getFirstChildWithType(JPA2Lexer.T_ORDER_BY);
    orderByField = orderByNode.getFirstChildWithType(JPA2Lexer.T_ORDER_BY_FIELD);
    assertEquals(2, orderByField.getChildCount());
    pathNode = (PathNode) orderByField.getChild(0);
    assertEquals("c", pathNode.getEntityVariableName());
    assertEquals("regNumber", pathNode.getChild(0).getText());
    assertEquals("desc", orderByField.getChild(1).getText());
}
Also used : PathEntityReference(com.haulmont.cuba.core.sys.jpql.transform.PathEntityReference) DomainModel(com.haulmont.cuba.core.sys.jpql.DomainModel) CommonTree(org.antlr.runtime.tree.CommonTree) CommonTree(org.antlr.runtime.tree.CommonTree) Tree(org.antlr.runtime.tree.Tree) QueryTreeTransformer(com.haulmont.cuba.core.sys.jpql.transform.QueryTreeTransformer) Test(org.junit.Test)

Example 2 with QueryTreeTransformer

use of com.haulmont.cuba.core.sys.jpql.transform.QueryTreeTransformer in project cuba by cuba-platform.

the class QueryAnalyzerTest method testTree.

@Test
public void testTree() throws RecognitionException {
    DomainModel model = prepareDomainModel();
    QueryTreeTransformer qa = new QueryTreeTransformer();
    String query = "select f from sec$SearchFolder f " + "left join f.user u " + "left join f.presentation p " + "where (f.user.id = ?1 or f.user is null) " + "order by f.sortOrder, f.name";
    qa.prepare(model, query);
}
Also used : DomainModel(com.haulmont.cuba.core.sys.jpql.DomainModel) QueryTreeTransformer(com.haulmont.cuba.core.sys.jpql.transform.QueryTreeTransformer) Test(org.junit.Test)

Example 3 with QueryTreeTransformer

use of com.haulmont.cuba.core.sys.jpql.transform.QueryTreeTransformer in project cuba by cuba-platform.

the class QueryAnalyzerTest method replaceWithCount_distinct.

@Test
public void replaceWithCount_distinct() throws RecognitionException {
    DomainModel model = prepareDomainModel();
    QueryTreeTransformer qa = new QueryTreeTransformer();
    qa.prepare(model, "select distinct d from Car c, in(c.drivers) d order by d.name");
    CommonTree tree = qa.getTree();
    CommonTree selectedItems = (CommonTree) tree.getFirstChildWithType(JPA2Lexer.T_SELECTED_ITEMS);
    Tree selectedItem = selectedItems.getFirstChildWithType(JPA2Lexer.T_SELECTED_ITEM);
    PathNode pathNode = (PathNode) selectedItem.getChild(0);
    assertEquals("d", pathNode.getEntityVariableName());
    assertEquals(0, pathNode.getChildCount());
    CommonTree orderByNode = (CommonTree) tree.getFirstChildWithType(JPA2Lexer.T_ORDER_BY);
    assertNotNull(orderByNode.getFirstChildWithType(JPA2Lexer.T_ORDER_BY_FIELD));
    qa.replaceWithCount(new VariableEntityReference("Driver", "d"));
    selectedItems = (CommonTree) tree.getFirstChildWithType(JPA2Lexer.T_SELECTED_ITEMS);
    assertEquals(1, selectedItems.getChildCount());
    selectedItem = selectedItems.getFirstChildWithType(JPA2Lexer.T_SELECTED_ITEM);
    assertTrue(selectedItem.getChild(0) instanceof AggregateExpressionNode);
    AggregateExpressionNode countExpr = (AggregateExpressionNode) selectedItem.getChild(0);
    assertEquals("count", countExpr.getChild(0).getText());
    assertEquals("distinct", countExpr.getChild(2).getText());
    assertEquals("d", countExpr.getChild(3).getText());
    assertEquals(5, countExpr.getChildCount());
    assertNull(orderByNode.getFirstChildWithType(JPA2Lexer.T_ORDER_BY));
}
Also used : DomainModel(com.haulmont.cuba.core.sys.jpql.DomainModel) CommonTree(org.antlr.runtime.tree.CommonTree) CommonTree(org.antlr.runtime.tree.CommonTree) Tree(org.antlr.runtime.tree.Tree) QueryTreeTransformer(com.haulmont.cuba.core.sys.jpql.transform.QueryTreeTransformer) VariableEntityReference(com.haulmont.cuba.core.sys.jpql.transform.VariableEntityReference) Test(org.junit.Test)

Example 4 with QueryTreeTransformer

use of com.haulmont.cuba.core.sys.jpql.transform.QueryTreeTransformer in project cuba by cuba-platform.

the class QueryAnalyzerTest method mixinJoinIntoTree_with_in_collections.

@Test
public void mixinJoinIntoTree_with_in_collections() throws RecognitionException {
    EntityBuilder builder = new EntityBuilder();
    builder.startNewEntity("HomeBase");
    builder.addStringAttribute("name");
    JpqlEntityModel homeBase = builder.produce();
    builder.startNewEntity("Driver");
    builder.addStringAttribute("name");
    builder.addStringAttribute("signal");
    JpqlEntityModel driver = builder.produce();
    builder.startNewEntity("Car");
    builder.addStringAttribute("model");
    builder.addCollectionReferenceAttribute("drivers", "Driver");
    builder.addReferenceAttribute("station", "HomeBase");
    JpqlEntityModel car = builder.produce();
    DomainModel model = new DomainModel(car, driver, homeBase);
    JoinVariableNode join = Parser.parseJoinClause("join c.station h").get(0);
    QueryTreeTransformer qa = new QueryTreeTransformer();
    qa.prepare(model, "select d.name from Car c, in(c.drivers) d");
    CommonTree tree = qa.getTree();
    CommonTree sources = (CommonTree) tree.getFirstChildWithType(JPA2Lexer.T_SOURCES);
    assertEquals(2, sources.getChildCount());
    assertTrue(sources.getChild(0) instanceof SelectionSourceNode);
    CommonTree source0 = (CommonTree) sources.getChild(0);
    assertEquals(1, source0.getChildCount());
    assertTrue(source0.getChild(0) instanceof IdentificationVariableNode);
    assertTrue(sources.getChild(1) instanceof SelectionSourceNode);
    CommonTree source1 = (CommonTree) sources.getChild(1);
    assertTrue(source1.getChild(0) instanceof CollectionMemberNode);
    qa.mixinJoinIntoTree(join, new VariableEntityReference("Car", "c"), true);
    tree = qa.getTree();
    sources = (CommonTree) tree.getFirstChildWithType(JPA2Lexer.T_SOURCES);
    assertEquals(2, sources.getChildCount());
    assertTrue(sources.getChild(0) instanceof SelectionSourceNode);
    source0 = (CommonTree) sources.getChild(0);
    assertEquals(2, source0.getChildCount());
    assertTrue(source0.getChild(0) instanceof IdentificationVariableNode);
    assertTrue(source0.getChild(1) instanceof JoinVariableNode);
    assertTrue(sources.getChild(1) instanceof SelectionSourceNode);
    source1 = (CommonTree) sources.getChild(1);
    assertTrue(source1.getChild(0) instanceof CollectionMemberNode);
}
Also used : DomainModel(com.haulmont.cuba.core.sys.jpql.DomainModel) CommonTree(org.antlr.runtime.tree.CommonTree) EntityBuilder(com.haulmont.cuba.core.sys.jpql.model.EntityBuilder) JpqlEntityModel(com.haulmont.cuba.core.sys.jpql.model.JpqlEntityModel) QueryTreeTransformer(com.haulmont.cuba.core.sys.jpql.transform.QueryTreeTransformer) VariableEntityReference(com.haulmont.cuba.core.sys.jpql.transform.VariableEntityReference) Test(org.junit.Test)

Example 5 with QueryTreeTransformer

use of com.haulmont.cuba.core.sys.jpql.transform.QueryTreeTransformer in project cuba by cuba-platform.

the class QueryAnalyzerTest method mixinWhereConditionsIntoTree.

@Test
public void mixinWhereConditionsIntoTree() throws RecognitionException {
    DomainModel model = prepareDomainModel();
    QueryTreeTransformer qa = new QueryTreeTransformer();
    qa.prepare(model, "select c from Car c");
    WhereNode where = (WhereNode) Parser.parseWhereClause("where c.model = ?1");
    CommonTree tree = qa.getTree();
    assertNull(tree.getFirstChildWithType(JPA2Lexer.T_CONDITION));
    qa.mixinWhereConditionsIntoTree(where);
    tree = qa.getTree();
    assertNotNull(tree.getFirstChildWithType(JPA2Lexer.T_CONDITION));
}
Also used : DomainModel(com.haulmont.cuba.core.sys.jpql.DomainModel) CommonTree(org.antlr.runtime.tree.CommonTree) QueryTreeTransformer(com.haulmont.cuba.core.sys.jpql.transform.QueryTreeTransformer) Test(org.junit.Test)

Aggregations

DomainModel (com.haulmont.cuba.core.sys.jpql.DomainModel)8 QueryTreeTransformer (com.haulmont.cuba.core.sys.jpql.transform.QueryTreeTransformer)8 Test (org.junit.Test)8 CommonTree (org.antlr.runtime.tree.CommonTree)7 VariableEntityReference (com.haulmont.cuba.core.sys.jpql.transform.VariableEntityReference)5 Tree (org.antlr.runtime.tree.Tree)3 TreeToQuery (com.haulmont.cuba.core.sys.jpql.TreeToQuery)1 EntityBuilder (com.haulmont.cuba.core.sys.jpql.model.EntityBuilder)1 JpqlEntityModel (com.haulmont.cuba.core.sys.jpql.model.JpqlEntityModel)1 PathEntityReference (com.haulmont.cuba.core.sys.jpql.transform.PathEntityReference)1 TreeVisitor (org.antlr.runtime.tree.TreeVisitor)1