[fix][dingo-calcite] Add lock operation for transaction

dingo-calcite/src/main/java/io/dingodb/calcite/DingoParser.java

@@ -19,9 +19,14 @@
 import com.google.common.collect.ImmutableList;
 import io.dingodb.calcite.grammar.ddl.SqlAnalyze;
 import io.dingodb.calcite.grammar.ddl.SqlBeginTx;
+import io.dingodb.calcite.grammar.ddl.SqlBlock;
 import io.dingodb.calcite.grammar.ddl.SqlCommit;
+import io.dingodb.calcite.grammar.ddl.SqlLockBlock;
+import io.dingodb.calcite.grammar.ddl.SqlLockTable;
 import io.dingodb.calcite.grammar.ddl.SqlRollback;
 import io.dingodb.calcite.grammar.ddl.SqlSetPassword;
+import io.dingodb.calcite.grammar.ddl.SqlUnLockBlock;
+import io.dingodb.calcite.grammar.ddl.SqlUnLockTable;
 import io.dingodb.calcite.grammar.dml.SqlExecute;
 import io.dingodb.calcite.grammar.dml.SqlPrepare;
 import io.dingodb.calcite.grammar.dql.SqlDesc;
@@ -76,6 +81,7 @@
 import org.checkerframework.checker.nullness.qual.NonNull;
 
 import java.sql.Connection;
