查询练习册修改

ssj-kmt-exercise-service/src/main/java/com/ssj/service/kmt/dto/ExerciseBookDto.java

@@ -2,6 +2,7 @@ package com.ssj.service.kmt.dto;
 
 import com.alibaba.fastjson.annotation.JSONField;
 import com.ssj.framework.core.util.StringUtil;
+import com.ssj.service.kmt.dto.exercise.ExerciseBookPartDto;
 import com.ssj.service.kmt.dto.exercise.ExerciseBookTypeDto;
 import com.ssj.service.kmt.dto.exercise.ExerciseBookYearDto;
 
@@ -159,12 +160,10 @@ public class ExerciseBookDto extends ExerciseBookBaseDto implements Serializable
     public void addBookYear (int bookYear, int bookType, String exerciseBookId){
         ExerciseBookYearDto dto = new ExerciseBookYearDto(bookYear);
         dto.addBookType(bookType, exerciseBookId);
+        dto.addBookPartType(bookType);
         this.bookYearList.add(dto);
     }
 
-    public void addBookPartType (){
-//        this.bookYearList.get
-    }
 
     public void addUnit(String unitId, String unitName, String courseId, String courseName){
         boolean isExist = false;

ssj-kmt-exercise-service/src/main/java/com/ssj/service/kmt/dto/exercise/ExerciseBookYearDto.java

@@ -63,4 +63,19 @@ public class ExerciseBookYearDto implements Serializable {
     public void setBookPartList(List<ExerciseBookPartDto> bookPartList) {
         this.bookPartList = bookPartList;
     }
+
+    public void addBookPartType (int bookType){
+        if (bookType == 3){
+            List<ExerciseBookPartDto> list = new ArrayList<>();
+            ExerciseBookPartDto temp = new ExerciseBookPartDto();
+            temp.setBookPartType(1);
+            temp.setBookPartName("全部");
+            list.add(temp);
+            temp = new ExerciseBookPartDto();
+            temp.setBookPartType(2);
+            temp.setBookPartName("不同");
+            list.add(temp);
+            this.setBookPartList(list);
+        }
+    }
 }

ssj-mybatis-service-impl/src/main/java/com/ssj/service/kmt/exercise/KmtExerciseMapperServiceImpl.java

@@ -1089,6 +1089,7 @@ public class KmtExerciseMapperServiceImpl implements KmtExerciseMapperService {
                         resultBookDto.addBookYear(bookDto.getBookYear(), bookDto.getBookType(), bookDto.getExerciseBookId());
                     } else {
                         resultBookYearDto.addBookType(bookDto.getBookType(), bookDto.getExerciseBookId());
+                        resultBookYearDto.addBookPartType(bookDto.getBookType());
                     }
                 }
             }