Search in sources :

Example 1 with IfNode

use of org.seasar.doma.internal.jdbc.sql.node.IfNode in project doma by domaframework.

the class NodePreparedSqlBuilderTest method testAndNode.

@Test
public void testAndNode() {
    SelectClauseNode select = new SelectClauseNode("select");
    select.appendNode(OtherNode.of(" * "));
    FromClauseNode from = new FromClauseNode("from");
    from.appendNode(WhitespaceNode.of(" "));
    from.appendNode(new WordNode("aaa"));
    from.appendNode(WhitespaceNode.of(" "));
    WhereClauseNode where = new WhereClauseNode("where");
    IfNode ifNode1 = new IfNode(location, "true", "/*if true*/");
    ifNode1.appendNode(WhitespaceNode.of(" "));
    ifNode1.appendNode(new WordNode("bbb"));
    ifNode1.appendNode(OtherNode.of(" = "));
    ifNode1.appendNode(new WordNode("ccc"));
    EndNode endNode1 = new EndNode("/*end*/");
    IfBlockNode ifBlock1 = new IfBlockNode();
    ifBlock1.setIfNode(ifNode1);
    ifBlock1.setEndNode(endNode1);
    where.appendNode(ifBlock1);
    IfNode ifNode2 = new IfNode(location, "true", "/*if true*/");
    ifNode2.appendNode(WhitespaceNode.of(" "));
    LogicalOperatorNode and = new LogicalOperatorNode("and");
    ifNode2.appendNode(and);
    and.appendNode(WhitespaceNode.of(" "));
    and.appendNode(new WordNode("ddd"));
    and.appendNode(OtherNode.of(" = "));
    and.appendNode(new WordNode("eee"));
    EndNode endNode2 = new EndNode("/*end*/");
    IfBlockNode ifBlock2 = new IfBlockNode();
    ifBlock2.setIfNode(ifNode2);
    ifBlock2.setEndNode(endNode2);
    where.appendNode(ifBlock2);
    SelectStatementNode statement = new SelectStatementNode();
    statement.setSelectClauseNode(select);
    statement.setFromClauseNode(from);
    statement.setWhereClauseNode(where);
    AnonymousNode root = new AnonymousNode();
    root.appendNode(statement);
    ExpressionEvaluator evaluator = new ExpressionEvaluator();
    NodePreparedSqlBuilder builder = new NodePreparedSqlBuilder(config, SqlKind.SELECT, "dummyPath", evaluator, SqlLogType.FORMATTED);
    PreparedSql sql = builder.build(statement, Function.identity());
    assertEquals("select * from aaa where bbb = ccc and ddd = eee", sql.getRawSql());
}
Also used : FromClauseNode(org.seasar.doma.internal.jdbc.sql.node.FromClauseNode) SelectStatementNode(org.seasar.doma.internal.jdbc.sql.node.SelectStatementNode) PreparedSql(org.seasar.doma.jdbc.PreparedSql) WordNode(org.seasar.doma.internal.jdbc.sql.node.WordNode) SelectClauseNode(org.seasar.doma.internal.jdbc.sql.node.SelectClauseNode) IfNode(org.seasar.doma.internal.jdbc.sql.node.IfNode) ExpressionEvaluator(org.seasar.doma.internal.expr.ExpressionEvaluator) AnonymousNode(org.seasar.doma.internal.jdbc.sql.node.AnonymousNode) EndNode(org.seasar.doma.internal.jdbc.sql.node.EndNode) LogicalOperatorNode(org.seasar.doma.internal.jdbc.sql.node.LogicalOperatorNode) IfBlockNode(org.seasar.doma.internal.jdbc.sql.node.IfBlockNode) WhereClauseNode(org.seasar.doma.internal.jdbc.sql.node.WhereClauseNode) Test(org.junit.jupiter.api.Test)

Example 2 with IfNode

use of org.seasar.doma.internal.jdbc.sql.node.IfNode in project doma by domaframework.

the class NodePreparedSqlBuilderTest method testIfNode_false.

