Examples with QueryCompilation org.datanucleus.query.compiler.QueryCompilation used on opensource projects

Example 1 with QueryCompilation

use of org.datanucleus.query.compiler.QueryCompilation in project datanucleus-rdbms by datanucleus.

the class QueryToSQLMapper method processVariableExpression.

/* (non-Javadoc)
     * @see org.datanucleus.query.evaluator.AbstractExpressionEvaluator#processVariableExpression(org.datanucleus.query.expression.VariableExpression)
     */
@Override
protected Object processVariableExpression(VariableExpression expr) {
    String varName = expr.getId();
    Symbol varSym = expr.getSymbol();
    if (varSym != null) {
        // Use name from symbol if possible
        varName = varSym.getQualifiedName();
    }
    if (hasSQLTableMappingForAlias(varName)) {
        // Variable already found
        SQLTableMapping tblMapping = getSQLTableMappingForAlias(varName);
        SQLExpression sqlExpr = exprFactory.newExpression(tblMapping.table.getSQLStatement(), tblMapping.table, tblMapping.mapping);
        stack.push(sqlExpr);
        return sqlExpr;
    } else if (compilation.getCompilationForSubquery(varName) != null) {
        // Subquery variable
        QueryCompilation subCompilation = compilation.getCompilationForSubquery(varName);
        AbstractClassMetaData subCmd = ec.getMetaDataManager().getMetaDataForClass(subCompilation.getCandidateClass(), ec.getClassLoaderResolver());
        // Create subquery statement, using any provided alias if possible
        String subAlias = null;
        if (subCompilation.getCandidateAlias() != null && !subCompilation.getCandidateAlias().equals(candidateAlias)) {
            subAlias = subCompilation.getCandidateAlias();
        }
        StatementResultMapping subqueryResultMapping = new StatementResultMapping();
        // TODO Fix "avg(something)" arg - not essential but is a hack right now
        SQLStatement subStmt = RDBMSQueryUtils.getStatementForCandidates(storeMgr, stmt, subCmd, null, ec, subCompilation.getCandidateClass(), true, "avg(something)", subAlias, null, null);
        QueryToSQLMapper sqlMapper = new QueryToSQLMapper(subStmt, subCompilation, parameters, null, subqueryResultMapping, subCmd, true, fetchPlan, ec, importsDefinition, options, extensionsByName);
        sqlMapper.setDefaultJoinType(defaultJoinType);
        sqlMapper.setDefaultJoinTypeFilter(defaultJoinTypeFilter);
        sqlMapper.setParentMapper(this);
        sqlMapper.compile();
        if (subqueryResultMapping.getNumberOfResultExpressions() > 1) {
            throw new NucleusUserException("Number of result expressions in subquery should be 1");
        }
        SQLExpression subExpr = null;
        // TODO Cater for subquery select of its own candidate
        if (subqueryResultMapping.getNumberOfResultExpressions() == 0) {
            subExpr = new org.datanucleus.store.rdbms.sql.expression.SubqueryExpression(stmt, subStmt);
        } else {
            JavaTypeMapping subMapping = ((StatementMappingIndex) subqueryResultMapping.getMappingForResultExpression(0)).getMapping();
            if (subMapping instanceof TemporalMapping) {
                subExpr = new TemporalSubqueryExpression(stmt, subStmt);
            } else if (subMapping instanceof StringMapping) {
                subExpr = new StringSubqueryExpression(stmt, subStmt);
            } else {
                subExpr = new NumericSubqueryExpression(stmt, subStmt);
            }
            if (subExpr.getJavaTypeMapping() == null) {
                subExpr.setJavaTypeMapping(subMapping);
            }
        }
        stack.push(subExpr);
        return subExpr;
    } else if (stmt.getParentStatement() != null && parentMapper != null && parentMapper.candidateAlias != null && parentMapper.candidateAlias.equals(varName)) {
        // Variable in subquery linking back to parent query
        SQLExpression varExpr = exprFactory.newExpression(stmt.getParentStatement(), stmt.getParentStatement().getPrimaryTable(), stmt.getParentStatement().getPrimaryTable().getTable().getIdMapping());
        stack.push(varExpr);
        return varExpr;
    } else {
        // Variable never met before, so return as UnboundExpression - process later if needing binding
        NucleusLogger.QUERY.debug("QueryToSQL.processVariable (unbound) variable=" + varName + " is not yet bound so returning UnboundExpression");
        UnboundExpression unbExpr = new UnboundExpression(stmt, varName);
        stack.push(unbExpr);
        return unbExpr;
    }
}

