/* * Hibernate, Relational Persistence for Idiomatic Java * * License: GNU Lesser General Public License (LGPL), version 2.1 or later. * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>. */ package org.hibernate.dialect.function; import java.util.ArrayList; import java.util.List; import org.hibernate.engine.spi.SessionFactoryImplementor; import org.hibernate.internal.CoreMessageLogger; import org.jboss.logging.Logger; /** * Delegate for handling function "templates". * * @author Steve Ebersole */ public class TemplateRenderer { private static final CoreMessageLogger LOG = Logger.getMessageLogger( CoreMessageLogger.class, TemplateRenderer.class.getName() ); private final String template; private final String[] chunks; private final int[] paramIndexes; /** * Constructs a template renderer * * @param template The template */ public TemplateRenderer(String template) { this.template = template; final List<String> chunkList = new ArrayList<String>(); final List<Integer> paramList = new ArrayList<Integer>(); final StringBuilder chunk = new StringBuilder( 10 ); final StringBuilder index = new StringBuilder( 2 ); int i = 0; final int len = template.length(); while ( i < len ) { char c = template.charAt( i ); if ( c == '?' ) { chunkList.add( chunk.toString() ); chunk.delete( 0, chunk.length() ); while ( ++i < template.length() ) { c = template.charAt( i ); if ( Character.isDigit( c ) ) { index.append( c ); } else { chunk.append( c ); break; } } paramList.add( Integer.valueOf( index.toString() ) ); index.delete( 0, index.length() ); } else { chunk.append( c ); } i++; } if ( chunk.length() > 0 ) { chunkList.add( chunk.toString() ); } chunks = chunkList.toArray( new String[chunkList.size()] ); paramIndexes = new int[paramList.size()]; for ( i = 0; i < paramIndexes.length; ++i ) { paramIndexes[i] = paramList.get( i ); } } public String getTemplate() { return template; } public int getAnticipatedNumberOfArguments() { return paramIndexes.length; } /** * The rendering code. * * @param args The arguments to inject into the template * @param factory The SessionFactory * * @return The rendered template with replacements */ @SuppressWarnings({ "UnusedDeclaration" }) public String render(List args, SessionFactoryImplementor factory) { final int numberOfArguments = args.size(); if ( getAnticipatedNumberOfArguments() > 0 && numberOfArguments != getAnticipatedNumberOfArguments() ) { LOG.missingArguments( getAnticipatedNumberOfArguments(), numberOfArguments ); } final StringBuilder buf = new StringBuilder(); for ( int i = 0; i < chunks.length; ++i ) { if ( i < paramIndexes.length ) { final int index = paramIndexes[i] - 1; final Object arg = index < numberOfArguments ? args.get( index ) : null; if ( arg != null ) { buf.append( chunks[i] ).append( arg ); } } else { buf.append( chunks[i] ); } } return buf.toString(); } }