小E数字提示问题

cloud-web-manage/src/main/java/com/yizhi/application/dashboard/controller/PromptStatisticsController.java

@@ -241,11 +241,11 @@ public class PromptStatisticsController {
             // 设置状态为未完成(进行中)的考试
             param.setState(1);
             param.setPageNo(1);
-            param.setPageSize(1000); // 设置一个较大的数值以获取所有未完成的考试
+            param.setPageSize(1); // 设置一个较大的数值以获取所有未完成的考试
 
             MyExamPageVO examData = myExamClient.getMyExamList(param);
             // 返回未完成考试的数量
-            return examData != null && examData.getExamVOS() != null ? examData.getExamVOS().size() : 0;
+            return examData != null  ? examData.getPageTotal() : 0;
         } catch (Exception e) {
             LOGGER.error("获取培训测试数量失败", e);
             return 0;

cloud-web-student/src/main/java/com/yizhi/application/promptStatistics/PromptStatisticsController.java

@@ -239,14 +239,22 @@ public class PromptStatisticsController {
             MyExamParameterVO param = new MyExamParameterVO();
             param.setSiteId(context.getSiteId());
             param.setAccountId(context.getAccountId());
+            List<Long> ids = context.getRelationIds();
+            if (null != ids) {
+                param.setIds(ids);
+            } else {
+                ids = new ArrayList<Long>();
+                ids.add(context.getAccountId());
+                param.setIds(ids);
+            }
             // 设置状态为未完成(进行中)的考试
             param.setState(1);
             param.setPageNo(1);
-            param.setPageSize(1000); // 设置一个较大的数值以获取所有未完成的考试
+            param.setPageSize(1); // 设置一个较大的数值以获取所有未完成的考试
             
             MyExamPageVO examData = myExamClient.getMyExamList(param);
             // 返回未完成考试的数量
-            return examData != null && examData.getExamVOS() != null ? examData.getExamVOS().size() : 0;
+            return examData != null ? examData.getPageTotal() : 0;
         } catch (Exception e) {
             LOGGER.error("获取培训测试数量失败", e);
             return 0;