org.frameworkset.persitent.template.SQLTemplateImpl2 Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bboss-persistent Show documentation
bboss is a j2ee framework include aop/ioc,mvc,persistent,taglib,rpc,event ,bean-xml serializable and so on.http://www.bbossgroups.com
The newest version!
/**
 *  Copyright 2008 biaoping.yin
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.  
 */
package org.frameworkset.persitent.template;

import bboss.org.apache.velocity.Template;
import bboss.org.apache.velocity.context.Context;
import bboss.org.apache.velocity.exception.*;
import bboss.org.apache.velocity.runtime.parser.ParseException;
import bboss.org.apache.velocity.runtime.parser.node.ASTText;
import bboss.org.apache.velocity.runtime.parser.node.Node;
import bboss.org.apache.velocity.runtime.parser.node.SimpleNode;
import bboss.org.apache.velocity.runtime.resource.ResourceManager;
import com.frameworkset.velocity.BBossVelocityUtil;
import org.frameworkset.persitent.util.SQLInfo;
import org.frameworkset.soa.BBossStringReader;

import java.io.*;
import java.util.List;
import java.util.Map;

/**
 *  SQLTemplate.java
 *  Description: sql语句模板
 *  bboss workgroup 
 *  Copyright (c) 2009 
 * 
 * @Date 2012-12-4 下午7:31:55
 * @author biaoping.yin
 * @version 1.0
 */
public class SQLTemplateImpl2 extends Template 
{
  

    private SQLInfo sqlinfo;
    
   private String sql;   
   private boolean processed;


    /** Default constructor */
    public SQLTemplateImpl2(SQLInfo sqlinfo)
    {
        super();
        scopeName = "sqltemplate";
        this.sqlinfo = sqlinfo;   
        this.sql = this.sqlinfo.getSql();
        this.setName(sqlinfo.getSqlname());
        setType(ResourceManager.RESOURCE_SQL);
    }
 
    /**
     * 根据解析出的语法结构确定sql语句是否是velocity模板
     * 如果不是则重置sqlinfo的istpl属性，相关的缓存就可以使用sqlname来作为key
     * 同时避免每次都对token进行拼接，提升系统性能
     */
    private void rechecksqlistpl()
    {
    	if(data != null)
    	{
    		Node[] childrens = ((SimpleNode)data).getChildren();
    		if(childrens != null && childrens.length > 0)
    		{
    			boolean switchcase = true;
    			for(Node node:childrens)
    			{
    				if(!(node instanceof ASTText))
    				{
    					switchcase = false;
    					break;
    				}
    			}
    			if(switchcase)
    			{
    				this.sqlinfo.setIstpl(false);
    				sqlinfo.setMultiparser(false);
    			}
    		}
    		
    	}
    }
    /**
     *  gets the named resource as a stream, parses and inits
     *
     * @return true if successful
     * @throws ResourceNotFoundException if template not found
     *          from any available source.
     * @throws ParseErrorException if template cannot be parsed due
     *          to syntax (or other) error.
     * @throws IOException problem reading input stream
     */
    public boolean process()
        throws ResourceNotFoundException, ParseErrorException
    {
    	if(processed)
    		return true;
    	synchronized(this)
    	{
    		if(processed)
        		return true;
    		
	        data = null;
	        
	        errorCondition = null;
	        Reader is = null;
	
	        /*
	         *  first, try to get the stream from the loader
	         */
	        try
	        {
	        	is = new BBossStringReader(sql);
	        }
	        catch( ResourceNotFoundException rnfe )
	        {
	            /*
	             *  remember and re-throw
	             */
	
	            errorCondition = rnfe;
	            processed = true;
	            throw rnfe;
	        }
	
	        /*
	         *  if that worked, lets protect in case a loader impl
	         *  forgets to throw a proper exception
	         */
	
	        if (is != null)
	        {
	            /*
	             *  now parse the template
	             */
                BufferedReader br = null;
	            try
	            {
	                br = new BufferedReader( is);
	                data = rsvc.parse( br, this);
	                initDocument();
	                processed = true;
	                try {
						rechecksqlistpl();
					} catch (Exception e) {
						
					}
	                return true;
	            }
	            
	            catch ( ParseException pex )
	            {
	                /*
	                 *  remember the error and convert
	                 */
	                errorCondition =  new ParseErrorException(pex, name);
	                processed = true;
	                throw errorCondition;
	            }
	            catch ( TemplateInitException pex )
	            {
	                errorCondition = new ParseErrorException( pex, name);
	                processed = true;
	                throw errorCondition;
	            }
	            /**
	             * pass through runtime exceptions
	             */
	            catch( RuntimeException e )
	            {
                    errorCondition = new VelocityException("Exception thrown processing Template "
                            +getName(), e, rsvc.getLogContext().getStackTrace());
	                processed = true;
	                throw errorCondition;
	            }
	            finally
	            {
	            	processed = true;
	                /*
	                 *  Make sure to close the inputstream when we are done.
	                 */
	                try
	                {
	                    is.close();
	                }
	                catch(IOException e)
	                {
	                                       
	                }

                    /*
                     *  Make sure to close the inputstream when we are done.
                     */
                    try
                    {
                        if(br != null)
                            br.close();
                    }
                    catch(IOException e)
                    {
                       
                    }
	                
	            }
	        }
	        else
	        {
                /*
                 *  is == null, therefore we have some kind of file issue
                 */
                errorCondition = new ResourceNotFoundException("Unknown resource error for resource " + name, null, rsvc.getLogContext().getStackTrace() );
	            processed = true;
	            throw errorCondition;
	        }
    	}
    }

      
    
    public String evaluate( Map variablevalues)
            throws ResourceNotFoundException, ParseErrorException, MethodInvocationException
    {
    	StringWriter writer = new StringWriter();    	
        merge(BBossVelocityUtil.buildVelocityContext(variablevalues), writer, null);
        return writer.toString();
    }


    /**
     * The AST node structure is merged with the
     * context to produce the final output.
     *
     *  @param context Context with data elements accessed by template
     *  @param writer output writer for rendered template
     *  @param macroLibraries a list of template files containing macros to be used when merging
     *  @throws ResourceNotFoundException if template not found
     *          from any available source.
     *  @throws ParseErrorException if template cannot be parsed due
     *          to syntax (or other) error.
     *  @throws MethodInvocationException When a method on a referenced object in the context could not invoked.
     *  @since 1.6
     */
    public void merge( Context context, Writer writer, List macroLibraries)
            throws ResourceNotFoundException, ParseErrorException, MethodInvocationException
    {       
        process();
        super.merge(context,writer,macroLibraries);
    }
 
  
}