mirror of
https://github.com/alibaba/druid.git
synced 2023-03-04 19:17:16 +00:00
hbse sql support
This commit is contained in:
@@ -1,21 +1,64 @@
|
||||
package com.alibaba.druid.hbase.exec;
|
||||
|
||||
import java.sql.SQLException;
|
||||
import java.util.LinkedHashMap;
|
||||
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.hbase.HBaseConnection;
|
||||
import com.alibaba.druid.hbase.HBasePreparedStatement;
|
||||
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 void execute(HBasePreparedStatement statement) throws SQLException {
|
||||
public boolean execute(HBasePreparedStatement statement) throws SQLException {
|
||||
try {
|
||||
HBaseConnection connection = statement.getConnection();
|
||||
HTableInterface htable = connection.getHTable(getTableName());
|
||||
String dbType = connection.getConnectProperties().getProperty("dbType");
|
||||
|
||||
throw new UnsupportedOperationException();
|
||||
Put put = null;
|
||||
for (Map.Entry<String, SQLExpr> entry : columns.entrySet()) {
|
||||
String column = entry.getKey();
|
||||
SQLExpr valueExpr = entry.getValue();
|
||||
|
||||
Object value = SQLEvalVisitorUtils.eval(dbType, valueExpr, statement.getParameters());
|
||||
|
||||
if (value == null) {
|
||||
continue;
|
||||
}
|
||||
|
||||
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 {
|
||||
throw new SQLException("TODO"); // TODO
|
||||
}
|
||||
|
||||
if (put == null) { // first value is key, TODO
|
||||
put = new Put(bytes);
|
||||
} else {
|
||||
byte[] qualifier = Bytes.toBytes(column);
|
||||
put.add(family, qualifier, bytes);
|
||||
}
|
||||
}
|
||||
|
||||
htable.put(put);
|
||||
|
||||
return false;
|
||||
} catch (SQLException e) {
|
||||
throw e;
|
||||
} catch (Exception e) {
|
||||
@@ -23,4 +66,12 @@ public class InsertExecutePlan extends SingleTableExecutePlan {
|
||||
}
|
||||
}
|
||||
|
||||
public Map<String, SQLExpr> getColumns() {
|
||||
return columns;
|
||||
}
|
||||
|
||||
public void setColumns(Map<String, SQLExpr> columns) {
|
||||
this.columns = columns;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user