hbse sql support

src/main/java/com/alibaba/druid/hbase/HBasePreparedStatement.java

@@ -7,14 +7,18 @@ import java.util.List;
 
 import com.alibaba.druid.common.jdbc.PreparedStatementBase;
 import com.alibaba.druid.hbase.exec.ExecutePlan;
+import com.alibaba.druid.hbase.exec.InsertExecutePlan;
 import com.alibaba.druid.hbase.exec.SingleTableQueryExecutePlan;
 import com.alibaba.druid.sql.SQLUtils;
+import com.alibaba.druid.sql.ast.SQLExpr;
 import com.alibaba.druid.sql.ast.SQLStatement;
 import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
 import com.alibaba.druid.sql.ast.statement.SQLExprTableSource;
 import com.alibaba.druid.sql.ast.statement.SQLInsertStatement;
 import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
 import com.alibaba.druid.sql.ast.statement.SQLSelectStatement;
+import com.alibaba.druid.sql.visitor.SQLEvalVisitor;
+import com.alibaba.druid.sql.visitor.SQLEvalVisitorUtils;
 
 public class HBasePreparedStatement extends PreparedStatementBase implements PreparedStatement, HBaseStatementInterface {
 
@@ -56,10 +60,24 @@ public class HBasePreparedStatement extends PreparedStatementBase implements Pre
             this.executePlan = singleTableQueryExecuetePlan;
         } else if (sqlStmt instanceof SQLInsertStatement) {
             SQLInsertStatement insertStmt = (SQLInsertStatement) sqlStmt;
+
+            SQLEvalVisitor evalVisitor = SQLEvalVisitorUtils.createEvalVisitor(dbType);
+            insertStmt.accept(evalVisitor);
             
             String tableName = ((SQLIdentifierExpr) insertStmt.getTableSource().getExpr()).getName();
             
-            throw new SQLException("TODO");
+            InsertExecutePlan insertExecutePlan = new InsertExecutePlan();
+            insertExecutePlan.setTableName(tableName);
+            
+            for (int i = 0; i < insertStmt.getColumns().size(); ++i) {
+                SQLExpr columnExpr = insertStmt.getColumns().get(i);
+                SQLExpr valueExpr = insertStmt.getValues().getValues().get(i);
+                
+                String columnName = ((SQLIdentifierExpr) columnExpr).getName();
+                insertExecutePlan.getColumns().put(columnName, valueExpr);
+            }
+            
+            this.executePlan = insertExecutePlan;
         } else {
             throw new SQLException("TODO");
         }
@@ -99,7 +117,7 @@ public class HBasePreparedStatement extends PreparedStatementBase implements Pre
 
     @Override
     public boolean execute() throws SQLException {
-        checkOpen();
+        this.executePlan.execute(this);
 
         return false;
     }

src/main/java/com/alibaba/druid/hbase/exec/ExecutePlan.java

@@ -9,5 +9,5 @@ public interface ExecutePlan {
 
     HBaseResultSet executeQuery(HBasePreparedStatement statement) throws SQLException;
 
-    void execute(HBasePreparedStatement statement) throws SQLException;
+    boolean execute(HBasePreparedStatement statement) throws SQLException;
 }

src/main/java/com/alibaba/druid/hbase/exec/ExecutePlanAdapter.java

@@ -13,7 +13,7 @@ public class ExecutePlanAdapter implements ExecutePlan {
     }
 
     @Override
-    public void execute(HBasePreparedStatement statement) throws SQLException {
+    public boolean execute(HBasePreparedStatement statement) throws SQLException {
         throw new UnsupportedOperationException();
     }
 

src/main/java/com/alibaba/druid/hbase/exec/InsertExecutePlan.java

@@ -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;
+    }
+
 }

src/main/java/com/alibaba/druid/hbase/exec/SingleTableExecutePlan.java

@@ -1,7 +1,7 @@
 package com.alibaba.druid.hbase.exec;
 
