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.
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.apache.camel.component.sql;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.Arrays;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.NoSuchElementException;
import java.util.StringJoiner;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.apache.camel.Exchange;
import org.apache.camel.RuntimeExchangeException;
import org.apache.camel.support.ObjectHelper;
import org.apache.camel.util.StringQuoteHelper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.jdbc.core.ArgumentPreparedStatementSetter;
import org.springframework.util.CompositeIterator;
/**
* Default {@link SqlPrepareStatementStrategy} that supports named query parameters as well index based.
*/
public class DefaultSqlPrepareStatementStrategy implements SqlPrepareStatementStrategy {
private static final Logger LOG = LoggerFactory.getLogger(DefaultSqlPrepareStatementStrategy.class);
private static final Pattern REPLACE_IN_PATTERN
= Pattern.compile("\\:\\?in\\:(\\w+|\\$\\{[^\\}]+\\}|\\$simple\\{[^\\}]+\\})", Pattern.MULTILINE);
private static final Pattern REPLACE_PATTERN
= Pattern.compile("\\:\\?\\w+|\\:\\?\\$\\{[^\\}]+\\}|\\:\\?\\$simple\\{[^\\}]+\\}", Pattern.MULTILINE);
private static final Pattern NAME_PATTERN = Pattern.compile(
"\\:\\?((in\\:(\\w+|\\$\\{[^\\}]+\\}|\\$simple\\{[^\\}]+\\}))|(\\w+|\\$\\{[^\\}]+\\}|\\$simple\\{[^\\}]+\\}))",
Pattern.MULTILINE);
private final char separator;
public DefaultSqlPrepareStatementStrategy() {
this(',');
}
public DefaultSqlPrepareStatementStrategy(char separator) {
this.separator = separator;
}
@Override
public String prepareQuery(String query, boolean allowNamedParameters, final Exchange exchange) throws SQLException {
String answer;
if (allowNamedParameters && hasNamedParameters(query)) {
if (exchange != null) {
// replace all :?in:word with a number of placeholders for how many values are expected in the IN values
Matcher matcher = REPLACE_IN_PATTERN.matcher(query);
while (matcher.find()) {
String found = matcher.group(1);
Object parameter = lookupParameter(found, exchange, exchange.getIn().getBody());
if (parameter != null) {
Iterator it = createInParameterIterator(parameter);
StringJoiner replaceBuilder = new StringJoiner(",");
while (it.hasNext()) {
it.next();
replaceBuilder.add("\\?");
}
String replace = replaceBuilder.toString();
String foundEscaped = found.replace("$", "\\$").replace("{", "\\{").replace("}", "\\}");
Matcher paramMatcher = Pattern.compile("\\:\\?in\\:" + foundEscaped, Pattern.MULTILINE).matcher(query);
query = paramMatcher.replaceAll(replace);
}
}
}
// replace all :?word and :?${foo} with just ?
answer = replaceParams(query);
} else {
answer = query;
}
LOG.trace("Prepared query: {}", answer);
return answer;
}
private String replaceParams(String query) {
// nested parameters are not replaced properly just by the REPLACE_PATTERN
// for example ":?${array[${index}]}"
query = replaceBracketedParams(query);
return REPLACE_PATTERN.matcher(query).replaceAll("\\?");
}
private String replaceBracketedParams(String query) {
while (query.contains(":?${")) {
int i = query.indexOf(":?${");
int j = findClosingBracket(query, i + 3);
if (j == -1) {
throw new IllegalArgumentException("String doesn't have equal opening and closing brackets: " + query);
}
query = query.substring(0, i) + "?" + query.substring(j + 1);
}
return query;
}
/**
* Finds closing bracket in text for named parameter.
*
* @param text
* @param openPosition position of the opening bracket
*
* @return index of corresponding closing bracket, or -1, if none was found
*/
private static int findClosingBracket(String text, int openPosition) {
if (text.charAt(openPosition) != '{') {
throw new IllegalArgumentException("Character at specified position is not an open bracket");
}
int remainingClosingBrackets = 0;
for (int i = openPosition; i < text.length(); i++) {
if (text.charAt(i) == '{') {
remainingClosingBrackets++;
} else if (text.charAt(i) == '}') {
remainingClosingBrackets--;
}
if (remainingClosingBrackets == 0) {
return i;
}
}
return -1;
}
@Override
public Iterator createPopulateIterator(
final String query, final String preparedQuery, final int expectedParams, final Exchange exchange,
final Object value)
throws SQLException {
if (hasNamedParameters(query)) {
// create an iterator that returns the value in the named order
return new PopulateIterator(query, exchange, value);
} else {
// if only 1 parameter and the body is a String then use body as is
if (expectedParams == 1 && value instanceof String) {
return Collections.singletonList(value).iterator();
} else {
// is the body a String
if (value instanceof String) {
// if the body is a String then honor quotes etc.
String[] tokens = StringQuoteHelper.splitSafeQuote((String) value, separator, true);
List list = Arrays.asList(tokens);
return list.iterator();
} else {
// just use a regular iterator
return exchange.getContext().getTypeConverter().convertTo(Iterator.class, value);
}
}
}
}
@Override
public void populateStatement(PreparedStatement ps, Iterator iterator, int expectedParams) throws SQLException {
if (expectedParams <= 0) {
return;
}
final Object[] args = new Object[expectedParams];
int i = 0;
int argNumber = 1;
while (iterator != null && iterator.hasNext()) {
Object value = iterator.next();
// special for SQL IN where we need to set dynamic number of values
if (value instanceof CompositeIterator) {
Iterator it = (Iterator) value;
while (it.hasNext()) {
Object val = it.next();
LOG.trace("Setting parameter #{} with value: {}", argNumber, val);
if (argNumber <= expectedParams) {
args[i] = val;
}
argNumber++;
i++;
}
} else {
LOG.trace("Setting parameter #{} with value: {}", argNumber, value);
if (argNumber <= expectedParams) {
args[i] = value;
}
argNumber++;
i++;
}
}
if (argNumber - 1 != expectedParams) {
throw new SQLException("Number of parameters mismatch. Expected: " + expectedParams + ", was: " + (argNumber - 1));
}
// use argument setter as it deals with various JDBC drivers setObject vs setLong/setInteger/setString etc.
ArgumentPreparedStatementSetter setter = new ArgumentPreparedStatementSetter(args);
setter.setValues(ps);
}
protected boolean hasNamedParameters(String query) {
NamedQueryParser parser = new NamedQueryParser(query);
return parser.next() != null;
}
private static final class NamedQueryParser {
private final String query;
private final Matcher matcher;
private NamedQueryParser(String query) {
this.query = query;
this.matcher = NAME_PATTERN.matcher(query);
}
public String next() {
if (matcher.find()) {
String param = matcher.group(1);
int openingBrackets = 0;
int closingBrackets = 0;
for (int i = 0; i < param.length(); i++) {
if (param.charAt(i) == '{') {
openingBrackets++;
}
if (param.charAt(i) == '}') {
closingBrackets++;
}
}
if (openingBrackets != closingBrackets) {
// nested parameters are not found properly by the NAME_PATTERN
// for example param ":?${array[?${index}]}"
// is detected as "${array[?${index}"
// we have to find correct closing bracket manually
String querySubstring = query.substring(matcher.start());
int i = querySubstring.indexOf('{');
int j = findClosingBracket(querySubstring, i);
param = "$" + querySubstring.substring(i, j + 1);
}
return param;
}
return null;
}
}
protected static Object lookupParameter(String nextParam, Exchange exchange, Object body) {
Map bodyMap = safeMap(exchange.getContext().getTypeConverter().tryConvertTo(Map.class, body));
Map headersMap = safeMap(exchange.getIn().getHeaders());
Object answer = null;
if ((nextParam.startsWith("$simple{") || nextParam.startsWith("${")) && nextParam.endsWith("}")) {
answer = exchange.getContext().resolveLanguage("simple").createExpression(nextParam).evaluate(exchange,
Object.class);
} else if (bodyMap.containsKey(nextParam)) {
answer = bodyMap.get(nextParam);
} else if (headersMap.containsKey(nextParam)) {
answer = headersMap.get(nextParam);
}
return answer;
}
protected static boolean hasParameter(String nextParam, Exchange exchange, Object body) {
Map bodyMap = safeMap(exchange.getContext().getTypeConverter().tryConvertTo(Map.class, body));
Map headersMap = safeMap(exchange.getIn().getHeaders());
if ((nextParam.startsWith("$simple{") || nextParam.startsWith("${")) && nextParam.endsWith("}")) {
return true;
} else if (bodyMap.containsKey(nextParam)) {
return true;
} else if (headersMap.containsKey(nextParam)) {
return true;
}
return false;
}
private static Map safeMap(Map map) {
return (map == null || map.isEmpty()) ? Collections.emptyMap() : map;
}
@SuppressWarnings("unchecked")
protected static CompositeIterator createInParameterIterator(Object value) {
Iterator it;
// if the body is a String then honor quotes etc.
if (value instanceof String) {
String[] tokens = StringQuoteHelper.splitSafeQuote((String) value, ',', true);
List list = Arrays.asList(tokens);
it = list.iterator();
} else {
it = ObjectHelper.createIterator(value, null);
}
CompositeIterator ci = new CompositeIterator();
ci.add(it);
return ci;
}
private static final class PopulateIterator implements Iterator
