From 90ea2c6194e527aaae240e74c67d3b236343ad01 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=A9=E5=B0=91?= Date: Mon, 21 May 2012 16:25:02 +0800 Subject: [PATCH] refactor --- .../com/alibaba/druid/hdriver/HConnection.java | 5 ++++- .../com/alibaba/druid/hdriver/HDriver.java | 2 +- .../com/alibaba/druid/hdriver/HResultSet.java | 3 ++- .../com/alibaba/druid/hdriver/HStatement.java | 4 +--- .../hdriver/impl/execute/ShowTablesPlan.java | 6 ++++++ .../hdriver/impl/jdbc/HScannerResultSet.java | 3 +-- .../alibaba/druid/util/jdbc/ResultSetBase.java | 18 +++++++++--------- 7 files changed, 24 insertions(+), 17 deletions(-) diff --git a/src/main/java/com/alibaba/druid/hdriver/HConnection.java b/src/main/java/com/alibaba/druid/hdriver/HConnection.java index f33ca3e2b9..bf35951fd7 100644 --- a/src/main/java/com/alibaba/druid/hdriver/HConnection.java +++ b/src/main/java/com/alibaba/druid/hdriver/HConnection.java @@ -1,7 +1,10 @@ package com.alibaba.druid.hdriver; import java.sql.Connection; +import java.sql.SQLException; + +import com.alibaba.druid.hdriver.impl.jdbc.HPreparedStatementImpl; public interface HConnection extends Connection { - + HPreparedStatementImpl prepareStatement(String sql) throws SQLException; } diff --git a/src/main/java/com/alibaba/druid/hdriver/HDriver.java b/src/main/java/com/alibaba/druid/hdriver/HDriver.java index cb36294dbe..693dd982b3 100644 --- a/src/main/java/com/alibaba/druid/hdriver/HDriver.java +++ b/src/main/java/com/alibaba/druid/hdriver/HDriver.java @@ -12,7 +12,7 @@ public class HDriver implements Driver { public static String PREFIX = "jdbc:druid-hbase:"; @Override - public HBaseConnectionImpl connect(String url, Properties info) throws SQLException { + public HConnection connect(String url, Properties info) throws SQLException { if (!acceptsURL(url)) { return null; } diff --git a/src/main/java/com/alibaba/druid/hdriver/HResultSet.java b/src/main/java/com/alibaba/druid/hdriver/HResultSet.java index 1605ec692e..779c202c8f 100644 --- a/src/main/java/com/alibaba/druid/hdriver/HResultSet.java +++ b/src/main/java/com/alibaba/druid/hdriver/HResultSet.java @@ -1,8 +1,9 @@ package com.alibaba.druid.hdriver; import java.sql.ResultSet; +import java.sql.SQLException; public interface HResultSet extends ResultSet { - + HStatement getStatement() throws SQLException; } diff --git a/src/main/java/com/alibaba/druid/hdriver/HStatement.java b/src/main/java/com/alibaba/druid/hdriver/HStatement.java index 47c498d365..dfc883c44d 100644 --- a/src/main/java/com/alibaba/druid/hdriver/HStatement.java +++ b/src/main/java/com/alibaba/druid/hdriver/HStatement.java @@ -3,8 +3,6 @@ package com.alibaba.druid.hdriver; import java.sql.SQLException; import java.sql.Statement; -import com.alibaba.druid.hdriver.impl.jdbc.HBaseConnectionImpl; - public interface HStatement extends Statement { - HBaseConnectionImpl getConnection() throws SQLException; + HConnection getConnection() throws SQLException; } diff --git a/src/main/java/com/alibaba/druid/hdriver/impl/execute/ShowTablesPlan.java b/src/main/java/com/alibaba/druid/hdriver/impl/execute/ShowTablesPlan.java index d39d2bb413..be432ea574 100644 --- a/src/main/java/com/alibaba/druid/hdriver/impl/execute/ShowTablesPlan.java +++ b/src/main/java/com/alibaba/druid/hdriver/impl/execute/ShowTablesPlan.java @@ -9,6 +9,7 @@ import org.apache.hadoop.hbase.HTableDescriptor; import org.apache.hadoop.hbase.client.HBaseAdmin; import com.alibaba.druid.hdriver.HResultSet; +import com.alibaba.druid.hdriver.HStatement; import com.alibaba.druid.hdriver.impl.jdbc.HPreparedStatementImpl; import com.alibaba.druid.hdriver.impl.jdbc.HResultSetMetaData; import com.alibaba.druid.util.jdbc.ResultSetBase; @@ -169,5 +170,10 @@ public class ShowTablesPlan implements ExecutePlan { @Override public void updateObject(int columnIndex, Object x) throws SQLException { } + + @Override + public HStatement getStatement() throws SQLException { + return (HStatement) statement; + } } } diff --git a/src/main/java/com/alibaba/druid/hdriver/impl/jdbc/HScannerResultSet.java b/src/main/java/com/alibaba/druid/hdriver/impl/jdbc/HScannerResultSet.java index 19f41532a1..9875f09a07 100644 --- a/src/main/java/com/alibaba/druid/hdriver/impl/jdbc/HScannerResultSet.java +++ b/src/main/java/com/alibaba/druid/hdriver/impl/jdbc/HScannerResultSet.java @@ -3,7 +3,6 @@ package com.alibaba.druid.hdriver.impl.jdbc; import java.io.IOException; import java.math.BigDecimal; import java.sql.SQLException; -import java.sql.Statement; import java.sql.Types; import org.apache.hadoop.hbase.client.HTableInterface; @@ -44,7 +43,7 @@ public class HScannerResultSet extends ResultSetBase implements HResultSet { } @Override - public Statement getStatement() throws SQLException { + public HStatement getStatement() throws SQLException { return statement; } diff --git a/src/main/java/com/alibaba/druid/util/jdbc/ResultSetBase.java b/src/main/java/com/alibaba/druid/util/jdbc/ResultSetBase.java index c73e2f9d91..7e354c8409 100644 --- a/src/main/java/com/alibaba/druid/util/jdbc/ResultSetBase.java +++ b/src/main/java/com/alibaba/druid/util/jdbc/ResultSetBase.java @@ -25,14 +25,14 @@ import java.util.Map; public abstract class ResultSetBase implements ResultSet { - protected boolean closed = false; - protected boolean wasNull = false; - private SQLWarning warning; - private String cursorName; - private int fetchSize = 0; - private int fetchDirection = 0; + protected boolean closed = false; + protected boolean wasNull = false; + private SQLWarning warning; + private String cursorName; + private int fetchSize = 0; + private int fetchDirection = 0; - private Statement statement; + protected Statement statement; protected ResultSetMetaData metaData; public ResultSetBase(Statement statement){ @@ -309,7 +309,7 @@ public abstract class ResultSetBase implements ResultSet { String columName = this.getMetaData().getColumnName(columnIndex); return getObject(columName); } - + return null; } @@ -1166,7 +1166,7 @@ public abstract class ResultSetBase implements ResultSet { return 0; } - + @Override public ResultSetMetaData getMetaData() throws SQLException { if (closed) {