添加项目数据隔离逻辑

fastbee-framework/src/main/java/com/fastbee/framework/config/ResourcesConfig.java

@@ -3,6 +3,7 @@ package com.fastbee.framework.config;
 import java.util.concurrent.TimeUnit;
 
 import com.fastbee.framework.interceptor.LanguageInterceptor;
+import com.fastbee.framework.interceptor.ProjectDataIsolationInterceptor;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
@@ -32,6 +33,9 @@ public class ResourcesConfig implements WebMvcConfigurer
     @Resource
     private LanguageInterceptor languageInterceptor;
 
+    @Autowired
+    private ProjectDataIsolationInterceptor projectDataIsolationInterceptor;
+
     @Override
     public void addResourceHandlers(ResourceHandlerRegistry registry)
     {
@@ -55,6 +59,9 @@ public class ResourcesConfig implements WebMvcConfigurer
         //这里配置国际化拦截器的白名单
         registry.addInterceptor(languageInterceptor).addPathPatterns("/**").excludePathPatterns("/v2/api-docs",
                 "/tool/gen/**");
+        //配置项目数据隔离拦截器
+        registry.addInterceptor(projectDataIsolationInterceptor)
+                .addPathPatterns("/gis/groups/**", "/gis/legend/**","/gis/site/**");
     }
 
     /**

fastbee-framework/src/main/java/com/fastbee/framework/interceptor/ProjectDataIsolationInterceptor.java

@@ -0,0 +1,71 @@
+package com.fastbee.framework.interceptor;
+
+
+import cn.hutool.core.codec.Base64;
+import cn.hutool.json.JSON;
+import cn.hutool.json.JSONArray;
+import cn.hutool.json.JSONUtil;
+import com.alibaba.fastjson.JSONObject;
+import com.fastbee.common.holder.ProjectHolder;
+import com.fastbee.common.holder.ProjectInfo;
+import lombok.extern.slf4j.Slf4j;
+
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.stereotype.Component;
+import org.springframework.web.servlet.HandlerInterceptor;
+
+import javax.annotation.Nullable;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import javax.validation.constraints.NotNull;
+import java.util.Arrays;
+import java.util.List;
+import java.util.Objects;
+import java.util.stream.Collectors;
+
+
+/**
+ * @author mijiupro
+ */
+
+@Component
+@Slf4j
+public class ProjectDataIsolationInterceptor implements HandlerInterceptor {
+
+    @Override
+    public boolean preHandle(@NotNull HttpServletRequest request, @NotNull HttpServletResponse response, @NotNull Object handler) throws Exception {
+
+        log.info("进入项目数据隔离拦截器");
+        //放行预检请求
+        if ("OPTIONS".equalsIgnoreCase(Objects.requireNonNull(request.getMethod()))) {
+            return true;
+        }
+        if ("DELETE".equalsIgnoreCase(request.getMethod())) {
+            return true;
+        }
+        if ("POST".equalsIgnoreCase(request.getMethod())) {
+                return true;
+            }
+        if ("PUT".equalsIgnoreCase(request.getMethod())) {
+            return true;
+        }
+            //获取路径中的查询字符串参数
+            String projectIds = request.getParameter("projectIds");
+            if (StringUtils.isBlank(projectIds)) {
+                return true;
+            }
+            byte[] decode = Base64.decode(projectIds);
+            //转成List集合
+            String decodedStr = new String(decode);
+            JSONArray objects = JSONUtil.parseArray(decodedStr);
+            ProjectHolder.setProjectInfo(ProjectInfo.builder()
+                    .projectIdList(objects.toList(Long.class))
+                    .build());
+            return true;
+    }
+    @Override
+    public void afterCompletion(@NotNull HttpServletRequest request, @NotNull HttpServletResponse response, @NotNull Object handler, @Nullable Exception ex)  {
+        // 清理本地线程
+        ProjectHolder.clear();
+    }
+}