You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@camel.apache.org by da...@apache.org on 2013/08/20 17:18:24 UTC

[2/2] git commit: CAMEL-6091: Added parametersCount option to sql producer. Thanks to Zemian Deng for the patch.

CAMEL-6091: Added parametersCount option to sql producer. Thanks to Zemian Deng for the patch.

Conflicts:
	components/camel-sql/src/main/java/org/apache/camel/component/sql/SqlEndpoint.java
	components/camel-sql/src/main/java/org/apache/camel/component/sql/SqlProducer.java


Project: http://git-wip-us.apache.org/repos/asf/camel/repo
Commit: http://git-wip-us.apache.org/repos/asf/camel/commit/e0946841
Tree: http://git-wip-us.apache.org/repos/asf/camel/tree/e0946841
Diff: http://git-wip-us.apache.org/repos/asf/camel/diff/e0946841

Branch: refs/heads/camel-2.11.x
Commit: e09468419bb134ebd8ae4b1f3bac6f0a975dff3b
Parents: 7c031be
Author: Claus Ibsen <da...@apache.org>
Authored: Tue Aug 20 17:15:02 2013 +0200
Committer: Claus Ibsen <da...@apache.org>
Committed: Tue Aug 20 17:17:56 2013 +0200

----------------------------------------------------------------------
 .../apache/camel/component/sql/SqlEndpoint.java | 13 +++-
 .../apache/camel/component/sql/SqlProducer.java |  6 +-
 .../sql/SqlProducerParametersCountTest.java     | 82 ++++++++++++++++++++
 3 files changed, 99 insertions(+), 2 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/camel/blob/e0946841/components/camel-sql/src/main/java/org/apache/camel/component/sql/SqlEndpoint.java
