用户常用模块收藏功能

portal-manager/src/main/java/com/mortals/xhx/module/model/ModelController.java

@@ -114,17 +114,69 @@ public class ModelController extends BaseJsonBodyController {
         return ret;
     }
 
-    private List<Map<String,List<ModelFeignVO>>> doGroup(List<ModelFeignVO> result){
+    /**
+     * 根据站点id查询模块列表
+     * @param query
+     * @return
+     */
+    @PostMapping({"list/test"})
+    public Rest<Object> getListT(@RequestBody ModelVO query) {
+        Rest<Object> ret = new Rest();
+        Map<String, Object> model = new HashMap();
+        String busiDesc = "查询站点编排" ;
+        int code=1;
+        try {
+            if(this.getCurUser()==null){
+                throw new AppException("未登录或者登录过期");
+            }
+            ModelPdu modelPdu = new ModelPdu();
+            modelPdu.setSiteId(query.getSiteId());
+            if(this.getCurUser().getId()==1 && StringUtils.isEmpty(this.getCurUser().getSiteIds())){
+                modelPdu.setIsAdmin(1);
+            }else {
+                modelPdu.setIsAdmin(0);
+            }
+            //ApiResp<List<ModelFeignVO>> apiResp = apiModelFeign.getModelBySiteId(modelPdu);
+            String resp = apiModelFeign.getModelBySiteId(modelPdu);
+            ApiResp<JSONObject> apiResp = JSON.parseObject(resp, ApiResp.class);
+            if (apiResp.getCode() != YesNoEnum.YES.getValue()) {
+                throw new AppException("获取站点列表信息失败：" + apiResp.getMsg());
+            }
+            Map<Long,Long> map = roleModelService.getModelIdsUserId(this.getCurUser().getId());
+            List<ModelFeignVO> modelFeignVOList = JSONObject.parseArray(apiResp.getData().get("data").toString(),ModelFeignVO.class);
+            List<ModelFeignVO> list = modelFeignVOList.stream().filter(item->map.containsKey(item.getId())).collect(Collectors.toList());
+            if(this.getCurUser().getId()==1){
+                model.put("data", doGroup(modelFeignVOList));
+            }else {
+                model.put("data", doGroup(list));
+            }
+            model.put("message_info", busiDesc + "成功");
+            this.recordSysLog(this.request, busiDesc + " 【成功】");
+        } catch (Exception var9) {
+            code = -1;
+            this.doException(this.request, busiDesc, model, var9);
+        }
+        ret.setCode(code);
+        ret.setData(model);
+        ret.setMsg(model.get("message_info") == null ? "" : model.remove("message_info").toString());
+        return ret;
+    }
+
+
+    private List<Map<String,Object>> doGroup(List<ModelFeignVO> result){
         Map<Integer,List<ModelFeignVO>> groupMap = result.stream().collect(Collectors.groupingBy(t -> t.getType()));
-        List<Map<String,List<ModelFeignVO>>> mapList = new ArrayList<>();
-        Map<String,List<ModelFeignVO>> jcMap = new HashMap<>();
-        jcMap.put(ModelTypeEnum.JC.getDesc(),groupMap.get(ModelTypeEnum.JC.getValue()));
+        List<Map<String,Object>> mapList = new ArrayList<>();
+        Map<String,Object> jcMap = new HashMap<>();
+        jcMap.put("type",ModelTypeEnum.JC.getDesc());
+        jcMap.put("list",groupMap.get(ModelTypeEnum.JC.getValue()));
         mapList.add(jcMap);
-        Map<String,List<ModelFeignVO>> g2gMap = new HashMap<>();
-        g2gMap.put(ModelTypeEnum.G2G.getDesc(),groupMap.get(ModelTypeEnum.G2G.getValue()));
+        Map<String,Object> g2gMap = new HashMap<>();
+        jcMap.put("type",ModelTypeEnum.G2G.getDesc());
+        jcMap.put("list",groupMap.get(ModelTypeEnum.G2G.getValue()));
         mapList.add(g2gMap);
-        Map<String,List<ModelFeignVO>> g2cMap = new HashMap<>();
-        g2cMap.put(ModelTypeEnum.G2C.getDesc(),groupMap.get(ModelTypeEnum.G2C.getValue()));
+        Map<String,Object> g2cMap = new HashMap<>();
+        jcMap.put("type",ModelTypeEnum.G2C.getDesc());
+        jcMap.put("list",groupMap.get(ModelTypeEnum.G2C.getValue()));
         mapList.add(g2cMap);
         return mapList;
     }