【8.1.1】【db】更新数据范围AOP和注解

kernel-d-db/db-api/src/main/java/cn/stylefeng/roses/kernel/db/api/constants/DbConstants.java

@@ -92,4 +92,9 @@ public interface DbConstants {
      */
     String DRUID_WEB_STAT_FILTER_PROFILE_ENABLE = "true";
 
+    /**
+     * 数据范围控制的AOP顺序
+     */
+    int DATA_SCOPE_AOP_ORDER = 100;
+
 }

kernel-d-db/db-sdk-mp/pom.xml

@@ -32,6 +32,12 @@
             <version>${roses.version}</version>
         </dependency>
 
+        <!-- aop -->
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-aop</artifactId>
+        </dependency>
+
     </dependencies>
 
 </project>

kernel-d-db/db-sdk-mp/src/main/java/cn/stylefeng/roses/kernel/db/mp/datascope/UserRoleDataScopeApi.java

@@ -0,0 +1,21 @@
+package cn.stylefeng.roses.kernel.db.mp.datascope;
+
+import cn.stylefeng.roses.kernel.db.mp.datascope.config.DataScopeConfig;
+
+/**
+ * 获取用户角色的数据范围
+ *
+ * @author fengshuonan
+ * @since 2024-03-01 13:39
+ */
+public interface UserRoleDataScopeApi {
+
+    /**
+     * 获取当前用户的数据范围
+     *
+     * @author fengshuonan
+     * @since 2024-03-01 13:41
+     */
+    DataScopeConfig getUserRoleDataScopeConfig();
+
+}

kernel-d-db/db-sdk-mp/src/main/java/cn/stylefeng/roses/kernel/db/mp/datascope/annotations/DataScope.java

@@ -0,0 +1,50 @@
+/*
+ * Copyright [2020-2030] [https://www.stylefeng.cn]
+ *
+ * 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.
+ *
+ * Guns采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
+ *
+ * 1.请不要删除和修改根目录下的LICENSE文件。
+ * 2.请不要删除和修改Guns源码头部的版权声明。
+ * 3.请保留源码和相关描述文件的项目出处，作者声明等。
+ * 4.分发源码时候，请注明软件出处 https://gitee.com/stylefeng/guns
+ * 5.在修改包名，模块名称，项目代码等时，请注明软件出处 https://gitee.com/stylefeng/guns
+ * 6.若您的项目无法满足以上几点，可申请商业授权
+ */
+package cn.stylefeng.roses.kernel.db.mp.datascope.annotations;
+
+import java.lang.annotation.*;
+
+/**
+ * 数据范围过滤的注解
+ *
+ * @author fengshuonan
+ * @since 2024-03-01 12:39
+ */
+@Inherited
+@Retention(RetentionPolicy.RUNTIME)
+@Target({ElementType.METHOD})
+public @interface DataScope {
+
+    /**
+     * 组织机构id字段的名称
+     */
+    String orgIdFieldName() default "";
+
+    /**
+     * 用户id字段的名称
+     */
+    String userIdFieldName() default "";
+
+}

kernel-d-db/db-sdk-mp/src/main/java/cn/stylefeng/roses/kernel/db/mp/datascope/aop/DataScopeAop.java

@@ -0,0 +1,115 @@
+/*
+ * Copyright [2020-2030] [https://www.stylefeng.cn]
+ *
+ * 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.
+ *
+ * Guns采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
+ *
+ * 1.请不要删除和修改根目录下的LICENSE文件。
+ * 2.请不要删除和修改Guns源码头部的版权声明。
+ * 3.请保留源码和相关描述文件的项目出处，作者声明等。
+ * 4.分发源码时候，请注明软件出处 https://gitee.com/stylefeng/guns
+ * 5.在修改包名，模块名称，项目代码等时，请注明软件出处 https://gitee.com/stylefeng/guns
+ * 6.若您的项目无法满足以上几点，可申请商业授权
+ */
+package cn.stylefeng.roses.kernel.db.mp.datascope.aop;
+
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.extra.spring.SpringUtil;
+import cn.stylefeng.roses.kernel.db.api.constants.DbConstants;
+import cn.stylefeng.roses.kernel.db.mp.datascope.UserRoleDataScopeApi;
+import cn.stylefeng.roses.kernel.db.mp.datascope.annotations.DataScope;
+import cn.stylefeng.roses.kernel.db.mp.datascope.config.DataScopeConfig;
+import cn.stylefeng.roses.kernel.db.mp.datascope.holder.DataScopeHolder;
+import lombok.extern.slf4j.Slf4j;
+import org.aspectj.lang.ProceedingJoinPoint;
+import org.aspectj.lang.Signature;
+import org.aspectj.lang.annotation.Around;
+import org.aspectj.lang.annotation.Aspect;
+import org.aspectj.lang.annotation.Pointcut;
+import org.aspectj.lang.reflect.MethodSignature;
+import org.springframework.core.Ordered;
+
+import java.lang.reflect.Method;
+
+
+/**
+ * 数据范围切换的AOP
+ *
+ * @author fengshuonan
+ * @since 2024-03-01 12:47
+ */
+@Aspect
+@Slf4j
+public class DataScopeAop implements Ordered {
+
+    @Pointcut(value = "@annotation(cn.stylefeng.roses.kernel.db.mp.datascope.annotations.DataScope)")
+    private void cutService() {
+
+    }
+
+    @Around("cutService()")
+    public Object around(ProceedingJoinPoint point) throws Throwable {
+
+        // 获取被aop拦截的方法
+        Signature signature = point.getSignature();
+        MethodSignature methodSignature = null;
+        if (!(signature instanceof MethodSignature)) {
+            throw new IllegalArgumentException("该注解只能用于方法");
+        }
+        methodSignature = (MethodSignature) signature;
+        Object target = point.getTarget();
+        Method currentMethod = target.getClass().getMethod(methodSignature.getName(), methodSignature.getParameterTypes());
+
+        // 获取注解
+        DataScope datasource = currentMethod.getAnnotation(DataScope.class);
+
+        // 获取当前用户拥有的数据范围
+        UserRoleDataScopeApi userRoleDataScopeApi = SpringUtil.getBean(UserRoleDataScopeApi.class);
+        DataScopeConfig userRoleDataScopeConfig = userRoleDataScopeApi.getUserRoleDataScopeConfig();
+
+        // 如果有单独配置特定的字段，以注解单独配置的字段为主
+        String userIdFieldName = datasource.userIdFieldName();
+        if (StrUtil.isNotBlank(userIdFieldName)) {
+            userRoleDataScopeConfig.setUserIdFieldName(userIdFieldName);
+        }
+
+        String orgIdFieldName = datasource.orgIdFieldName();
+        if (StrUtil.isNotBlank(orgIdFieldName)) {
+            userRoleDataScopeConfig.setOrgIdFieldName(orgIdFieldName);
+        }
+
+        try {
+            // 放入到临时上下文中
+            DataScopeHolder.set(userRoleDataScopeConfig);
+
+            // 执行原有业务逻辑
+            return point.proceed();
+        } finally {
+            // 清空数据范围的上下文环境
+            DataScopeHolder.remove();
+        }
+    }
+
+    /**
+     * aop的顺序要早于spring的事务
+     *
+     * @author fengshuonan
+     * @since 2020/10/31 22:55
+     */
+    @Override
+    public int getOrder() {
+        return DbConstants.DATA_SCOPE_AOP_ORDER;
+    }
+
+}