----------------------------------------------------------------------
diff --git a/components/camel-sql/src/main/java/org/apache/camel/component/sql/SqlEndpoint.java b/components/camel-sql/src/main/java/org/apache/camel/component/sql/SqlEndpoint.java
index 70c70e4..6747df7 100644
--- a/components/camel-sql/src/main/java/org/apache/camel/component/sql/SqlEndpoint.java
+++ b/components/camel-sql/src/main/java/org/apache/camel/component/sql/SqlEndpoint.java
@@ -42,6 +42,7 @@ public class SqlEndpoint extends DefaultPollingEndpoint {
     private boolean allowNamedParameters = true;
     private boolean alwaysPopulateStatement;
     private char separator = ',';
+    private int parametersCount;
 
     public SqlEndpoint() {
     }
@@ -66,7 +67,9 @@ public class SqlEndpoint extends DefaultPollingEndpoint {
 
     public Producer createProducer() throws Exception {
         SqlPrepareStatementStrategy prepareStrategy = prepareStatementStrategy != null ? prepareStatementStrategy : new DefaultSqlPrepareStatementStrategy(separator);
-        return new SqlProducer(this, query, jdbcTemplate, prepareStrategy, batch, alwaysPopulateStatement);
+        SqlProducer result = new SqlProducer(this, query, jdbcTemplate, prepareStrategy, batch, alwaysPopulateStatement);
+        result.setParametersCount(parametersCount);
+        return result;
     }
 
     public boolean isSingleton() {
@@ -169,6 +172,14 @@ public class SqlEndpoint extends DefaultPollingEndpoint {
         this.separator = separator;
     }
 
+    public int getParametersCount() {
+        return parametersCount;
+    }
+
+    public void setParametersCount(int parametersCount) {
+        this.parametersCount = parametersCount;
+    }
+
     @Override
     protected String createEndpointUri() {
         // Make sure it's properly encoded

http://git-wip-us.apache.org/repos/asf/camel/blob/e0946841/components/camel-sql/src/main/java/org/apache/camel/component/sql/SqlProducer.java
----------------------------------------------------------------------
diff --git a/components/camel-sql/src/main/java/org/apache/camel/component/sql/SqlProducer.java b/components/camel-sql/src/main/java/org/apache/camel/component/sql/SqlProducer.java
index 97ff150..04d211c 100644
--- a/components/camel-sql/src/main/java/org/apache/camel/component/sql/SqlProducer.java
+++ b/components/camel-sql/src/main/java/org/apache/camel/component/sql/SqlProducer.java
@@ -35,6 +35,7 @@ public class SqlProducer extends DefaultProducer {
     private boolean batch;
     private boolean alwaysPopulateStatement;
     private SqlPrepareStatementStrategy sqlPrepareStatementStrategy;
+    private int parametersCount;
 
     public SqlProducer(SqlEndpoint endpoint, String query, JdbcTemplate jdbcTemplate, SqlPrepareStatementStrategy sqlPrepareStatementStrategy,
                        boolean batch, boolean alwaysPopulateStatement) {
@@ -59,7 +60,7 @@ public class SqlProducer extends DefaultProducer {
 
         jdbcTemplate.execute(preparedQuery, new PreparedStatementCallback<Map<?, ?>>() {
             public Map<?, ?> doInPreparedStatement(PreparedStatement ps) throws SQLException {
-                int expected = ps.getParameterMetaData().getParameterCount();
+                int expected = parametersCount > 0 ? parametersCount : ps.getParameterMetaData().getParameterCount();
 
                 // only populate if really needed
                 if (alwaysPopulateStatement || expected > 0) {
@@ -106,4 +107,7 @@ public class SqlProducer extends DefaultProducer {
         });
     }
 
+    public void setParametersCount(int parametersCount) {
+        this.parametersCount = parametersCount;
+    }
 }

http://git-wip-us.apache.org/repos/asf/camel/blob/e0946841/components/camel-sql/src/test/java/org/apache/camel/component/sql/SqlProducerParametersCountTest.java
----------------------------------------------------------------------
diff --git a/components/camel-sql/src/test/java/org/apache/camel/component/sql/SqlProducerParametersCountTest.java b/components/camel-sql/src/test/java/org/apache/camel/component/sql/SqlProducerParametersCountTest.java
new file mode 100644
index 0000000..a04b0af
--- /dev/null
+++ b/components/camel-sql/src/test/java/org/apache/camel/component/sql/SqlProducerParametersCountTest.java
@@ -0,0 +1,82 @@
+/**
+ * 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.util.List;
+import java.util.Map;
+
+import org.apache.camel.builder.RouteBuilder;
+import org.apache.camel.component.mock.MockEndpoint;
+import org.apache.camel.test.junit4.CamelTestSupport;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.springframework.jdbc.datasource.embedded.EmbeddedDatabase;
+import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseBuilder;
+import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseType;
+
+/**
+ * @version
+ */
+public class SqlProducerParametersCountTest extends CamelTestSupport {
+
+    private EmbeddedDatabase db;
+
+    @Before
+    public void setUp() throws Exception {
+        db = new EmbeddedDatabaseBuilder()
+                .setType(EmbeddedDatabaseType.DERBY).addScript("sql/createAndPopulateDatabase.sql").build();
+
+        super.setUp();
+    }
+
+    @After
+    public void tearDown() throws Exception {
+        super.tearDown();
+
+        db.shutdown();
+    }
+
+    @Test
+    public void testExplicitParametersCount() throws Exception {
+        MockEndpoint mock = getMockEndpoint("mock:result");
+        mock.expectedMessageCount(1);
+
+        template.sendBodyAndHeader("direct:start", null, "lic", "ASF");
+
+        List<?> received = assertIsInstanceOf(List.class, mock.getReceivedExchanges().get(0).getIn().getBody());
+        assertEquals(2, received.size());
+        Map<?, ?> row = assertIsInstanceOf(Map.class, received.get(0));
+        assertEquals("Camel", row.get("PROJECT"));
+
+        row = assertIsInstanceOf(Map.class, received.get(1));
+        assertEquals("AMQ", row.get("PROJECT"));
+    }
+
+    @Override
+    protected RouteBuilder createRouteBuilder() throws Exception {
+        return new RouteBuilder() {
+            public void configure() {
+                getContext().getComponent("sql", SqlComponent.class).setDataSource(db);
+
+                from("direct:start")
+                        .to("sql:select * from projects where license = :#lic order by id?parametersCount=1")
+                        .to("mock:result");
+            }
+        };
+    }
+}