diff --git a/src/main/java/com/sztzjy/financial_bigdata/controller/tea/UserController.java b/src/main/java/com/sztzjy/financial_bigdata/controller/tea/UserController.java index 25ef68c..34a5502 100644 --- a/src/main/java/com/sztzjy/financial_bigdata/controller/tea/UserController.java +++ b/src/main/java/com/sztzjy/financial_bigdata/controller/tea/UserController.java @@ -223,17 +223,20 @@ public class UserController { stuUser.setCreateTime(new Date()); users.add(stuUser); - stuClass.setClassId(zyUserInfo.getClassId().toString()); - stuClass.setClassName(zyUserInfo.getClassName()); - stuClass.setSchoolName(zyUserInfo.getSchoolName()); - stuClass.setCreateTime(new Date()); - map.put(zyUserInfo.getClassId().toString(), stuClass); + if (roleId == 4) { + stuClass.setClassId(zyUserInfo.getClassId().toString()); + stuClass.setClassName(zyUserInfo.getClassName()); + stuClass.setSchoolName(zyUserInfo.getSchoolName()); + stuClass.setCreateTime(new Date()); + map.put(zyUserInfo.getClassId().toString(), stuClass); + } } - for (Map.Entry stringStuClassEntry : map.entrySet()) { - StuClass stuClass = stringStuClassEntry.getValue(); - classMapper.insert(stuClass); + if (!map.isEmpty()) { + for (Map.Entry stringStuClassEntry : map.entrySet()) { + StuClass stuClass = stringStuClassEntry.getValue(); + classMapper.insert(stuClass); + } } - if (users.isEmpty()) { return "账号已全部存在"; }