+import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -88,6 +94,7 @@
 @Slf4j
 public class DingoParser {
     private static Map<String, String> sensitiveKey = new HashMap();
+
     static {
         sensitiveKey.put(".\"USER\"", ".USER");
     }
@@ -241,17 +248,16 @@ protected static boolean compatibleMysql(SqlNode sqlNode) {
             return false;
         } else if (sqlNode instanceof SqlSetOption && !(sqlNode instanceof SqlSetPassword)) {
             return true;
-        } else if (sqlNode instanceof SqlPrepare) {
-            return true;
-        } else if (sqlNode instanceof SqlExecute) {
-            return true;
-        } else if (sqlNode instanceof SqlAnalyze) {
-            return true;
-        } else if (sqlNode instanceof SqlBeginTx) {
-            return true;
-        } else if (sqlNode instanceof SqlCommit) {
-            return true;
-        } else if (sqlNode instanceof SqlRollback) {
+        } else if (sqlNode instanceof SqlPrepare
+            || sqlNode instanceof SqlExecute
+            || sqlNode instanceof SqlAnalyze
+            || sqlNode instanceof SqlBeginTx
+            || sqlNode instanceof SqlCommit
+            || sqlNode instanceof SqlRollback
+            || sqlNode instanceof SqlLockTable
+            || sqlNode instanceof SqlLockBlock
+            || sqlNode instanceof SqlUnLockTable
+            || sqlNode instanceof SqlUnLockBlock) {
             return true;
         }
         return false;

dingo-calcite/src/main/java/io/dingodb/calcite/grammar/ddl/SqlLockBlock.java

@@ -16,7 +16,7 @@
 
 package io.dingodb.calcite.grammar.ddl;
 
-import org.apache.calcite.sql.SqlDdl;
+import lombok.Getter;
 import org.apache.calcite.sql.SqlKind;
 import org.apache.calcite.sql.SqlNode;
 import org.apache.calcite.sql.SqlOperator;
@@ -28,7 +28,8 @@
 
 public class SqlLockBlock extends SqlLock {
 
-    List<SqlBlock> sqlBlockList;
+    @Getter
+    private List<SqlBlock> sqlBlockList;
 
     private static final SqlOperator OPERATOR =
         new SqlSpecialOperator("LOCK BLOCK", SqlKind.OTHER_DDL);

dingo-calcite/src/main/java/io/dingodb/calcite/grammar/ddl/SqlUnLockBlock.java

@@ -16,6 +16,7 @@
 
 package io.dingodb.calcite.grammar.ddl;
 
+import lombok.Getter;
 import org.apache.calcite.sql.SqlKind;
 import org.apache.calcite.sql.SqlOperator;
 import org.apache.calcite.sql.SqlSpecialOperator;
@@ -26,6 +27,7 @@
 
 public class SqlUnLockBlock extends SqlUnLock {
 
+    @Getter
     List<SqlBlock> sqlUnBlockList;
 
     private static final SqlOperator OPERATOR =

dingo-calcite/src/main/java/io/dingodb/calcite/operation/LockBlockOperation.java

@@ -0,0 +1,37 @@
+/*
+ * Copyright 2021 DataCanvas
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.dingodb.calcite.operation;
+
+import io.dingodb.calcite.grammar.ddl.SqlBlock;
+
+import java.sql.Connection;
+import java.util.List;
+
+public class LockBlockOperation implements DdlOperation {
+    private Connection connection;
+    private List<SqlBlock> sqlBlockList;
+
+    public LockBlockOperation(Connection connection, List<SqlBlock> sqlBlockList) {
+        this.connection = connection;
+        this.sqlBlockList = sqlBlockList;
+    }
+
+    @Override
+    public void execute() {
+
+    }
+}

dingo-calcite/src/main/java/io/dingodb/calcite/operation/LockTableOperation.java

@@ -0,0 +1,37 @@
+/*
+ * Copyright 2021 DataCanvas
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.dingodb.calcite.operation;
+
+import java.sql.Connection;
+import java.util.List;
+
+public class LockTableOperation implements DdlOperation {
+
+    private Connection connection;
+
+    private List<String> tableList;
+
+    public LockTableOperation(Connection connection, List<String> tableList) {
+        this.connection = connection;
+        this.tableList = tableList;
+    }
+
+    @Override
+    public void execute() {
+
+    }
+}

dingo-calcite/src/main/java/io/dingodb/calcite/operation/SqlToOperationConverter.java

@@ -20,7 +20,11 @@
 import io.dingodb.calcite.grammar.ddl.SqlAnalyze;
 import io.dingodb.calcite.grammar.ddl.SqlBeginTx;
 import io.dingodb.calcite.grammar.ddl.SqlCommit;
+import io.dingodb.calcite.grammar.ddl.SqlLockBlock;
+import io.dingodb.calcite.grammar.ddl.SqlLockTable;
 import io.dingodb.calcite.grammar.ddl.SqlRollback;
+import io.dingodb.calcite.grammar.ddl.SqlUnLockBlock;
+import io.dingodb.calcite.grammar.ddl.SqlUnLockTable;
 import io.dingodb.calcite.grammar.dql.SqlDesc;
 import io.dingodb.calcite.grammar.dql.SqlNextAutoIncrement;
 import io.dingodb.calcite.grammar.dql.SqlShowColumns;
@@ -132,6 +136,18 @@ public static Optional<Operation> convert(SqlNode sqlNode, Connection connection
         } else if (sqlNode instanceof SqlBeginTx) {
             SqlBeginTx sqlBeginTx = (SqlBeginTx) sqlNode;
             return Optional.of(new StartTransactionOperation(connection, sqlBeginTx.pessimistic));
+        } else if (sqlNode instanceof SqlLockTable) {
+            SqlLockTable sqlLockTable = (SqlLockTable) sqlNode;
+            return Optional.of(new LockTableOperation(connection, sqlLockTable.tableList));
+        } else if (sqlNode instanceof SqlLockBlock) {
+            SqlLockBlock sqlLockBlock = (SqlLockBlock) sqlNode;
+            return Optional.of(new LockBlockOperation(connection, sqlLockBlock.getSqlBlockList()));
+        } else if (sqlNode instanceof SqlUnLockTable) {
+            SqlUnLockTable sqlUnLockTable = (SqlUnLockTable) sqlNode;
+            return Optional.of(new UnlockTableOperation(connection, sqlUnLockTable.tableList));
+        } else if (sqlNode instanceof SqlUnLockBlock) {
+            SqlUnLockBlock sqlUnLockBlock = (SqlUnLockBlock) sqlNode;
+            return Optional.of(new UnlockBlockOperation(connection, sqlUnLockBlock.getSqlUnBlockList()));
         } else {
             return Optional.empty();
         }

dingo-calcite/src/main/java/io/dingodb/calcite/operation/UnlockBlockOperation.java

@@ -0,0 +1,38 @@
+/*
+ * Copyright 2021 DataCanvas
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.dingodb.calcite.operation;
+
+import io.dingodb.calcite.grammar.ddl.SqlBlock;
+
+import java.sql.Connection;
+import java.util.List;
+
+public class UnlockBlockOperation implements DdlOperation {
+
+    private Connection connection;
+    private List<SqlBlock> sqlBlockList;
+
+    public UnlockBlockOperation(Connection connection, List<SqlBlock> sqlBlockList) {
+        this.connection = connection;
+        this.sqlBlockList = sqlBlockList;
+    }
+
+    @Override
+    public void execute() {
+
+    }
+}

dingo-calcite/src/main/java/io/dingodb/calcite/operation/UnlockTableOperation.java

@@ -0,0 +1,37 @@
+/*
+ * Copyright 2021 DataCanvas
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.dingodb.calcite.operation;
+
+import java.sql.Connection;
+import java.util.List;
+
+public class UnlockTableOperation implements DdlOperation {
+
+    private Connection connection;
+
+    private List<String> tableList;
+
+    public UnlockTableOperation(Connection connection, List<String> tableList) {
+        this.connection = connection;
+        this.tableList = tableList;
+    }
+
+    @Override
+    public void execute() {
+
+    }
+}