Example 2 with QueryCompilation

use of org.datanucleus.query.compiler.QueryCompilation in project tests by datanucleus.

the class JPQLCompilerTest method testFilterWithStringIndexOfLiteral.

/**
 * Tests for "String.indexOf(Literal, int)" in filter.
 */
public void testFilterWithStringIndexOfLiteral() {
    JavaQueryCompiler compiler = null;
    QueryCompilation compilation = null;
    try {
        compiler = new JPQLCompiler(nucCtx, nucCtx.getClassLoaderResolver(null), null, Project.class, null, "LENGTH(name) > 5", null, null, null, null, null, null, null);
        compilation = compiler.compile(new HashMap(), null);
    } catch (NucleusException ne) {
        NucleusLogger.QUERY.error("Exception during compile", ne);
        fail("compilation of filter with valid field threw exception : " + ne.getMessage());
    }
    Expression expr = compilation.getExprFilter();
    assertTrue("Compiled expression should have been DyadicExpression but wasnt", expr instanceof DyadicExpression);
    DyadicExpression dyExpr = (DyadicExpression) expr;
    assertTrue("DyadicExpression operator should have been > but wasnt", dyExpr.getOperator() == Expression.OP_GT);
    assertTrue("DyadicExpression left should have been InvokeExpression but wasnt", dyExpr.getLeft() instanceof InvokeExpression);
    InvokeExpression leftInvExpr = (InvokeExpression) dyExpr.getLeft();
    assertTrue("InvokeExpression invoked object should have been PrimaryExpression but wasnt", leftInvExpr.getLeft() instanceof PrimaryExpression);
    PrimaryExpression invPrimExpr = (PrimaryExpression) leftInvExpr.getLeft();
    assertEquals("PrimaryExpression field is wrong", "name", invPrimExpr.getId());
    assertEquals("InvokeExpression method is wrong", "length", leftInvExpr.getOperation());
    List args = leftInvExpr.getArguments();
    assertTrue("Number of args should be 0 but isnt", args == null || args.isEmpty());
    assertTrue("DyadicExpression left should have been Literal but wasnt", dyExpr.getRight() instanceof Literal);
    Literal rightExpr = (Literal) dyExpr.getRight();
    assertEquals("Parameter2 to indexOf() has wrong value", new Long(5), rightExpr.getLiteral());
}

Example 3 with QueryCompilation

use of org.datanucleus.query.compiler.QueryCompilation in project tests by datanucleus.

the class JPQLCompilerTest method testFilterComparisonWithAndOr.

/**
 * Tests for filter with field-literal comparison AND another comparison, and ORed with
 * another set of expressions.
 */
