update 3.3.56

data authorization support union all sql

Author: xlorne

example/example-application/pom.xml

@@ -5,7 +5,7 @@
     <parent>
         <artifactId>springboot-example</artifactId>
         <groupId>com.codingapi.springboot</groupId>
-        <version>3.3.55</version>
+        <version>3.3.56</version>
     </parent>
     <modelVersion>4.0.0</modelVersion>
 

example/example-domain/pom.xml

@@ -5,7 +5,7 @@
     <parent>
         <artifactId>springboot-example</artifactId>
         <groupId>com.codingapi.springboot</groupId>
-        <version>3.3.55</version>
+        <version>3.3.56</version>
     </parent>
     <modelVersion>4.0.0</modelVersion>
 

example/example-infra-flow/pom.xml

@@ -5,7 +5,7 @@
     <parent>
         <artifactId>springboot-example</artifactId>
         <groupId>com.codingapi.springboot</groupId>
-        <version>3.3.55</version>
+        <version>3.3.56</version>
     </parent>
     <modelVersion>4.0.0</modelVersion>
 

example/example-infra-jpa/pom.xml

@@ -5,7 +5,7 @@
     <parent>
         <artifactId>springboot-example</artifactId>
         <groupId>com.codingapi.springboot</groupId>
-        <version>3.3.55</version>
+        <version>3.3.56</version>
     </parent>
     <modelVersion>4.0.0</modelVersion>
 

example/example-server/pom.xml

@@ -5,7 +5,7 @@
     <parent>
         <artifactId>springboot-example</artifactId>
         <groupId>com.codingapi.springboot</groupId>
-        <version>3.3.55</version>
+        <version>3.3.56</version>
     </parent>
     <modelVersion>4.0.0</modelVersion>
 

example/pom.xml

@@ -17,7 +17,7 @@
     </parent>
 
     <artifactId>springboot-example</artifactId>
-    <version>3.3.55</version>
+    <version>3.3.56</version>
 
     <name>springboot-example</name>
     <description>springboot-example project for Spring Boot</description>

pom.xml

@@ -15,7 +15,7 @@
 
     <groupId>com.codingapi.springboot</groupId>
     <artifactId>springboot-parent</artifactId>
-    <version>3.3.55</version>
+    <version>3.3.56</version>
 
     <url>https://github.com/codingapi/springboot-framewrok</url>
     <name>springboot-parent</name>

springboot-starter-data-authorization/pom.xml

@@ -6,7 +6,7 @@
     <parent>
         <groupId>com.codingapi.springboot</groupId>
         <artifactId>springboot-parent</artifactId>
-        <version>3.3.55</version>
+        <version>3.3.56</version>
     </parent>
 
     <artifactId>springboot-starter-data-authorization</artifactId>

springboot-starter-data-authorization/src/main/java/com/codingapi/springboot/authorization/enhancer/DataPermissionSQLEnhancer.java

@@ -8,12 +8,10 @@
 import net.sf.jsqlparser.parser.CCJSqlParserUtil;
 import net.sf.jsqlparser.schema.Table;
 import net.sf.jsqlparser.statement.Statement;
-import net.sf.jsqlparser.statement.select.FromItem;
-import net.sf.jsqlparser.statement.select.Join;
-import net.sf.jsqlparser.statement.select.PlainSelect;
-import net.sf.jsqlparser.statement.select.Select;
+import net.sf.jsqlparser.statement.select.*;
 
 import java.sql.SQLException;
+import java.util.List;
 
 /**
  * 数据权限 SQL 增强器
@@ -45,8 +43,7 @@ public String getNewSQL() throws SQLException {
             if (statement instanceof Select) {
                 tableColumnAliasHolder.holderAlias();
                 Select select = (Select) statement;
-                PlainSelect plainSelect = select.getPlainSelect();
-                this.enhanceDataPermissionInSelect(plainSelect);
+                this.deepMatch(select);
                 return statement.toString();
             }
         } catch (Exception e) {
@@ -55,6 +52,21 @@ public String getNewSQL() throws SQLException {
         return sql;
     }
 
+
+    private void deepMatch(Select select) throws Exception {
+        if (select instanceof PlainSelect) {
+            PlainSelect plainSelect = select.getPlainSelect();
+            this.enhanceDataPermissionInSelect(plainSelect);
+        }
+        if (select instanceof SetOperationList) {
+            SetOperationList setOperationList = select.getSetOperationList();
+            List<Select> selectList = setOperationList.getSelects();
+            for (Select selectItem : selectList) {
+                this.deepMatch(selectItem.getPlainSelect());
+            }
+        }
+    }
+
     public TableColumnAliasContext getTableAlias() {
         return tableColumnAliasHolder.getAliasContext();
     }

springboot-starter-data-authorization/src/main/java/com/codingapi/springboot/authorization/enhancer/TableColumnAliasHolder.java

@@ -29,9 +29,22 @@ public TableColumnAliasHolder(Statement statement) {
      */
     public void holderAlias() {
         Select select = (Select) statement;
-        PlainSelect plainSelect = select.getPlainSelect();
-        this.searchSubSelect(null, plainSelect);
-        aliasContext.columnKeyToMap();
+        this.deepSearch(select);
+    }
+
+
+    private void deepSearch(Select select) {
+        if (select instanceof PlainSelect) {
+            PlainSelect plainSelect = select.getPlainSelect();
+            this.searchSubSelect(null, plainSelect);
+            aliasContext.columnKeyToMap();
+        } else if (select instanceof SetOperationList) {
+            SetOperationList setOperationList = select.getSetOperationList();
+            List<Select> selectList = setOperationList.getSelects();
+            for (Select selectItem : selectList) {
+                this.deepSearch(selectItem);
+            }
+        }
     }