更新文件 WechatScanLoginController.java

src/main/java/com/template/controller/WechatScanLoginController.java

@@ -651,8 +651,7 @@ public class WechatScanLoginController implements WechatScanLoginControllerAPI {
 
     @Override
     @PassToken
-    @DESRespondSecret(validated = true)
-    public CommonResult queryPhoneBook(Integer id) {
+    public CommonResult operatePhoneBook(Integer id) {
         if (id == null) {
             return CommonResult.fail("请传入id");
         }
@@ -706,6 +705,73 @@ public class WechatScanLoginController implements WechatScanLoginControllerAPI {
             gradeBook.setClassBookList(classBookList);
             gradeBookList.add(gradeBook);
         }
+        GlobalData.setAddressBook(gradeBookList);
+        return CommonResult.ok(gradeBookList);
+    }
+
+    @Override
+    @PassToken
+    @DESRespondSecret(validated = false)
+    public CommonResult queryPhoneBook(Integer id) {
+        if (id == null) {
+            return CommonResult.fail("请传入id");
+        }
+        SmartUser user = smartUserService.getSmartById(id);
+        if (user == null) {
+            return CommonResult.fail("未查到用户信息");
+        }
+        if (user.getId() < 3) {
+            return CommonResult.fail("无权限查询");
+        }
+        List<GradeBook> gradeBookList = null;
+        if (GlobalData.getAddressBook()==null){
+            QueryWrapper<SmartGrade> queryWrapper1 = new QueryWrapper<>();
+            queryWrapper1.eq("deleted", 0);
+            List<SmartGrade> grades = smartGradeService.list(queryWrapper1);
+            gradeBookList = new ArrayList<>();
+            for (int i = 0; i < grades.size(); i++) {
+                GradeBook gradeBook = new GradeBook();
+                gradeBook.setGradeName(grades.get(i).getName());
+                QueryWrapper<SmartClass> queryWrapper2 = new QueryWrapper<>();
+                queryWrapper2.eq("deleted", 0);
+                queryWrapper2.eq("grade_id", grades.get(i).getId());
+                List<SmartClass> classes = smartClassService.list(queryWrapper2);
+                List<ClassBook> classBookList = new ArrayList<>();
+                for (int j = 0; j < classes.size(); j++) {
+                    ClassBook classBook = new ClassBook();
+                    classBook.setClassName(classes.get(j).getName());
+                    QueryWrapper<SmartUser> queryWrapper3 = new QueryWrapper<>();
+                    queryWrapper3.eq("deleted", 0);
+                    queryWrapper3.eq("is_cancel", 0);
+                    queryWrapper3.eq("school_class", classes.get(j).getId());
+                    queryWrapper3.eq("identity_id", 2);
+                    List<SmartUser> users = smartUserService.list(queryWrapper3);
+                    List<BookUser> bookUserList = new ArrayList<>();
+                    for (int k = 0; k < users.size(); k++) {
+                        if (Objects.equals(users.get(k).getSchoolClass(), user.getSchoolClass())) {
+                            String ob = JSONObject.toJSONString(users.get(k));
+                            BookUser bookUser = JSONObject.parseObject(ob, BookUser.class);
+                            ;
+                            QueryWrapper<SmartUser> queryWrapper4 = new QueryWrapper<>();
+                            queryWrapper4.eq("deleted", 0);
+                            queryWrapper4.eq("is_cancel", 0);
+                            queryWrapper4.eq("affiliate", users.get(k).getId());
+                            queryWrapper4.eq("identity_id", 1);
+                            List<SmartUser> usersPrarent = smartUserService.list(queryWrapper4);
+                            bookUser.setUserParents(usersPrarent);
+                            bookUserList.add(bookUser);
+                        }
+                    }
+                    classBook.setBookUserList(bookUserList);
+                    classBookList.add(classBook);
+                }
+                gradeBook.setClassBookList(classBookList);
+                gradeBookList.add(gradeBook);
+            }
+            GlobalData.setAddressBook(gradeBookList);
+        } else {
+            gradeBookList = GlobalData.getAddressBook();
+        }
         return CommonResult.ok(gradeBookList);
     }