-
 public class SingleTableExecutePlan extends ExecutePlanAdapter {
+
     private String tableName;
 
     public String getTableName() {
@@ -11,4 +11,5 @@ public class SingleTableExecutePlan extends ExecutePlanAdapter {
     public void setTableName(String tableName) {
         this.tableName = tableName;
     }
+
 }

src/main/java/com/alibaba/druid/sql/dialect/mysql/visitor/MySqlEvalVisitorImpl.java

@@ -11,9 +11,11 @@ import com.alibaba.druid.sql.visitor.SQLEvalVisitorUtils;
 
 public class MySqlEvalVisitorImpl extends MySqlASTVisitorAdapter implements SQLEvalVisitor {
 
-    private List<Object> parameters   = new ArrayList<Object>();
+    private List<Object> parameters       = new ArrayList<Object>();
 
-    private int          variantIndex = -1;
+    private int          variantIndex     = -1;
+
+    private boolean      markVariantIndex = true;
 
     public MySqlEvalVisitorImpl(){
         this(new ArrayList<Object>(1));
@@ -46,8 +48,17 @@ public class MySqlEvalVisitorImpl extends MySqlASTVisitorAdapter implements SQLE
     public boolean visit(SQLVariantRefExpr x) {
         return SQLEvalVisitorUtils.visit(this, x);
     }
-    
+
     public boolean visit(SQLBinaryOpExpr x) {
         return SQLEvalVisitorUtils.visit(this, x);
     }
+
+    public boolean isMarkVariantIndex() {
+        return markVariantIndex;
+    }
+
+    public void setMarkVariantIndex(boolean markVariantIndex) {
+        this.markVariantIndex = markVariantIndex;
+    }
+
 }

src/main/java/com/alibaba/druid/sql/visitor/SQLEvalVisitor.java

@@ -13,4 +13,8 @@ public interface SQLEvalVisitor extends SQLASTVisitor {
     void setParameters(List<Object> parameters);
 
     int incrementAndGetVariantIndex();
+    
+    boolean isMarkVariantIndex();
+    
+    void setMarkVariantIndex(boolean markVariantIndex);
 }

src/main/java/com/alibaba/druid/sql/visitor/SQLEvalVisitorImpl.java

@@ -9,9 +9,11 @@ import com.alibaba.druid.sql.ast.expr.SQLVariantRefExpr;
 
 public class SQLEvalVisitorImpl extends SQLASTVisitorAdapter implements SQLEvalVisitor {
 
-    private List<Object> parameters   = new ArrayList<Object>();
+    private List<Object> parameters       = new ArrayList<Object>();
 
-    private int          variantIndex = -1;
+    private int          variantIndex     = -1;
+
+    private boolean      markVariantIndex = true;
 
     public SQLEvalVisitorImpl(){
         this(new ArrayList<Object>(1));
@@ -48,4 +50,13 @@ public class SQLEvalVisitorImpl extends SQLASTVisitorAdapter implements SQLEvalV
     public boolean visit(SQLBinaryOpExpr x) {
         return SQLEvalVisitorUtils.visit(this, x);
     }
+
+    public boolean isMarkVariantIndex() {
+        return markVariantIndex;
+    }
+
+    public void setMarkVariantIndex(boolean markVariantIndex) {
+        this.markVariantIndex = markVariantIndex;
+    }
+
 }

src/main/java/com/alibaba/druid/sql/visitor/SQLEvalVisitorUtils.java

@@ -67,24 +67,24 @@ public class SQLEvalVisitorUtils {
         x.putAttribute(EVAL_VALUE, x.getText());
         return true;
     }
-    
+
     public static boolean visit(SQLEvalVisitor visitor, SQLBinaryOpExpr x) {
         x.getLeft().accept(visitor);
         x.getRight().accept(visitor);
-        
+
         if (!x.getLeft().getAttributes().containsKey(EVAL_VALUE)) {
             return false;
         }
-        
+
         if (!x.getRight().getAttributes().containsKey(EVAL_VALUE)) {
             return false;
         }
-        
+
         if (SQLBinaryOperator.Add.equals(x.getOperator())) {
             Object value = _add(x.getLeft().getAttribute(EVAL_VALUE), x.getRight().getAttributes().get(EVAL_VALUE));
             x.putAttribute(EVAL_VALUE, value);
         }
-        
+
         return false;
     }
 
@@ -96,10 +96,12 @@ public class SQLEvalVisitorUtils {
             x.putAttribute(EVAL_VAR_INDEX, varIndex);
         }
 
-        boolean containsValue = attributes.containsKey(EVAL_VALUE);
-        if (!containsValue) {
-            Object value = visitor.getParameters().get(varIndex.intValue());
-            attributes.put(EVAL_VALUE, value);
+        if (visitor.getParameters().size() > 0) {
+            boolean containsValue = attributes.containsKey(EVAL_VALUE);
+            if (!containsValue) {
+                Object value = visitor.getParameters().get(varIndex.intValue());
+                attributes.put(EVAL_VALUE, value);
+            }
         }
 
         return false;