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:04:43 +08:00
parent 401ec28bbf
commit dbe458d97c
7 changed files with 77 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/hdriver/impl/execute/HBaseUtils.java
-37
View File
@@ -1,37 +0,0 @@
package com.alibaba.druid.hdriver.impl.execute;
import java.io.IOException;
import java.math.BigDecimal;
import org.apache.hadoop.hbase.util.Bytes;
public class HBaseUtils {
public static byte[] toBytes(Object value) throws IOException {
if (value == null) {
return null;
}
byte[] bytes;
if (value instanceof String) {
String strValue = (String) value;
bytes = Bytes.toBytes(strValue);
} else if (value instanceof Integer) {
int intValue = ((Integer) value).intValue();
bytes = Bytes.toBytes(intValue);
} else if (value instanceof Long) {
long longValue = ((Long) value).longValue();
bytes = Bytes.toBytes(longValue);
} else if (value instanceof Boolean) {
boolean booleanValue = ((Boolean) value).booleanValue();
bytes = Bytes.toBytes(booleanValue);
} else if (value instanceof BigDecimal) {
BigDecimal decimalValue = (BigDecimal) value;
bytes = Bytes.toBytes(decimalValue);
} else {
throw new IOException("TODO"); // TODO
}
return bytes;
}
}
src/main/java/com/alibaba/druid/hdriver/impl/execute/InsertExecutePlan.java
+12 -6
View File
@@ -6,21 +6,26 @@ import java.util.Map;
import org.apache.hadoop.hbase.client.HTableInterface;
import org.apache.hadoop.hbase.client.Put;
import org.apache.hadoop.hbase.util.Bytes;
import com.alibaba.druid.hdriver.impl.HBaseConnectionImpl;
import com.alibaba.druid.hdriver.impl.HPreparedStatementImpl;
import com.alibaba.druid.hdriver.impl.mapping.HMapping;
import com.alibaba.druid.hdriver.impl.mapping.HMappingDefaultImpl;
import com.alibaba.druid.sql.ast.SQLExpr;
import com.alibaba.druid.sql.visitor.SQLEvalVisitorUtils;
public class InsertExecutePlan extends SingleTableExecutePlan {
private Map<String, SQLExpr> columns = new LinkedHashMap<String, SQLExpr>();
private byte[] family = Bytes.toBytes("d");
@Override
public boolean execute(HPreparedStatementImpl statement) throws SQLException {
try {
HMapping mapping = this.getMapping();
if (mapping == null) {
mapping = new HMappingDefaultImpl();
}
HBaseConnectionImpl connection = statement.getConnection();
String dbType = connection.getConnectProperties().getProperty("dbType");
@@ -34,13 +39,14 @@ public class InsertExecutePlan extends SingleTableExecutePlan {
if (value == null) {
continue;
}
byte[] bytes = mapping.toBytes(column, value);
byte[] bytes = HBaseUtils.toBytes(value);
if (put == null) { // first value is key, TODO
if (mapping.isRow(column)) {
put = new Put(bytes);
} else {
byte[] qualifier = Bytes.toBytes(column);
byte[] family = mapping.getFamily(column);
byte[] qualifier = mapping.getQualifier(column);
put.add(family, qualifier, bytes);
}
}
src/main/java/com/alibaba/druid/hdriver/impl/execute/SingleTableExecutePlan.java
+13 -1
View File
@@ -1,8 +1,12 @@
package com.alibaba.druid.hdriver.impl.execute;
import com.alibaba.druid.hdriver.impl.mapping.HMapping;
public class SingleTableExecutePlan extends ExecutePlanAdapter {
private String tableName;
private String tableName;
private HMapping mapping;
public String getTableName() {
return tableName;
@@ -12,4 +16,12 @@ public class SingleTableExecutePlan extends ExecutePlanAdapter {
this.tableName = tableName;
}
public HMapping getMapping() {
return mapping;
}
public void setMapping(HMapping mapping) {
this.mapping = mapping;
}
}
src/main/java/com/alibaba/druid/hdriver/impl/execute/SingleTableQueryExecutePlan.java
+10 -16
View File
@@ -13,13 +13,13 @@ import org.apache.hadoop.hbase.filter.Filter;
import org.apache.hadoop.hbase.filter.FilterList;
import org.apache.hadoop.hbase.filter.RowFilter;
import org.apache.hadoop.hbase.filter.SingleColumnValueFilter;
import org.apache.hadoop.hbase.util.Bytes;
import com.alibaba.druid.hdriver.impl.HBaseConnectionImpl;
import com.alibaba.druid.hdriver.impl.HPreparedStatementImpl;
import com.alibaba.druid.hdriver.impl.HResultSetMetaDataImpl;
import com.alibaba.druid.hdriver.impl.HScannerResultSetImpl;
import com.alibaba.druid.hdriver.impl.mapping.HMapping;
import com.alibaba.druid.hdriver.impl.mapping.HMappingDefaultImpl;
import com.alibaba.druid.sql.ast.SQLExpr;
import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
import com.alibaba.druid.sql.ast.expr.SQLBinaryOperator;
@@ -31,24 +31,12 @@ public class SingleTableQueryExecutePlan extends SingleTableExecutePlan {
private List<String> columeNames = new ArrayList<String>();
private List<SQLExpr> conditions = new ArrayList<SQLExpr>();
private byte[] family = Bytes.toBytes("d");
private HResultSetMetaDataImpl resultMetaData;
private HMapping mapping;
public SingleTableQueryExecutePlan(){
}
public HMapping getMapping() {
return mapping;
}
public void setMapping(HMapping mapping) {
this.mapping = mapping;
}
public List<SQLExpr> getConditions() {
return conditions;
}
@@ -72,6 +60,11 @@ public class SingleTableQueryExecutePlan extends SingleTableExecutePlan {
@Override
public HScannerResultSetImpl executeQuery(HPreparedStatementImpl statement) throws SQLException {
try {
HMapping mapping = this.getMapping();
if (mapping == null) {
mapping = new HMappingDefaultImpl();
}
HBaseConnectionImpl connection = statement.getConnection();
String dbType = connection.getConnectProperties().getProperty("dbType");
@@ -82,8 +75,8 @@ public class SingleTableQueryExecutePlan extends SingleTableExecutePlan {
String fieldName = ((SQLIdentifierExpr) condition.getLeft()).getName();
Object value = SQLEvalVisitorUtils.eval(dbType, condition.getRight(), statement.getParameters());
byte[] bytes = HBaseUtils.toBytes(value);
if ("id".equals(fieldName)) {
byte[] bytes = mapping.toBytes(fieldName, value);
if (mapping.isRow(fieldName)) {
if (condition.getOperator() == SQLBinaryOperator.GreaterThanOrEqual) {
scan.setStartRow(bytes);
} else if (condition.getOperator() == SQLBinaryOperator.LessThan) {
@@ -95,7 +88,8 @@ public class SingleTableQueryExecutePlan extends SingleTableExecutePlan {
throw new SQLException("TODO");
}
} else {
byte[] qualifier = Bytes.toBytes(fieldName);
byte[] qualifier = mapping.getQualifier(fieldName);
byte[] family = mapping.getFamily(fieldName);
CompareOp compareOp;
if (condition.getOperator() == SQLBinaryOperator.Equality) {
src/main/java/com/alibaba/druid/hdriver/impl/mapping/HMapping.java
+7 -3
View File
@@ -1,5 +1,7 @@
package com.alibaba.druid.hdriver.impl.mapping;
import java.io.IOException;
import org.apache.hadoop.hbase.client.Result;
public interface HMapping {
@@ -7,10 +9,12 @@ public interface HMapping {
byte[] getFamily(String columnName);
byte[] getQualifier(String columnName);
byte[] getRow(Result result, String columnName);
boolean isRow(String columnName);
Object getObject(Result result, String name);
Object getObject(Result result, String columnName);
byte[] toBytes(String columnName, Object value) throws IOException;
}
src/main/java/com/alibaba/druid/hdriver/impl/mapping/HMappingAdapter.java
+31
View File
@@ -1,5 +1,8 @@
package com.alibaba.druid.hdriver.impl.mapping;
import java.io.IOException;
import java.math.BigDecimal;
import org.apache.hadoop.hbase.client.Result;
import org.apache.hadoop.hbase.util.Bytes;
@@ -30,4 +33,32 @@ public abstract class HMappingAdapter implements HMapping {
return bytes;
}
@Override
public byte[] toBytes(String columnName, Object value) throws IOException {
if (value == null) {
return null;
}
byte[] bytes;
if (value instanceof String) {
String strValue = (String) value;
bytes = Bytes.toBytes(strValue);
} else if (value instanceof Integer) {
int intValue = ((Integer) value).intValue();
bytes = Bytes.toBytes(intValue);
} else if (value instanceof Long) {
long longValue = ((Long) value).longValue();
bytes = Bytes.toBytes(longValue);
} else if (value instanceof Boolean) {
boolean booleanValue = ((Boolean) value).booleanValue();
bytes = Bytes.toBytes(booleanValue);
} else if (value instanceof BigDecimal) {
BigDecimal decimalValue = (BigDecimal) value;
bytes = Bytes.toBytes(decimalValue);
} else {
throw new IOException("TODO"); // TODO
}
return bytes;
}
}
src/main/java/com/alibaba/druid/hdriver/impl/mapping/HMappingDefaultImpl.java
+4
View File
@@ -1,5 +1,8 @@
package com.alibaba.druid.hdriver.impl.mapping;
import java.io.IOException;
import java.math.BigDecimal;
import org.apache.hadoop.hbase.util.Bytes;
public class HMappingDefaultImpl extends HMappingAdapter implements HMapping {
@@ -35,4 +38,5 @@ public class HMappingDefaultImpl extends HMappingAdapter implements HMapping {
return false;
}
}