bug fix: SENSEI-269

wonlay · Dec 12, 2012 · 88a665a · 88a665a
1 parent e49a0e5
commit 88a665a
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 7 deletions.
diff --git a/sensei-gateways/src/main/java/com/senseidb/gateway/jdbc/JdbcDataProviderBuilder.java b/sensei-gateways/src/main/java/com/senseidb/gateway/jdbc/JdbcDataProviderBuilder.java
@@ -41,10 +41,10 @@ public StreamDataProvider<JSONObject> buildDataProvider(final DataSourceFilter<R
       Set<Integer> partitions) throws Exception
   {
 
-	       final String url = config.get("jdbc.url");
-	       final String username = config.get("jdbc.username");
-	       final String password = config.get("jdbc.password");
-	       final String driver = config.get("jdbc.driver");
+	     final String url = config.get("jdbc.url");
+	     final String username = config.get("jdbc.username");
+	     final String password = config.get("jdbc.password");
+	     final String driver = config.get("jdbc.driver");
          final String adaptor = config.get("jdbc.adaptor");
 
          final SenseiJDBCAdaptor senseiAdaptor =
@@ -56,6 +56,7 @@ public StreamDataProvider<JSONObject> buildDataProvider(final DataSourceFilter<R
            throw new ConfigurationException("adaptor not found: " + adaptor);
 	       }
 
+	       senseiAdaptor.setPartitions(partitions);
 
 		   JDBCConnectionFactory connFactory = new JDBCConnectionFactory() {
 

diff --git a/sensei-gateways/src/main/java/com/senseidb/gateway/jdbc/SenseiJDBCAdaptor.java b/sensei-gateways/src/main/java/com/senseidb/gateway/jdbc/SenseiJDBCAdaptor.java
@@ -4,8 +4,16 @@
 import java.sql.PreparedStatement;
 import java.sql.ResultSet;
 import java.sql.SQLException;
+import java.util.Set;
 
-public interface SenseiJDBCAdaptor {
-  PreparedStatement buildStatment(Connection conn,String fromVersion) throws SQLException;
-  String extractVersion(ResultSet resultSet) throws SQLException;
+public abstract class SenseiJDBCAdaptor {
+
+  protected Set<Integer> partitions = null;
+
+  abstract public PreparedStatement buildStatment(Connection conn,String fromVersion) throws SQLException;
+  abstract public String extractVersion(ResultSet resultSet) throws SQLException;
+
+  public void setPartitions(Set<Integer> partitions){
+    this.partitions = partitions;
+  }
 }