HSQLUtil.java

package railo.runtime.db;

import java.io.ByteArrayInputStream;
import java.util.Enumeration;
import java.util.HashSet;
import java.util.Set;
import java.util.Vector;

import railo.runtime.sql.old.ParseException;
import railo.runtime.sql.old.ZExp;
import railo.runtime.sql.old.ZExpression;
import railo.runtime.sql.old.ZFromItem;
import railo.runtime.sql.old.ZQuery;
import railo.runtime.sql.old.ZStatement;
import railo.runtime.sql.old.ZqlParser;


/**
 * utilities for sql statements
 */
public final class HSQLUtil {
	
	private ZqlParser parser;
	private String sql;
	private boolean isUnion;

	/**or of the class
	 * construct
	 * @param sql SQl Statement as String
	 */
	public HSQLUtil(String sql) {
		this.sql=SQLPrettyfier.prettyfie(sql,true);//sqlToZQL(sql,true);
		parser = new ZqlParser(new ByteArrayInputStream(this.sql.getBytes()));
	}
	
	/* *
	 * transalte SQL syntax to a ZQL combatible form
	 * @param sql sql to transalte
	 * @param changePlaceHolder
	 * @return translated sql
	 * /
	private static String sqlToZQL(String sql,boolean changePlaceHolder) {
		sql=sql.trim();
		char c=' ';//,last=' ';
		int len=sql.length();
		boolean insideString=false;
		StringBuffer sb=new StringBuffer(len);
		
		
		
		for(int i=0;i<len;i++) {
		    c=sql.charAt(i);
		    if(insideString) {
		        if(c=='\'') {
		            if(i+1>=len || sql.charAt(i+1)!='\'')insideString=false;
		        }
		    }
		    else {
		        if(c=='\'')insideString=true;
		        else if(changePlaceHolder && c=='?') {
		            sb.append("QUESTION_MARK_SIGN");
				    //last=c;
		            continue;
		        }
		        else if(c=='a'|| c=='A') {
		            if(
		                    (i!=0 && isWhiteSpace(sql.charAt(i-1)))
		                    &&
		                    (i+1<len && (sql.charAt(i+1)=='s' || sql.charAt(i+1)=='S'))
		                    &&
		                    (i+2<len && isWhiteSpace(sql.charAt(i+2)))
		            )	{
		                i++;
		    		    //last=c;
		                continue;
		            }
		        }
		        else if(c=='*') {
		        	
		        }
		    }
		    //last=c;
		    sb.append(c);
		}
		
		if(c!=';')sb.append(';');
		
		return sb.toString();
		
	}*/
	
	/*private static boolean isWhiteSpace(char c) {
	    return (c==' ' || c=='\t' || c=='\b' || c=='\r' || c=='\n');
	}*/

	/**
	 * @return return the sql state inside
	 */
	public String getSQL() {
		return sql;
	}
	
	/**
	 * return all invoked tables by a sql statement
	 * @return invoked tables in a ArrayList
	 * @throws ParseException
	 */
	public Set<String> getInvokedTables() throws ParseException {
		
	  	
	  	// Read all SQL statements from input
	  	ZStatement st;
	  	Set<String> tables=new HashSet<String>();
	  	
		while((st = parser.readStatement()) != null) {
			this.sql=st.toString();
		  	if(st instanceof ZQuery) { // An SQL query: query the DB
		      getInvokedTables((ZQuery)st,tables);	          
		    }
		    break;
		}
		return tables;
	}
	
	private void getInvokedTables(ZQuery query, Set<String> tablesNames) {
		//print.out("qry:"+query.getSet());
		Vector tables=query.getFrom();
		Enumeration e = tables.elements();
		
		// from
			while(e.hasMoreElements()) {
				ZFromItem fromItem=(ZFromItem) e.nextElement();
				tablesNames.add(fromItem.getFullName());
			}
		// where
			ZExp where = query.getWhere();
			if(where instanceof ZExpression) {
				parseZExpression((ZExpression) where, tablesNames);
			}
		// set
			ZExpression set = query.getSet();
			if(set!=null){
				isUnion=true;
				ZExp op = set.getOperand(0);
				if(op instanceof ZQuery) getInvokedTables((ZQuery)op, tablesNames);
			}
	}
	


	public boolean isUnion() {
		return isUnion;
	}
	
	private void parseZExpression(ZExpression expression, Set tablesNames) {
		Vector operands = expression.getOperands();
		Enumeration e = operands.elements();
		while(e.hasMoreElements()) {
			Object el=e.nextElement();
			if(el instanceof ZExpression)parseZExpression((ZExpression)el,tablesNames);
			else if(el instanceof ZQuery)getInvokedTables((ZQuery)el,tablesNames);
		}
	}
}