Search in sources :

Example 1 with TrimSpecification

use of org.hibernate.sql.ast.tree.expression.TrimSpecification in project hibernate-orm by hibernate.

the class AnsiTrimEmulationFunctionTest method render.

private String render(Dialect dialect, TrimFunction function, TrimSpec trimSpec, char trimCharacter, String trimSource) {
    SessionFactoryImplementor factory = Mockito.mock(SessionFactoryImplementor.class);
    JdbcServices jdbcServices = Mockito.mock(JdbcServices.class);
    Mockito.doReturn(jdbcServices).when(factory).getJdbcServices();
    Mockito.doReturn(dialect).when(jdbcServices).getDialect();
    StandardSqlAstTranslator<JdbcOperation> walker = new StandardSqlAstTranslator<>(factory, null);
    List<SqlAstNode> sqlAstArguments = new ArrayList<>();
    sqlAstArguments.add(new TrimSpecification(trimSpec));
    sqlAstArguments.add(new QueryLiteral<>(trimCharacter, new BasicTypeImpl<>(CharacterJavaType.INSTANCE, CharJdbcType.INSTANCE)));
    sqlAstArguments.add(new SelfRenderingExpression() {

        @Override
        public void renderToSql(SqlAppender sqlAppender, SqlAstTranslator<?> walker, SessionFactoryImplementor sessionFactory) {
            sqlAppender.appendSql(trimSource);
        }

        @Override
        public JdbcMappingContainer getExpressionType() {
            return null;
        }
    });
    function.render(walker, sqlAstArguments, walker);
    return walker.getSql();
}
Also used : SqlAppender(org.hibernate.sql.ast.spi.SqlAppender) SessionFactoryImplementor(org.hibernate.engine.spi.SessionFactoryImplementor) BasicTypeImpl(org.hibernate.type.internal.BasicTypeImpl) JdbcOperation(org.hibernate.sql.exec.spi.JdbcOperation) ArrayList(java.util.ArrayList) TrimSpecification(org.hibernate.sql.ast.tree.expression.TrimSpecification) JdbcServices(org.hibernate.engine.jdbc.spi.JdbcServices) SqlAstNode(org.hibernate.sql.ast.tree.SqlAstNode) JdbcMappingContainer(org.hibernate.metamodel.mapping.JdbcMappingContainer) SelfRenderingExpression(org.hibernate.sql.ast.tree.expression.SelfRenderingExpression) StandardSqlAstTranslator(org.hibernate.sql.ast.spi.StandardSqlAstTranslator)

Example 2 with TrimSpecification

use of org.hibernate.sql.ast.tree.expression.TrimSpecification in project hibernate-orm by hibernate.

the class TrimFunction method render.

@Override
public void render(SqlAppender sqlAppender, List<? extends SqlAstNode> sqlAstArguments, SqlAstTranslator<?> walker) {
    final TrimSpec specification = ((TrimSpecification) sqlAstArguments.get(0)).getSpecification();
    final Object trimCharacter = ((Literal) sqlAstArguments.get(1)).getLiteralValue();
    final Expression sourceExpr = (Expression) sqlAstArguments.get(2);
    String trim = dialect.trimPattern(specification, (char) trimCharacter);
    new PatternRenderer(trim).render(sqlAppender, Collections.singletonList(sourceExpr), walker);
}
Also used : TrimSpec(org.hibernate.query.sqm.TrimSpec) Expression(org.hibernate.sql.ast.tree.expression.Expression) Literal(org.hibernate.sql.ast.tree.expression.Literal) TrimSpecification(org.hibernate.sql.ast.tree.expression.TrimSpecification) PatternRenderer(org.hibernate.query.sqm.produce.function.internal.PatternRenderer)

Aggregations

TrimSpecification (org.hibernate.sql.ast.tree.expression.TrimSpecification)2 ArrayList (java.util.ArrayList)1 JdbcServices (org.hibernate.engine.jdbc.spi.JdbcServices)1 SessionFactoryImplementor (org.hibernate.engine.spi.SessionFactoryImplementor)1 JdbcMappingContainer (org.hibernate.metamodel.mapping.JdbcMappingContainer)1 TrimSpec (org.hibernate.query.sqm.TrimSpec)1 PatternRenderer (org.hibernate.query.sqm.produce.function.internal.PatternRenderer)1 SqlAppender (org.hibernate.sql.ast.spi.SqlAppender)1 StandardSqlAstTranslator (org.hibernate.sql.ast.spi.StandardSqlAstTranslator)1 SqlAstNode (org.hibernate.sql.ast.tree.SqlAstNode)1 Expression (org.hibernate.sql.ast.tree.expression.Expression)1 Literal (org.hibernate.sql.ast.tree.expression.Literal)1 SelfRenderingExpression (org.hibernate.sql.ast.tree.expression.SelfRenderingExpression)1 JdbcOperation (org.hibernate.sql.exec.spi.JdbcOperation)1 BasicTypeImpl (org.hibernate.type.internal.BasicTypeImpl)1