日期:2014-05-16 浏览次数:20603 次
import java.util.ArrayList;
import java.util.List;
/**
* 此类封装NamedParameterSql
*
* @author zl
*
*/
public class ParsedSql {
private String originalSql;
//参数名
private List<String> paramNames = new ArrayList<String>();
//参数在sql中对应的位置
private List<int[]> paramIndexs = new ArrayList<int[]>();
//统计参数个数(不包含重复)
private int namedParamCount;
//统计sql中?的个数
private int unnamedParamCount;
private int totalParamCount;
public ParsedSql(String originalSql){
this.originalSql = originalSql;
}
public List<String> getParamNames() {
return paramNames;
}
public void addParamNames(String paramName,int startIndex,int endIndex) {
paramNames.add(paramName);
paramIndexs.add(new int[]{startIndex,endIndex});
}
public int[] getParamIndexs(int position) {
return paramIndexs.get(position);
}
public String getOriginalSql() {
return originalSql;
}
public int getNamedParamCount() {
return namedParamCount;
}
public void setNamedParamCount(int namedParamCount) {
this.namedParamCount = namedParamCount;
}
public int getUnnamedParamCount() {
return unnamedParamCount;
}
public void setUnnamedParamCount(int unnamedParamCount) {
this.unnamedParamCount = unnamedParamCount;
}
public int getTotalParamCount() {
return totalParamCount;
}
public void setTotalParamCount(int totalParamCount) {
this.totalParamCount = totalParamCount;
}
public String toString(){
return this.originalSql;
}
}
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
/**
* 带参数sql处理工具类
*
* @author zl
*
*/
public class NamedParameterUtils {
//定义特殊字符(参考spring jdbc N多)
private static final char[] PARAMETER_SEPARATORS =
new char[] {'"', '\'', ':', '&', ',', ';', '(', ')', '|', '=', '+', '-', '*', '%', '/', '\\', '<', '>', '^'};
/**
* 对带参数sql的统计式封装,便于后续肢解拼装(恐怖啊。。。。。)
* @param originalSql
* @return
*/
public static ParsedSql parserSqlStatement(String originalSql) {
ParsedSql parsedSql = new ParsedSql(originalSql);
Set<String> paramNames = new HashSet<String>();
char[] sqlchars = originalSql.toCharArray();
int namedParamCount = 0;
int unNamedParamCount = 0;
int totalParamCount = 0;
int i = 0;
while(i<sqlchars.length){
char statement = sqlchars[i];
if(statement==':'||statement=='&'){
int j = i+1;
while(j<sqlchars.length&&!isSeparatorsChar(sqlchars[j])){
j++;
}
if(j-i>1){
String paramName = originalSql.substring(i+1, j);
if(!paramNames.contains(paramName)){
paramNames.add(paramName);
namedParamCount++;
}
parsedSql.addParamNames(paramName, i, j);
totalParamCount++;
}
i=j-1;
}else if(statement=='?'){
unNamedParamCount++;
totalParamCount++;
}
i++;
}
parsedSql.setNamedParamCount(namedParamCount);
parsedSql.setUnnamedParamCount(unNamedParamCount);
parsedSql.setTotalParamCount(totalParamCount);
return parsedSql;
}
/**
* 获得不带参数的sql,即替换参数为?
* @param parsedSql
* @param params
* @return
*/
public static String substituteNamedParams(ParsedSql parsedSql,Map<String,Object> params){
String original =parsedSql.getOriginalSql();
StringBuffer actual = new StringBuffer("");
int lastIndex = 0;
List<String> paramNames = parsedSql.getParamNames();
for(int i=0;i<paramNames.size();i++){
int[] indexs = parsedSql.getParamIndexs(i);
int startIndex = indexs[0];
int endIndex = indexs[1];
String paramName = paramNames.get(i);
actual.append(original.substring(lastIndex, startIndex));
if(params!=null&¶ms.containsKey(paramName)){
// if(){}