public void testFilterComparisonWithAndOr() {
    JavaQueryCompiler compiler = null;
    QueryCompilation compilation = null;
    try {
        compiler = new JPQLCompiler(nucCtx, nucCtx.getClassLoaderResolver(null), null, Project.class, null, "(budget = 2 AND 'Sales' = name) OR (budget >= 50 AND name = 'Marketing')", null, null, null, null, null, null, null);
        compilation = compiler.compile(null, null);
    } catch (NucleusException ne) {
        NucleusLogger.QUERY.error("Exception during compile", ne);
        fail("compilation of filter with valid field threw exception : " + ne.getMessage());
    }
    Expression expr = compilation.getExprFilter();
    assertTrue("Compiled expression should have been DyadicExpression but wasnt", expr instanceof DyadicExpression);
    DyadicExpression dyExpr = (DyadicExpression) expr;
    assertTrue("Compiled left expression should be DyadicExpression but isnt", dyExpr.getLeft() instanceof DyadicExpression);
    assertTrue("Compiled right expression should be DyadicExpression but isnt", dyExpr.getRight() instanceof DyadicExpression);
    DyadicExpression dyExpr1 = (DyadicExpression) dyExpr.getLeft();
    DyadicExpression dyExpr2 = (DyadicExpression) dyExpr.getRight();
    assertEquals("Operator between left and right is incorrect", Expression.OP_OR, dyExpr.getOperator());
    assertTrue("Compiled left(left) expression should be DyadicExpression but isnt", dyExpr1.getLeft() instanceof DyadicExpression);
    assertTrue("Compiled left(right) expression should be DyadicExpression but isnt", dyExpr1.getRight() instanceof DyadicExpression);
    DyadicExpression dyExpr1a = (DyadicExpression) dyExpr1.getLeft();
    DyadicExpression dyExpr1b = (DyadicExpression) dyExpr1.getRight();
    // 1a : budget == 2
    assertTrue("Compiled left expression should be PrimaryExpression but isnt", dyExpr1a.getLeft() instanceof PrimaryExpression);
    assertTrue("Compiled right expression should be Literal but isnt", dyExpr1a.getRight() instanceof Literal);
    assertEquals("Operator between left (left and right) is incorrect", Expression.OP_EQ, dyExpr1a.getOperator());
    PrimaryExpression leftExpr1a = (PrimaryExpression) dyExpr1a.getLeft();
    assertEquals("Compiled left expression has incorrect number of tuples", 1, leftExpr1a.getTuples().size());
    assertEquals("Compiled left expression 'id' is incorrect", "budget", leftExpr1a.getId());
    Literal rightExpr1a = (Literal) dyExpr1a.getRight();
    assertTrue("Compiled right expression literal is of incorrect type", rightExpr1a.getLiteral() instanceof Long);
    assertEquals("Compiled right expression literal has incorrect value", 2, ((Long) rightExpr1a.getLiteral()).longValue());
    // 1b : "Sales" == name
    assertTrue("Compiled right expression should be PrimaryExpression but isnt", dyExpr1b.getRight() instanceof PrimaryExpression);
    assertTrue("Compiled left expression should be Literal but isnt", dyExpr1b.getLeft() instanceof Literal);
    assertEquals("Operator between right (left and right) is incorrect", Expression.OP_EQ, dyExpr1b.getOperator());
    PrimaryExpression rightExpr1b = (PrimaryExpression) dyExpr1b.getRight();
    assertEquals("Compiled left expression has incorrect number of tuples", 1, rightExpr1b.getTuples().size());
    assertEquals("Compiled left expression 'id' is incorrect", "name", rightExpr1b.getId());
    Literal leftExpr1b = (Literal) dyExpr1b.getLeft();
    assertTrue("Compiled right expression literal is of incorrect type", leftExpr1b.getLiteral() instanceof String);
    assertEquals("Compiled right expression literal has incorrect value", "Sales", ((String) leftExpr1b.getLiteral()));
    assertTrue("Compiled right(left) expression should be DyadicExpression but isnt", dyExpr2.getLeft() instanceof DyadicExpression);
    assertTrue("Compiled right(right) expression should be DyadicExpression but isnt", dyExpr2.getRight() instanceof DyadicExpression);
    DyadicExpression dyExpr2a = (DyadicExpression) dyExpr2.getLeft();
    DyadicExpression dyExpr2b = (DyadicExpression) dyExpr2.getRight();
    // 2a : budget >= 50
    assertTrue("Compiled left expression should be PrimaryExpression but isnt", dyExpr2a.getLeft() instanceof PrimaryExpression);
    assertTrue("Compiled right expression should be Literal but isnt", dyExpr2a.getRight() instanceof Literal);
    assertEquals("Operator between right (left and right) is incorrect", Expression.OP_GTEQ, dyExpr2a.getOperator());
    PrimaryExpression leftExpr2a = (PrimaryExpression) dyExpr2a.getLeft();
    assertEquals("Compiled left expression has incorrect number of tuples", 1, leftExpr2a.getTuples().size());
    assertEquals("Compiled left expression 'id' is incorrect", "budget", leftExpr2a.getId());
    Literal rightExpr2a = (Literal) dyExpr2a.getRight();
    assertTrue("Compiled right expression literal is of incorrect type " + rightExpr2a.getLiteral().getClass().getName(), rightExpr2a.getLiteral() instanceof Long);
    assertEquals("Compiled right expression literal has incorrect value", 50, ((Long) rightExpr2a.getLiteral()).longValue());
    // 2b : name == "Marketing"
    assertTrue("Compiled left expression should be PrimaryExpression but isnt", dyExpr2b.getLeft() instanceof PrimaryExpression);
    assertTrue("Compiled right expression should be Literal but isnt", dyExpr2b.getRight() instanceof Literal);
    assertEquals("Operator between right (left and right) is incorrect", Expression.OP_EQ, dyExpr2b.getOperator());
    PrimaryExpression leftExpr2b = (PrimaryExpression) dyExpr2b.getLeft();
    assertEquals("Compiled left expression has incorrect number of tuples", 1, leftExpr2b.getTuples().size());
    assertEquals("Compiled left expression 'id' is incorrect", "name", leftExpr2b.getId());
    Literal rightExpr2b = (Literal) dyExpr2b.getRight();
    assertTrue("Compiled right expression literal is of incorrect type", rightExpr2b.getLiteral() instanceof String);
    assertEquals("Compiled right expression literal has incorrect value", "Marketing", ((String) rightExpr2b.getLiteral()));
}

