Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

hbase support

Browse Source
This commit is contained in:
温少
2012-05-21 21:59:46 +08:00
parent 664a4b1258
commit ec2debdfd7
2 changed files with 86 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/hdriver/impl/execute/SingleTableQueryExecutePlan.java
+49 -19
View File
@@ -76,9 +76,13 @@ public class SingleTableQueryExecutePlan extends SingleTableExecutePlan {
scan = new Scan();
this.statement = statement;
Filter filter = null;
for (SQLExpr item : conditions) {
SQLBinaryOpExpr condition = (SQLBinaryOpExpr) item;
setFilter(condition);
filter = setFilter(condition, filter, true);
}
if (filter != null) {
scan.setFilter(filter);
}
HTableInterface htable = connection.getHTable(getTableName());
@@ -98,24 +102,35 @@ public class SingleTableQueryExecutePlan extends SingleTableExecutePlan {
}
}
private void setFilter(SQLBinaryOpExpr condition) throws IOException,
SQLException {
private Filter setFilter(SQLBinaryOpExpr condition, Filter filter, boolean and) throws IOException, SQLException {
HMapping mapping = this.getMapping();
if (condition.getOperator() == SQLBinaryOperator.BooleanAnd) {
filter = setFilter((SQLBinaryOpExpr) condition.getLeft(), filter, true);
filter = setFilter((SQLBinaryOpExpr) condition.getRight(), filter, true);
return filter;
} else if (condition.getOperator() == SQLBinaryOperator.BooleanOr) {
filter = setFilter((SQLBinaryOpExpr) condition.getLeft(), filter, false);
filter = setFilter((SQLBinaryOpExpr) condition.getRight(), filter, false);
return filter;
}
String fieldName = ((SQLIdentifierExpr) condition.getLeft()).getName();
Object value = SQLEvalVisitorUtils.eval(dbType, condition.getRight(), statement.getParameters());
byte[] bytes = mapping.toBytes(fieldName, value);
if (mapping.isRow(fieldName)) {
if (condition.getOperator() == SQLBinaryOperator.GreaterThanOrEqual) {
if (filter == null && condition.getOperator() == SQLBinaryOperator.GreaterThanOrEqual) {
scan.setStartRow(bytes);
} else if (condition.getOperator() == SQLBinaryOperator.LessThan) {
return null;
} else if (filter == null && condition.getOperator() == SQLBinaryOperator.LessThan) {
scan.setStopRow(bytes);
} else if (condition.getOperator() == SQLBinaryOperator.LessThanOrEqual) {
RowFilter filter = new RowFilter(CompareOp.LESS_OR_EQUAL, new BinaryComparator(bytes));
setFilter(scan, filter);
return null;
} else {
throw new SQLException("TODO");
CompareOp compareOp = toCompareOp(condition.getOperator());
RowFilter rowFilter = new RowFilter(compareOp, new BinaryComparator(bytes));
return setFilter(filter, rowFilter, and);
}
} else {
byte[] qualifier = mapping.getQualifier(fieldName);
@@ -123,9 +138,10 @@ public class SingleTableQueryExecutePlan extends SingleTableExecutePlan {
CompareOp compareOp = toCompareOp(condition.getOperator());
SingleColumnValueFilter filter = new SingleColumnValueFilter(family, qualifier, compareOp, bytes);
setFilter(scan, filter);
SingleColumnValueFilter columnFilter = new SingleColumnValueFilter(family, qualifier, compareOp, bytes);
return setFilter(filter, columnFilter, and);
}
}
CompareOp toCompareOp(SQLBinaryOperator operator) {
@@ -147,15 +163,29 @@ public class SingleTableQueryExecutePlan extends SingleTableExecutePlan {
}
}
void setFilter(Scan scan, Filter filter) {
if (scan.getFilter() == null) {
scan.setFilter(filter);
} else if (scan.getFilter() instanceof FilterList) {
FilterList filterList = (FilterList) scan.getFilter();
filterList.addFilter(filter);
Filter setFilter(Filter parentFilter, Filter filter, boolean and) {
if (parentFilter == null) {
return filter;
} else if (parentFilter instanceof FilterList) {
FilterList filterList = (FilterList) parentFilter;
if (and) {
if (filterList.getOperator() == FilterList.Operator.MUST_PASS_ALL) {
filterList.addFilter(filter);
return filterList;
} else {
return new FilterList(FilterList.Operator.MUST_PASS_ALL, parentFilter, filter);
}
} else {
if (filterList.getOperator() == FilterList.Operator.MUST_PASS_ONE) {
filterList.addFilter(filter);
return filterList;
} else {
return new FilterList(FilterList.Operator.MUST_PASS_ONE, parentFilter, filter);
}
}
} else {
FilterList filterList = new FilterList(scan.getFilter(), filter);
scan.setFilter(filterList);
FilterList.Operator filterOp = and ? FilterList.Operator.MUST_PASS_ALL : FilterList.Operator.MUST_PASS_ONE;
return new FilterList(filterOp, parentFilter, filter);
}
}
src/test/java/com/alibaba/druid/hbase/HBaseDriverTest_getIndex_or.java
+37
View File
@@ -0,0 +1,37 @@
package com.alibaba.druid.hbase;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.util.Properties;
import com.alibaba.druid.hdriver.HDriver;
import junit.framework.TestCase;
public class HBaseDriverTest_getIndex_or extends TestCase {
public void test_select_0 () throws Exception {
HDriver driver = new HDriver();
Connection conn = driver.connect("jdbc:druid-hbase:10.20.153.63", new Properties());
PreparedStatement stmt = conn.prepareStatement("SELECT id, name, gender, salary FROM test_user where id = 3 or id = ?");
stmt.setInt(1, 6);
ResultSet rs = stmt.executeQuery();
while (rs.next()) {
System.out.print(rs.getInt(1));
System.out.print(' ');
System.out.print(rs.getString(2));
System.out.print(' ');
System.out.print(rs.getString(3));
System.out.print(' ');
System.out.print(rs.getBigDecimal(4));
System.out.println();
}
rs.close();
stmt.close();
conn.close();
}
}