[shardingsphere] branch master updated: Add more Privilege and SQL statement mapping (#11230)

[panjuan]

This is an automated email from the ASF dual-hosted git repository.

panjuan pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git


The following commit(s) were added to refs/heads/master by this push:
     new 9eaf98f  Add more Privilege and SQL statement mapping (#11230)
9eaf98f is described below

commit 9eaf98f2afc1366d3909c43967945c91981bb947
Author: Zhu jun <zhujunxx...@163.com>
AuthorDate: Fri Jul 9 07:31:12 2021 +0800

    Add more Privilege and SQL statement mapping (#11230)
---
 .../authority/checker/AuthorityChecker.java        | 61 ++++++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git 
a/shardingsphere-infra/shardingsphere-infra-authority/shardingsphere-infra-authority-common/src/main/java/org/apache/shardingsphere/authority/checker/AuthorityChecker.java
 
b/shardingsphere-infra/shardingsphere-infra-authority/shardingsphere-infra-authority-common/src/main/java/org/apache/shardingsphere/authority/checker/AuthorityChecker.java
index 513ba3a..cfa3875 100644
--- 
a/shardingsphere-infra/shardingsphere-infra-authority/shardingsphere-infra-authority-common/src/main/java/org/apache/shardingsphere/authority/checker/AuthorityChecker.java
+++ 
b/shardingsphere-infra/shardingsphere-infra-authority/shardingsphere-infra-authority-common/src/main/java/org/apache/shardingsphere/authority/checker/AuthorityChecker.java
@@ -27,6 +27,20 @@ import 
org.apache.shardingsphere.infra.metadata.ShardingSphereMetaData;
 import org.apache.shardingsphere.infra.metadata.user.Grantee;
 import org.apache.shardingsphere.infra.metadata.user.ShardingSphereUser;
 import org.apache.shardingsphere.sql.parser.sql.common.statement.SQLStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.AlterDatabaseStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.AlterTableStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.CreateDatabaseStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.CreateFunctionStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.CreateTableStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.DDLStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.DropDatabaseStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.DropTableStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.TruncateStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.dml.DMLStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.dml.DeleteStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.dml.InsertStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.dml.SelectStatement;
+import 
org.apache.shardingsphere.sql.parser.sql.common.statement.dml.UpdateStatement;
 import 
org.apache.shardingsphere.sql.parser.sql.dialect.statement.mysql.dal.MySQLShowDatabasesStatement;
 
 import java.util.Collections;
@@ -74,9 +88,56 @@ public final class AuthorityChecker implements 
SQLChecker<AuthorityRule> {
         if (sqlStatement instanceof MySQLShowDatabasesStatement) {
             return PrivilegeType.SHOW_DB;
         }
+        if (sqlStatement instanceof DMLStatement) {
+            return getDMLPrivilege(sqlStatement);
+        }
+        if (sqlStatement instanceof DDLStatement) {
+            return getDDLPrivilege(sqlStatement);
+        }
         // TODO add more Privilege and SQL statement mapping
         return null;
     }
+
+    private PrivilegeType getDMLPrivilege(final SQLStatement sqlStatement) {
+        if (sqlStatement instanceof SelectStatement) {
+            return PrivilegeType.SELECT;
+        }
+        if (sqlStatement instanceof InsertStatement) {
+            return PrivilegeType.INSERT;
+        }
+        if (sqlStatement instanceof UpdateStatement) {
+            return PrivilegeType.UPDATE;
+        }
+        if (sqlStatement instanceof DeleteStatement) {
+            return PrivilegeType.DELETE;
+        }
+        return null;
+    }
+
+    private PrivilegeType getDDLPrivilege(final SQLStatement sqlStatement) {
+        if (sqlStatement instanceof AlterDatabaseStatement) {
+            return PrivilegeType.ALTER_ANY_DATABASE;
+        }
+        if (sqlStatement instanceof AlterTableStatement) {
+            return PrivilegeType.ALTER;
+        }
+        if (sqlStatement instanceof CreateDatabaseStatement) {
+            return PrivilegeType.CREATE_DATABASE;
+        }
+        if (sqlStatement instanceof CreateTableStatement) {
+            return PrivilegeType.CREATE_TABLE;
+        }
+        if (sqlStatement instanceof CreateFunctionStatement) {
+            return PrivilegeType.CREATE_FUNCTION;
+        }
+        if (sqlStatement instanceof DropTableStatement || sqlStatement 
instanceof DropDatabaseStatement) {
+            return PrivilegeType.DROP;
+        }
+        if (sqlStatement instanceof TruncateStatement) {
+            return PrivilegeType.TRUNCATE;
+        }
+        return null;
+    }
     
     @Override
     public int getOrder() {