@Test
public void testIfNode_false() {
    SelectClauseNode select = new SelectClauseNode("select");
    select.appendNode(OtherNode.of(" * "));
    FromClauseNode from = new FromClauseNode("from");
    from.appendNode(WhitespaceNode.of(" "));
    from.appendNode(new WordNode("aaa"));
    from.appendNode(WhitespaceNode.of(" "));
    WhereClauseNode where = new WhereClauseNode("where");
    where.appendNode(WhitespaceNode.of(" "));
    IfNode ifNode = new IfNode(location, "false", "/*if false*/");
    ifNode.appendNode(new WordNode("bbb"));
    ifNode.appendNode(OtherNode.of(" = "));
    ifNode.appendNode(new WordNode("ccc"));
    IfBlockNode ifBlockNode = new IfBlockNode();
    ifBlockNode.setIfNode(ifNode);
    ifBlockNode.setEndNode(new EndNode("/*end*/"));
    where.appendNode(ifBlockNode);
    SelectStatementNode statement = new SelectStatementNode();
    statement.setSelectClauseNode(select);
    statement.setFromClauseNode(from);
    statement.setWhereClauseNode(where);
    AnonymousNode root = new AnonymousNode();
    root.appendNode(statement);
    ExpressionEvaluator evaluator = new ExpressionEvaluator();
    NodePreparedSqlBuilder builder = new NodePreparedSqlBuilder(config, SqlKind.SELECT, "dummyPath", evaluator, SqlLogType.FORMATTED);
    PreparedSql sql = builder.build(statement, Function.identity());
    assertEquals("select * from aaa", sql.getRawSql());
}
Also used : AnonymousNode(org.seasar.doma.internal.jdbc.sql.node.AnonymousNode) FromClauseNode(org.seasar.doma.internal.jdbc.sql.node.FromClauseNode) SelectStatementNode(org.seasar.doma.internal.jdbc.sql.node.SelectStatementNode) PreparedSql(org.seasar.doma.jdbc.PreparedSql) WordNode(org.seasar.doma.internal.jdbc.sql.node.WordNode) EndNode(org.seasar.doma.internal.jdbc.sql.node.EndNode) IfBlockNode(org.seasar.doma.internal.jdbc.sql.node.IfBlockNode) SelectClauseNode(org.seasar.doma.internal.jdbc.sql.node.SelectClauseNode) IfNode(org.seasar.doma.internal.jdbc.sql.node.IfNode) ExpressionEvaluator(org.seasar.doma.internal.expr.ExpressionEvaluator) WhereClauseNode(org.seasar.doma.internal.jdbc.sql.node.WhereClauseNode) Test(org.junit.jupiter.api.Test)

Example 3 with IfNode

use of org.seasar.doma.internal.jdbc.sql.node.IfNode in project doma by domaframework.

the class NodePreparedSqlBuilderTest method testElseNode.

@Test
public void testElseNode() {
    SelectClauseNode select = new SelectClauseNode("select");
    select.appendNode(OtherNode.of(" * "));
    FromClauseNode from = new FromClauseNode("from");
    from.appendNode(WhitespaceNode.of(" "));
    from.appendNode(new WordNode("aaa"));
    from.appendNode(WhitespaceNode.of(" "));
    WhereClauseNode where = new WhereClauseNode("where");
    where.appendNode(WhitespaceNode.of(" "));
    IfNode ifNode = new IfNode(location, "false", "/*if false*/");
    ifNode.appendNode(new WordNode("bbb"));
    ifNode.appendNode(OtherNode.of(" = "));
    ifNode.appendNode(new WordNode("ccc"));
    ElseNode elseNode = new ElseNode("/*else*/");
    elseNode.appendNode(new WordNode("ddd"));
    elseNode.appendNode(OtherNode.of(" = "));
    elseNode.appendNode(new WordNode("eee"));
    EndNode endNode = new EndNode("/*end*/");
    IfBlockNode ifBlock = new IfBlockNode();
    ifBlock.setIfNode(ifNode);
    ifBlock.setElseNode(elseNode);
    ifBlock.setEndNode(endNode);
    where.appendNode(ifBlock);
    SelectStatementNode statement = new SelectStatementNode();
    statement.setSelectClauseNode(select);
    statement.setFromClauseNode(from);
    statement.setWhereClauseNode(where);
    AnonymousNode root = new AnonymousNode();
    root.appendNode(statement);
    ExpressionEvaluator evaluator = new ExpressionEvaluator();
    NodePreparedSqlBuilder builder = new NodePreparedSqlBuilder(config, SqlKind.SELECT, "dummyPath", evaluator, SqlLogType.FORMATTED);
    PreparedSql sql = builder.build(statement, Function.identity());
    assertEquals("select * from aaa where ddd = eee", sql.getRawSql());
}
Also used : FromClauseNode(org.seasar.doma.internal.jdbc.sql.node.FromClauseNode) SelectStatementNode(org.seasar.doma.internal.jdbc.sql.node.SelectStatementNode) PreparedSql(org.seasar.doma.jdbc.PreparedSql) WordNode(org.seasar.doma.internal.jdbc.sql.node.WordNode) SelectClauseNode(org.seasar.doma.internal.jdbc.sql.node.SelectClauseNode) IfNode(org.seasar.doma.internal.jdbc.sql.node.IfNode) ExpressionEvaluator(org.seasar.doma.internal.expr.ExpressionEvaluator) AnonymousNode(org.seasar.doma.internal.jdbc.sql.node.AnonymousNode) EndNode(org.seasar.doma.internal.jdbc.sql.node.EndNode) ElseNode(org.seasar.doma.internal.jdbc.sql.node.ElseNode) IfBlockNode(org.seasar.doma.internal.jdbc.sql.node.IfBlockNode) WhereClauseNode(org.seasar.doma.internal.jdbc.sql.node.WhereClauseNode) Test(org.junit.jupiter.api.Test)