Example 4 with QueryCompilation

use of org.datanucleus.query.compiler.QueryCompilation in project tests by datanucleus.

the class JPQLCompilerTest method testFilterWithStringEqualsLiteral.

/**
 * Tests for "String = Literal" in filter.
 */
public void testFilterWithStringEqualsLiteral() {
    JavaQueryCompiler compiler = null;
    QueryCompilation compilation = null;
    try {
        compiler = new JPQLCompiler(nucCtx, nucCtx.getClassLoaderResolver(null), null, Project.class, null, "name = \"Kettle\"", null, null, null, null, null, null, null);
        compilation = compiler.compile(new HashMap(), null);
    } catch (NucleusException ne) {
        NucleusLogger.QUERY.error("Exception during compile", ne);
        fail("compilation of filter with valid field threw exception : " + ne.getMessage());
    }
    Expression expr = compilation.getExprFilter();
    assertTrue("Compiled expression should have been DyadicExpression but wasnt", expr instanceof DyadicExpression);
    DyadicExpression dyExpr = (DyadicExpression) expr;
    assertTrue("DyadicExpression should have left of PrimaryExpression but hasnt", dyExpr.getLeft() instanceof PrimaryExpression);
    assertEquals("Name of field upon which we invoke the method was wrong", "name", ((PrimaryExpression) dyExpr.getLeft()).getId());
    assertTrue("Name of invoked method was wrong", dyExpr.getOperator() == Expression.OP_EQ);
    assertTrue("DyadicExpression should right of Literal but hasnt", dyExpr.getRight() instanceof Literal);
    Literal paramLit = (Literal) dyExpr.getRight();
    assertEquals("Parameter to equals() has wrong value", "Kettle", paramLit.getLiteral());
}

Example 5 with QueryCompilation

use of org.datanucleus.query.compiler.QueryCompilation in project tests by datanucleus.

the class JPQLCompilerTest method testFilterUnaryMinus.

/**
 * Test for unary minus.
 */
public void testFilterUnaryMinus() {
    JavaQueryCompiler compiler = null;
    QueryCompilation compilation = null;
    try {
        compiler = new JPQLCompiler(nucCtx, nucCtx.getClassLoaderResolver(null), null, Project.class, null, "1 > -1", null, null, null, null, null, null, null);
        compilation = compiler.compile(new HashMap(), null);
    } catch (NucleusException ne) {
        NucleusLogger.QUERY.error("Exception during compile", ne);
        fail("compilation of filter with valid field threw exception : " + ne.getMessage());
    }
    // TODO Check the content of the compilation when it is compiled
    NucleusLogger.QUERY.info(">> compilation=" + compilation);
}