This is an automated email from the ASF dual-hosted git repository.
dschneider pushed a commit to branch feature/GEODE-3781
in repository https://gitbox.apache.org/repos/asf/geode.git
commit 0b077931777aff56805c7f05f71ca906bde45ed8
Author: Darrel Schneider <dschnei...@pivotal.io>
AuthorDate: Wed Nov 1 12:16:36 2017 -0700
Added JDBCLoader and integration test
---
.../geode/connectors/jdbc/JDBCAsyncWriter.java | 5 +-
.../apache/geode/connectors/jdbc/JDBCLoader.java | 47 +++++++
.../connectors/jdbc/JDBCLoaderIntegrationTest.java | 143 +++++++++++++++++++++
3 files changed, 191 insertions(+), 4 deletions(-)
diff --git
a/geode-connectors/src/main/java/org/apache/geode/connectors/jdbc/JDBCAsyncWriter.java
b/geode-connectors/src/main/java/org/apache/geode/connectors/jdbc/JDBCAsyncWriter.java
index 12688d6..05306bb 100644
---
a/geode-connectors/src/main/java/org/apache/geode/connectors/jdbc/JDBCAsyncWriter.java
+++
b/geode-connectors/src/main/java/org/apache/geode/connectors/jdbc/JDBCAsyncWriter.java
@@ -42,10 +42,7 @@ public class JDBCAsyncWriter implements AsyncEventListener {
private JDBCManager manager;
@Override
- public void close() {
- // TODO Auto-generated method stub
-
- }
+ public void close() {}
@Override
public boolean processEvents(List<AsyncEvent> events) {
diff --git
a/geode-connectors/src/main/java/org/apache/geode/connectors/jdbc/JDBCLoader.java
b/geode-connectors/src/main/java/org/apache/geode/connectors/jdbc/JDBCLoader.java
new file mode 100644
index 0000000..5219576
--- /dev/null
+++
b/geode-connectors/src/main/java/org/apache/geode/connectors/jdbc/JDBCLoader.java
@@ -0,0 +1,47 @@
+/*
+ * 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.geode.connectors.jdbc;
+
+import org.apache.geode.cache.LoaderHelper;
+import org.apache.geode.connectors.jdbc.internal.JDBCConfiguration;
+import org.apache.geode.connectors.jdbc.internal.JDBCManager;
+
+import java.util.Properties;
+
+import org.apache.geode.cache.CacheLoader;
+import org.apache.geode.cache.CacheLoaderException;
+
+/*
+ * This class provides loading from a data source using JDBC.
+ *
+ * @since Geode 1.4
+ */
+public class JDBCLoader<K, V> implements CacheLoader<K, V> {
+ private JDBCManager manager;
+
+ @Override
+ public void close() {}
+
+ @SuppressWarnings("unchecked")
+ @Override
+ public V load(LoaderHelper<K, V> helper) throws CacheLoaderException {
+ return (V) this.manager.read(helper.getRegion(), helper.getKey());
+ }
+
+ public void init(Properties props) {
+ JDBCConfiguration config = new JDBCConfiguration(props);
+ this.manager = new JDBCManager(config);
+ };
+}
diff --git
a/geode-connectors/src/test/java/org/apache/geode/connectors/jdbc/JDBCLoaderIntegrationTest.java
b/geode-connectors/src/test/java/org/apache/geode/connectors/jdbc/JDBCLoaderIntegrationTest.java
new file mode 100644
index 0000000..b57375b
--- /dev/null
+++
b/geode-connectors/src/test/java/org/apache/geode/connectors/jdbc/JDBCLoaderIntegrationTest.java
@@ -0,0 +1,143 @@
+/*
+ * 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.geode.connectors.jdbc;
+
+import static org.apache.geode.distributed.ConfigurationProperties.MCAST_PORT;
+import static org.assertj.core.api.Assertions.*;
+
+import java.sql.*;
+import java.util.Properties;
+
+import org.junit.*;
+import org.junit.experimental.categories.Category;
+import org.apache.geode.cache.Cache;
+import org.apache.geode.cache.CacheFactory;
+import org.apache.geode.cache.Region;
+import org.apache.geode.cache.RegionFactory;
+import org.apache.geode.cache.RegionShortcut;
+import org.apache.geode.internal.cache.GemFireCacheImpl;
+import org.apache.geode.pdx.PdxInstance;
+import org.apache.geode.test.junit.categories.IntegrationTest;
+
+@Category(IntegrationTest.class)
+public class JDBCLoaderIntegrationTest {
+ private JDBCLoader<String, String> jdbcLoader;
+ private Cache cache;
+ private Connection conn;
+ private Statement stmt;
+ private String dbName = "DerbyDB";
+ private String regionTableName = "employees";
+ private String driver = "org.apache.derby.jdbc.EmbeddedDriver";
+ private String connectionURL = "jdbc:derby:memory:" + dbName +
";create=true";
+
+ @Before
+ public void setup() throws Exception {
+ try {
+ cache = CacheFactory.getAnyInstance();
+ } catch (Exception e) {
+ // ignore
+ }
+ if (null == cache) {
+ cache = (GemFireCacheImpl) new
CacheFactory().setPdxReadSerialized(false).set(MCAST_PORT, "0")
+ .create();
+ }
+ setupDB();
+ }
+
+ @After
+ public void tearDown() throws Exception {
+ if (cache != null && !cache.isClosed()) {
+ cache.close();
+ cache = null;
+ }
+ closeDB();
+ }
+
+ public void setupDB() throws Exception {
+ Class.forName(driver);
+ conn = DriverManager.getConnection(connectionURL);
+ stmt = conn.createStatement();
+ stmt.execute("Create Table " + regionTableName
+ + " (id varchar(10) primary key not null, name varchar(10), age int)");
+ }
+
+ public void closeDB() throws Exception {
+ if (stmt == null) {
+ stmt = conn.createStatement();
+ }
+ stmt.execute("Drop table " + regionTableName);
+ stmt.close();
+
+ if (conn != null) {
+ conn.close();
+ }
+ }
+
+ private Properties getRequiredProperties() {
+ Properties props = new Properties();
+ props.setProperty("driver", this.driver);
+ props.setProperty("url", this.connectionURL);
+ return props;
+ }
+
+ private Region createRegionWithJDBCLoader(String regionName, Properties
props) {
+ this.jdbcLoader = new JDBCLoader<>();
+ this.jdbcLoader.init(props);
+ RegionFactory<String, String> rf =
cache.createRegionFactory(RegionShortcut.REPLICATE);
+ rf.setCacheLoader(jdbcLoader);
+ return rf.create(regionName);
+ }
+
+ @Test
+ public void verifySimpleGet() throws SQLException {
+ stmt.execute("Insert into " + regionTableName + " values('1', 'Emp1',
21)");
+ Region region = createRegionWithJDBCLoader(this.regionTableName,
getRequiredProperties());
+ Object result = region.get("1");
+ assertThat(result).isNotNull();
+ PdxInstance pdx = (PdxInstance) result;
+ assertThat(pdx.getField("name")).isEqualTo("Emp1");
+ assertThat(pdx.getField("age")).isEqualTo(21);
+ }
+
+ @Test
+ public void verifySimpleMiss() throws SQLException {
+ Region region = createRegionWithJDBCLoader(this.regionTableName,
getRequiredProperties());
+ Object result = region.get("1");
+ assertThat(result).isNull();
+ }
+
+ private void validateTableRowCount(int expected) throws Exception {
+ ResultSet rs = stmt.executeQuery("select count(*) from " +
regionTableName);
+ rs.next();
+ int size = rs.getInt(1);
+ assertThat(size).isEqualTo(expected);
+ }
+
+ private void printTable() throws Exception {
+ ResultSet rs = stmt.executeQuery("select * from " + regionTableName);
+ ResultSetMetaData rsmd = rs.getMetaData();
+ int columnsNumber = rsmd.getColumnCount();
+ while (rs.next()) {
+ for (int i = 1; i <= columnsNumber; i++) {
+ if (i > 1)
+ System.out.print(", ");
+ String columnValue = rs.getString(i);
+ System.out.print(rsmd.getColumnName(i) + ": " + columnValue);
+ }
+ System.out.println("");
+ }
+ }
+
+}
--
To stop receiving notification emails like this one, please contact
"commits@geode.apache.org" <commits@geode.apache.org>.