Example 4 with IfNode

use of org.seasar.doma.internal.jdbc.sql.node.IfNode in project doma by domaframework.

the class NodePreparedSqlBuilder method handleIfNode.

protected boolean handleIfNode(IfBlockNode node, Context p) {
    IfNode ifNode = node.getIfNode();
    SqlLocation location = ifNode.getLocation();
    String expression = ifNode.getExpression();
    EvaluationResult ifResult = p.evaluate(location, expression);
    if (ifResult.getBooleanValue()) {
        ifNode.accept(this, p);
        return true;
    }
    return false;
}
Also used : SqlLocation(org.seasar.doma.internal.jdbc.sql.node.SqlLocation) IfNode(org.seasar.doma.internal.jdbc.sql.node.IfNode) EvaluationResult(org.seasar.doma.internal.expr.EvaluationResult)

Example 5 with IfNode

use of org.seasar.doma.internal.jdbc.sql.node.IfNode in project doma by domaframework.

the class SqlParser method parseIfBlockComment.

protected void parseIfBlockComment() {
    IfBlockNode ifBlockNode = new IfBlockNode();
    appendNode(ifBlockNode);
    push(ifBlockNode);
    String expression = tokenType.extract(token);
    IfNode ifNode = new IfNode(getLocation(), expression, token);
    ifBlockNode.setIfNode(ifNode);
    push(ifNode);
}
Also used : IfBlockNode(org.seasar.doma.internal.jdbc.sql.node.IfBlockNode) IfNode(org.seasar.doma.internal.jdbc.sql.node.IfNode)

Aggregations

IfNode (org.seasar.doma.internal.jdbc.sql.node.IfNode)7 IfBlockNode (org.seasar.doma.internal.jdbc.sql.node.IfBlockNode)6 Test (org.junit.jupiter.api.Test)5 ExpressionEvaluator (org.seasar.doma.internal.expr.ExpressionEvaluator)5 AnonymousNode (org.seasar.doma.internal.jdbc.sql.node.AnonymousNode)5 EndNode (org.seasar.doma.internal.jdbc.sql.node.EndNode)5 FromClauseNode (org.seasar.doma.internal.jdbc.sql.node.FromClauseNode)5 SelectClauseNode (org.seasar.doma.internal.jdbc.sql.node.SelectClauseNode)5 SelectStatementNode (org.seasar.doma.internal.jdbc.sql.node.SelectStatementNode)5 WhereClauseNode (org.seasar.doma.internal.jdbc.sql.node.WhereClauseNode)5 WordNode (org.seasar.doma.internal.jdbc.sql.node.WordNode)5 PreparedSql (org.seasar.doma.jdbc.PreparedSql)5 LogicalOperatorNode (org.seasar.doma.internal.jdbc.sql.node.LogicalOperatorNode)2 EvaluationResult (org.seasar.doma.internal.expr.EvaluationResult)1 ElseNode (org.seasar.doma.internal.jdbc.sql.node.ElseNode)1 SqlLocation (org.seasar.doma.internal.jdbc.sql.node.SqlLocation)1