更新文件 WechatScanLoginController.java

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

@@ -520,4 +520,63 @@ public class WechatScanLoginController implements WechatScanLoginControllerAPI {
         jsonObject.put("userhead", AesUtils.encrypt(String.valueOf(user.get(0).getId())));
         return CommonResult.ok(jsonObject);
     }
+
+    @Override
+	@PassToken
+	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("无权限查询");
+		}
+		QueryWrapper<SmartGrade> queryWrapper1 = new QueryWrapper<>();
+		queryWrapper1.eq("deleted", 0);
+        List<SmartGrade> grades = smartGradeService.list(queryWrapper1);
+		List<GradeBook> 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);
+		}
+		return CommonResult.ok(gradeBookList);
+	}
+    
 }