/**
* 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.ResultSet;
import java.sql.SQLDataException;
import java.sql.SQLException;
import java.util.List;
import java.util.Map;
import org.apache.camel.Component;
import org.apache.camel.Consumer;
import org.apache.camel.Processor;
import org.apache.camel.Producer;
import org.apache.camel.impl.DefaultPollingEndpoint;
import org.apache.camel.spi.UriEndpoint;
import org.apache.camel.spi.UriParam;
import org.apache.camel.spi.UriPath;
import org.apache.camel.util.UnsafeUriCharactersEncoder;
import org.springframework.jdbc.core.BeanPropertyRowMapper;
import org.springframework.jdbc.core.ColumnMapRowMapper;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.jdbc.core.RowMapperResultSetExtractor;
/**
* SQL Endpoint. Endpoint URI should contain valid SQL statement, but instead of
* question marks (that are parameter placeholders), sharp signs should be used.
* This is because in camel question mark has other meaning.
*/
@UriEndpoint(scheme = "sql", consumerClass = SqlConsumer.class)
public class SqlEndpoint extends DefaultPollingEndpoint {
private JdbcTemplate jdbcTemplate;
@UriPath
private String query;
@UriParam
private boolean batch;
@UriParam
private int maxMessagesPerPoll;
@UriParam
private SqlProcessingStrategy processingStrategy;
@UriParam
private SqlPrepareStatementStrategy prepareStatementStrategy;
@UriParam
private String onConsume;
@UriParam
private String onConsumeFailed;
@UriParam
private String onConsumeBatchComplete;
@UriParam
private boolean allowNamedParameters = true;
@UriParam
private boolean alwaysPopulateStatement;
@UriParam
private char separator = ',';
@UriParam
private SqlOutputType outputType = SqlOutputType.SelectList;
@UriParam
private String outputClass;
@UriParam
private int parametersCount;
@UriParam
private boolean noop;
public SqlEndpoint() {
}
public SqlEndpoint(String uri, Component component, JdbcTemplate jdbcTemplate, String query) {
super(uri, component);
this.jdbcTemplate = jdbcTemplate;
this.query = query;
}
public Consumer createConsumer(Processor processor) throws Exception {
SqlPrepareStatementStrategy prepareStrategy = prepareStatementStrategy != null ? prepareStatementStrategy : new DefaultSqlPrepareStatementStrategy(separator);
SqlProcessingStrategy proStrategy = processingStrategy != null ? processingStrategy : new DefaultSqlProcessingStrategy(prepareStrategy);
SqlConsumer consumer = new SqlConsumer(this, processor, jdbcTemplate, query, prepareStrategy, proStrategy);
consumer.setMaxMessagesPerPoll(getMaxMessagesPerPoll());
consumer.setOnConsume(getOnConsume());
consumer.setOnConsumeFailed(getOnConsumeFailed());
consumer.setOnConsumeBatchComplete(getOnConsumeBatchComplete());
configureConsumer(consumer);
return consumer;
}
public Producer createProducer() throws Exception {
SqlPrepareStatementStrategy prepareStrategy = prepareStatementStrategy != null ? prepareStatementStrategy : new DefaultSqlPrepareStatementStrategy(separator);
SqlProducer result = new SqlProducer(this, query, jdbcTemplate, prepareStrategy, batch, alwaysPopulateStatement);
result.setParametersCount(parametersCount);
return result;
}
public boolean isSingleton() {
return true;
}
public JdbcTemplate getJdbcTemplate() {
return jdbcTemplate;
}
public void setJdbcTemplate(JdbcTemplate jdbcTemplate) {
this.jdbcTemplate = jdbcTemplate;
}
public String getQuery() {
return query;
}
/**
* Sets the SQL query to perform
*/
public void setQuery(String query) {
this.query = query;
}
public boolean isBatch() {
return batch;
}
/**
* Enables or disables batch mode
*/
public void setBatch(boolean batch) {
this.batch = batch;
}
public int getMaxMessagesPerPoll() {
return maxMessagesPerPoll;
}
/**
* Sets the maximum number of messages to poll
*/
public void setMaxMessagesPerPoll(int maxMessagesPerPoll) {
this.maxMessagesPerPoll = maxMessagesPerPoll;
}
public SqlProcessingStrategy getProcessingStrategy() {
return processingStrategy;
}
/**
* Allows to plugin to use a custom org.apache.camel.component.sql.SqlProcessingStrategy to execute queries when the consumer has processed the rows/batch.
*/
public void setProcessingStrategy(SqlProcessingStrategy processingStrategy) {
this.processingStrategy = processingStrategy;
}
public SqlPrepareStatementStrategy getPrepareStatementStrategy() {
return prepareStatementStrategy;
}
/**
* Allows to plugin to use a custom org.apache.camel.component.sql.SqlPrepareStatementStrategy to control preparation of the query and prepared statement.
*/
public void setPrepareStatementStrategy(SqlPrepareStatementStrategy prepareStatementStrategy) {
this.prepareStatementStrategy = prepareStatementStrategy;
}
public String getOnConsume() {
return onConsume;
}
/**
* After processing each row then this query can be executed, if the Exchange was processed successfully, for example to mark the row as processed. The query can have parameter.
*/
public void setOnConsume(String onConsume) {
this.onConsume = onConsume;
}
public String getOnConsumeFailed() {
return onConsumeFailed;
}
/**
* After processing each row then this query can be executed, if the Exchange failed, for example to mark the row as failed. The query can have parameter.
*/
public void setOnConsumeFailed(String onConsumeFailed) {
this.onConsumeFailed = onConsumeFailed;
}
public String getOnConsumeBatchComplete() {
return onConsumeBatchComplete;
}
/**
* After processing the entire batch, this query can be executed to bulk update rows etc. The query cannot have parameters.
*/
public void setOnConsumeBatchComplete(String onConsumeBatchComplete) {
this.onConsumeBatchComplete = onConsumeBatchComplete;
}
public boolean isAllowNamedParameters() {
return allowNamedParameters;
}
/**
* Whether to allow using named parameters in the queries.
*/
public void setAllowNamedParameters(boolean allowNamedParameters) {
this.allowNamedParameters = allowNamedParameters;
}
public boolean isAlwaysPopulateStatement() {
return alwaysPopulateStatement;
}
/**
* If enabled then the populateStatement method from org.apache.camel.component.sql.SqlPrepareStatementStrategy is always invoked,
* also if there is no expected parameters to be prepared. When this is false then the populateStatement is only invoked if there
* is 1 or more expected parameters to be set; for example this avoids reading the message body/headers for SQL queries with no parameters.
*/
public void setAlwaysPopulateStatement(boolean alwaysPopulateStatement) {
this.alwaysPopulateStatement = alwaysPopulateStatement;
}
public char getSeparator() {
return separator;
}
/**
* The separator to use when parameter values is taken from message body (if the body is a String type), to be inserted at # placeholders.
* Notice if you use named parameters, then a Map type is used instead.
* <p/>
* The default value is ,
*/
public void setSeparator(char separator) {
this.separator = separator;
}
public SqlOutputType getOutputType() {
return outputType;
}
/**
* Make the output of consumer or producer to SelectList as List of Map, or SelectOne as single Java object in the following way:
* a) If the query has only single column, then that JDBC Column object is returned. (such as SELECT COUNT( * ) FROM PROJECT will return a Long object.
* b) If the query has more than one column, then it will return a Map of that result.
* c) If the outputClass is set, then it will convert the query result into an Java bean object by calling all the setters that match the column names.
* It will assume your class has a default constructor to create instance with.
* d) If the query resulted in more than one rows, it throws an non-unique result exception.
*/
public void setOutputType(SqlOutputType outputType) {
this.outputType = outputType;
}
public String getOutputClass() {
return outputClass;
}
/**
* Specify the full package and class name to use as conversion when outputType=SelectOne.
*/
public void setOutputClass(String outputClass) {
this.outputClass = outputClass;
}
public int getParametersCount() {
return parametersCount;
}
/**
* If set greater than zero, then Camel will use this count value of parameters to replace instead of querying via JDBC metadata API.
* This is useful if the JDBC vendor could not return correct parameters count, then user may override instead.
*/
public void setParametersCount(int parametersCount) {
this.parametersCount = parametersCount;
}
public boolean isNoop() {
return noop;
}
/**
* If set, will ignore the results of the SQL query and use the existing IN message as the OUT message for the continuation of processing
*/
public void setNoop(boolean noop) {
this.noop = noop;
}
@Override
protected String createEndpointUri() {
// Make sure it's properly encoded
return "sql:" + UnsafeUriCharactersEncoder.encode(query);
}
protected List<Map<String, Object>> queryForList(ResultSet rs) throws SQLException {
ColumnMapRowMapper rowMapper = new ColumnMapRowMapper();
RowMapperResultSetExtractor<Map<String, Object>> mapper = new RowMapperResultSetExtractor<Map<String, Object>>(rowMapper);
List<Map<String, Object>> data = mapper.extractData(rs);
return data;
}
@SuppressWarnings("unchecked")
protected Object queryForObject(ResultSet rs) throws SQLException {
Object result = null;
if (outputClass == null) {
RowMapper rowMapper = new ColumnMapRowMapper();
RowMapperResultSetExtractor<Map<String, Object>> mapper = new RowMapperResultSetExtractor<Map<String, Object>>(rowMapper);
List<Map<String, Object>> data = mapper.extractData(rs);
if (data.size() > 1) {
throw new SQLDataException("Query result not unique for outputType=SelectOne. Got " + data.size() + " count instead.");
} else if (data.size() == 1) {
// Set content depend on number of column from query result
Map<String, Object> row = data.get(0);
if (row.size() == 1) {
result = row.values().iterator().next();
} else {
result = row;
}
}
} else {
Class<?> outputClzz = getCamelContext().getClassResolver().resolveClass(outputClass);
RowMapper rowMapper = new BeanPropertyRowMapper(outputClzz);
RowMapperResultSetExtractor<?> mapper = new RowMapperResultSetExtractor(rowMapper);
List<?> data = mapper.extractData(rs);
if (data.size() > 1) {
throw new SQLDataException("Query result not unique for outputType=SelectOne. Got " + data.size() + " count instead.");
} else if (data.size() == 1) {
result = data.get(0);
}
}
// If data.size is zero, let result be null.
return result;
}
}