Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.landawn.abacus.util.SQLParser Maven / Gradle / Ivy
/*
* Copyright (C) 2015 HaiYang Li
*
* 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 com.landawn.abacus.util;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;
/**
*
* @author Haiyang Li
* @since 0.8
*/
public final class SQLParser {
private static final String KEEP_COMMENTS = "-- Keep comments";
private static final char TAB = '\t';
private static final char ENTER = '\n';
private static final char ENTER_2 = '\r';
private static final Set seperators = N.newHashSet();
static {
seperators.add(TAB);
seperators.add(ENTER);
seperators.add(ENTER_2);
seperators.add(' ');
seperators.add('?');
seperators.add(',');
seperators.add('~');
seperators.add('!');
seperators.add('@');
seperators.add('^');
seperators.add('#');
seperators.add("!!");
seperators.add(';');
seperators.add('(');
seperators.add(')');
seperators.add('=');
seperators.add("==");
seperators.add(":=");
seperators.add("^=");
seperators.add("~=");
seperators.add("+=");
seperators.add("-=");
seperators.add("*=");
seperators.add("/=");
seperators.add("%=");
seperators.add("&=");
seperators.add("|=");
seperators.add("!=");
seperators.add("!<");
seperators.add("!>");
seperators.add('>');
seperators.add(">>");
seperators.add(">=");
seperators.add("@>");
seperators.add("&>");
seperators.add(">^");
seperators.add('<');
seperators.add("<<");
seperators.add("<=");
seperators.add("<@");
seperators.add("&<");
seperators.add("<^");
seperators.add('+');
seperators.add('-');
seperators.add('%');
seperators.add('/');
seperators.add('*');
seperators.add('&');
seperators.add("&&");
seperators.add('|');
seperators.add("||");
seperators.add("|/");
seperators.add("||/");
seperators.add('^');
seperators.add('~');
seperators.add('!');
seperators.add("->");
seperators.add('#');
seperators.add("##");
seperators.add("@@");
seperators.add("@-@");
seperators.add("@@@");
seperators.add("->>");
seperators.add("<->");
seperators.add("<=>");
seperators.add(">>=");
seperators.add("<<=");
seperators.add("<<|");
seperators.add("|>>");
seperators.add("&<|");
seperators.add("|&>");
seperators.add("|>>");
seperators.add("(+)");
seperators.add("?#");
seperators.add("?-");
seperators.add("?-");
seperators.add("?|");
seperators.add("?-|");
seperators.add("?||");
seperators.add("~*");
seperators.add("!~");
seperators.add("!~*");
seperators.add("^-=");
seperators.add("|*=");
}
private static final Map compositeWords = new ObjectPool<>(64);
static {
compositeWords.put(WD.LEFT_JOIN, new String[] { "LEFT", "JOIN" });
compositeWords.put(WD.RIGHT_JOIN, new String[] { "RIGHT", "JOIN" });
compositeWords.put(WD.FULL_JOIN, new String[] { "FULL", "JOIN" });
compositeWords.put(WD.CROSS_JOIN, new String[] { "CROSS", "JOIN" });
compositeWords.put(WD.INNER_JOIN, new String[] { "INNER", "JOIN" });
compositeWords.put(WD.NATURAL_JOIN, new String[] { "NATURAL", "JOIN" });
compositeWords.put(WD.INNER_JOIN, new String[] { "INNER", "JOIN" });
compositeWords.put(WD.GROUP_BY, new String[] { "GROUP", "BY" });
compositeWords.put(WD.ORDER_BY, new String[] { "ORDER", "BY" });
compositeWords.put(WD.FOR_UPDATE, new String[] { "FOR", "UPDATE" });
compositeWords.put(WD.FETCH_FIRST, new String[] { "FETCH", "FIRST" });
compositeWords.put(WD.FETCH_NEXT, new String[] { "FETCH", "NEXT" });
compositeWords.put(WD.ROWS_ONLY, new String[] { "ROWS", "ONLY" });
compositeWords.put(WD.UNION_ALL, new String[] { "UNION", "ALL" });
compositeWords.put(WD.IS_NOT, new String[] { "IS", "NOT" });
compositeWords.put(WD.IS_NULL, new String[] { "IS", "NULL" });
compositeWords.put(WD.IS_NOT_NULL, new String[] { "IS", "NOT", "NULL" });
compositeWords.put(WD.IS_EMPTY, new String[] { "IS", "EMPTY" });
compositeWords.put(WD.IS_NOT_EMPTY, new String[] { "IS", "NOT", "EMPTY" });
compositeWords.put(WD.IS_BLANK, new String[] { "IS", "BLANK" });
compositeWords.put(WD.IS_NOT_BLANK, new String[] { "IS", "NOT", "BLANK" });
compositeWords.put(WD.NOT_IN, new String[] { "NOT", "IN" });
compositeWords.put(WD.NOT_EXISTS, new String[] { "NOT", "EXISTS" });
final List list = new ArrayList<>(compositeWords.keySet());
for (String e : list) {
e = e.toLowerCase();
if (!compositeWords.containsKey(e)) {
compositeWords.put(e, Splitter.with(WD.SPACE).trimResults().splitToArray(e));
}
e = e.toUpperCase();
if (!compositeWords.containsKey(e)) {
compositeWords.put(e, Splitter.with(WD.SPACE).trimResults().splitToArray(e));
}
}
}
private SQLParser() {
}
/**
*
* @param sql
* @return
*/
public static List parse(final String sql) {
final int sqlLength = sql.length();
final StringBuilder sb = Objectory.createStringBuilder();
final List words = new ArrayList<>();
String temp = "";
char quoteChar = 0;
int keepComments = -1;
for (int index = 0; index < sqlLength; index++) {
// TODO [performance improvement]. will it improve performance if
// change to char array?
// char c = sqlCharArray[charIndex];
char c = sql.charAt(index);
if (quoteChar != 0) {
// is it in a quoted identifier?
sb.append(c);
// end in quote.
if (c == quoteChar && sql.charAt(index - 1) != '\\') {
words.add(sb.toString());
sb.setLength(0);
quoteChar = 0;
}
} else if (c == '-' && index < sqlLength - 1 && sql.charAt(index + 1) == '-') {
if (sb.length() > 0) {
words.add(sb.toString());
sb.setLength(0);
}
// if (keepComments == -1) {
// keepComments = Strings.startsWithIgnoreCase(sql, "-- Keep comments") ? 1 : 0;
// }
//
// if (keepComments == 1) {
// sb.append(c);
//
// while (++index < sqlLength) {
// c = sql.charAt(index);
// sb.append(c);
//
// if (c == ENTER || c == ENTER_2) {
// final String tmp = sb.toString();
//
// if (!Strings.startsWithIgnoreCase(tmp, "-- Keep comments")) {
// words.add(sb.toString());
// }
//
// sb.setLength(0);
//
// break;
// }
// }
// } else {
// while (++index < sqlLength) {
// c = sql.charAt(index);
//
// if (c == ENTER || c == ENTER_2) {
// break;
// }
// }
// }
while (++index < sqlLength) {
c = sql.charAt(index);
if (c == ENTER || c == ENTER_2) {
break;
}
}
} else if (c == '/' && index < sqlLength - 1 && sql.charAt(index + 1) == '*') {
if (sb.length() > 0) {
words.add(sb.toString());
sb.setLength(0);
}
if (keepComments == -1) {
keepComments = Strings.startsWithIgnoreCase(sql, KEEP_COMMENTS) ? 1 : 0;
}
if (keepComments == 1) {
sb.append(c);
while (++index < sqlLength) {
c = sql.charAt(index);
sb.append(c);
if (c == '*' && index < sqlLength - 1 && sql.charAt(index + 1) == '/') {
sb.append(sql.charAt(++index));
words.add(sb.toString());
sb.setLength(0);
break;
}
}
} else {
while (++index < sqlLength) {
c = sql.charAt(index);
if (c == '*' && index < sqlLength - 1 && sql.charAt(index + 1) == '/') {
index++;
break;
}
}
}
} else if (isSeperator(sql, sqlLength, index, c)) {
if (sb.length() > 0) {
words.add(sb.toString());
sb.setLength(0);
}
if ((index < (sqlLength - 2)) && seperators.contains(temp = sql.substring(index, index + 3))) {
words.add(temp);
index += 2;
} else if ((index < (sqlLength - 1)) && seperators.contains(temp = sql.substring(index, index + 2))) {
words.add(temp);
index += 1;
} else if (c == WD._SPACE || c == TAB || c == ENTER || c == ENTER_2) {
if ((words.size() > 0) && !words.get(words.size() - 1).equals(WD.SPACE)) {
words.add(WD.SPACE);
}
} else {
words.add(String.valueOf(c));
}
} else {
sb.append(c);
if ((c == WD._QUOTATION_S) || (c == WD._QUOTATION_D)) {
quoteChar = c;
}
}
}
if (sb.length() > 0) {
words.add(sb.toString());
sb.setLength(0);
}
Objectory.recycle(sb);
return words;
}
/**
*
* @param sql
* @param word
* @param fromIndex
* @param caseSensitive
* @return
*/
public static int indexWord(final String sql, final String word, final int fromIndex, final boolean caseSensitive) {
String[] subWords = compositeWords.get(word);
if (subWords == null) {
subWords = Splitter.with(WD.SPACE).trimResults().splitToArray(word);
compositeWords.put(word, subWords);
}
if ((subWords == null) || (subWords.length <= 1)) {
int result = N.INDEX_NOT_FOUND;
final StringBuilder sb = Objectory.createStringBuilder();
final int sqlLength = sql.length();
String temp = "";
char quoteChar = 0;
for (int index = fromIndex; index < sqlLength; index++) {
final char c = sql.charAt(index);
// is it in a quoted identifier?
if (quoteChar != 0) {
sb.append(c);
// end in quote.
if (c == quoteChar) {
temp = sb.toString();
if (word.equals(temp) || (!caseSensitive && word.equalsIgnoreCase(temp))) {
result = index - word.length() + 1;
break;
}
sb.setLength(0);
quoteChar = 0;
}
} else if (isSeperator(sql, sqlLength, index, c)) {
if (sb.length() > 0) {
temp = sb.toString();
if (word.equals(temp) || (!caseSensitive && word.equalsIgnoreCase(temp))) {
result = index - word.length();
break;
}
sb.setLength(0);
} else if (c == WD._SPACE || c == TAB || c == ENTER || c == ENTER_2) {
// skip white char
continue;
}
if ((index < (sqlLength - 2)) && seperators.contains(temp = sql.substring(index, index + 3))) {
if (word.equals(temp) || (!caseSensitive && word.equalsIgnoreCase(temp))) {
result = index;
break;
}
index += 2;
} else if ((index < (sqlLength - 1)) && seperators.contains(temp = sql.substring(index, index + 2))) {
if (word.equals(temp) || (!caseSensitive && word.equalsIgnoreCase(temp))) {
result = index;
break;
}
index += 1;
} else if (word.equals(String.valueOf(c)) || (!caseSensitive && word.equalsIgnoreCase(String.valueOf(c)))) {
result = index;
break;
}
} else {
sb.append(c);
if ((c == WD._QUOTATION_S) || (c == WD._QUOTATION_D)) {
quoteChar = c;
}
}
}
if (result < 0 && sb.length() > 0) {
temp = sb.toString();
if (word.equals(temp) || (!caseSensitive && word.equalsIgnoreCase(temp))) {
result = sqlLength - word.length();
}
}
Objectory.recycle(sb);
return result;
} else {
int result = indexWord(sql, subWords[0], fromIndex, caseSensitive);
if (result >= 0) {
int tmpIndex = result + subWords[0].length();
String nextWord = null;
for (int i = 1; i < subWords.length; i++) {
nextWord = nextWord(sql, tmpIndex);
if (Strings.isNotEmpty(nextWord) && (nextWord.equals(subWords[i]) || (!caseSensitive && nextWord.equalsIgnoreCase(subWords[i])))) {
tmpIndex += (subWords[i].length() + 1);
} else {
result = -1;
break;
}
}
}
return result;
}
}
/**
*
* @param sql
* @param fromIndex
* @return
*/
public static String nextWord(final String sql, final int fromIndex) {
final int sqlLength = sql.length();
final StringBuilder sb = Objectory.createStringBuilder();
String temp = "";
char quoteChar = 0;
for (int index = fromIndex; index < sqlLength; index++) {
final char c = sql.charAt(index);
// is it in a quoted identifier?
if (quoteChar != 0) {
sb.append(c);
// end in quote.
if (c == quoteChar) {
break;
}
} else if (isSeperator(sql, sqlLength, index, c)) {
if (sb.length() > 0) {
break;
} else if (c == WD._SPACE || c == TAB || c == ENTER || c == ENTER_2) {
// skip white char
continue;
}
if (((index < (sqlLength - 2)) && seperators.contains(temp = sql.substring(index, index + 3)))
|| ((index < (sqlLength - 1)) && seperators.contains(temp = sql.substring(index, index + 2)))) {
sb.append(temp);
} else {
sb.append(c);
}
break;
} else {
sb.append(c);
if ((c == WD._QUOTATION_S) || (c == WD._QUOTATION_D)) {
quoteChar = c;
}
}
}
final String st = (sb.length() == 0) ? "" : sb.toString();
Objectory.recycle(sb);
return st;
}
/**
*
* @param seperator
*/
public static void registerSeperator(final char seperator) {
N.checkArgPositive(seperator, "seperator");
seperators.add(seperator);
}
/**
*
* @param seperator
*/
public static void registerSeperator(final String seperator) {
N.checkArgNotNull(seperator, "seperator");
seperators.add(seperator);
if (seperator.length() == 1) {
seperators.add(seperator.charAt(0));
}
}
/**
* Checks if is seperator.
*
* @param str
* @param len
* @param index
* @param ch
* @return true, if is seperator
*/
public static boolean isSeperator(final String str, final int len, final int index, final char ch) {
// for Ibatis/Mybatis
if (ch == '#' && index < len - 1 && str.charAt(index + 1) == '{') {
return false;
}
return seperators.contains(ch);
}
/**
* Checks if is function name.
*
* @param words
* @param len
* @param index
* @return true, if is function name
*/
public static boolean isFunctionName(final List words, final int len, final int index) {
// return (i < len - 1 && words.get(i + 1).charAt(0) == WD._PARENTHESES_L)
// || (i < len - 2 && WD.SPACE.equals(words.get(i + 1)) && words.get(i + 2).charAt(0) == WD._PARENTHESES_L);
if ((index < len - 1 && words.get(index + 1).charAt(0) == WD._PARENTHESES_L)) {
return true;
}
for (int i = index + 1; i < len; i++) {
if (words.get(i).charAt(0) == WD._PARENTHESES_L) {
return true;
} else if (!WD.SPACE.equals(words.get(i))) {
return false;
}
}
return false;
}
}
