refactor

2023-03-04 19:17:16 +00:00 · 2012-05-21 16:11:51 +08:00
parent 74c5f89b05
commit 98e8e226c5
38 changed files with 72 additions and 63 deletions
@@ -0,0 +1,67 @@
+package com.alibaba.druid.hdriver.execute;
+
+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.hdriver.jdbc.HBaseConnection;
+import com.alibaba.druid.hdriver.jdbc.HPreparedStatementImpl;
+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 {
+            HBaseConnection connection = statement.getConnection();
+            String dbType = connection.getConnectProperties().getProperty("dbType");
+
+            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 = HBaseUtils.toBytes(value);
+
+                if (put == null) { // first value is key, TODO
+                    put = new Put(bytes);
+                } else {
+                    byte[] qualifier = Bytes.toBytes(column);
+                    put.add(family, qualifier, bytes);
+                }
+            }
+
+            HTableInterface htable = connection.getHTable(getTableName());
+            htable.put(put);
+
+            return false;
+        } catch (SQLException e) {
+            throw e;
+        } catch (Exception e) {
+            throw new SQLException("executeQuery error", e);
+        }
+    }
+
+    public Map<String, SQLExpr> getColumns() {
+        return columns;
+    }
+
+    public void setColumns(Map<String, SQLExpr> columns) {
+        this.columns = columns;
+    }
+
+}