排版支持多个

src/main/java/com/gogirl/application/store/store/impl/StoreClassesServiceImpl.java

@@ -11,9 +11,9 @@ import com.gogirl.infrastructure.mapper.store.store.StoreClassesMapper;
 import lombok.AllArgsConstructor;
 import org.springframework.stereotype.Service;
 
-import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
 
 @Service
 @AllArgsConstructor
@@ -26,23 +26,10 @@ public class StoreClassesServiceImpl extends ServiceImpl<StoreClassesMapper, Sto
         List<StoreClasses> classList = this.list(new LambdaQueryWrapper<StoreClasses>()
                 .eq(StoreClasses::getDepartmentId, departmentId));
 
-        //处理一个对应关系只返回012
-        Map<Integer, Integer> map = new HashMap<>();
-        map.put(0, 0);
-        for (int i = 0; i < classList.size(); i++) {
-            StoreClasses item = classList.get(i);
-            if (item.getName().equals("早班")) {
-                map.put(item.getId(), 1);
-            } else if (item.getName().equals("晚班")) {
-                map.put(item.getId(), 2);
-            }
-        }
+        Map<Integer, StoreClasses> classesMap = classList.stream().collect(Collectors.toMap(StoreClasses::getId, storeClasses -> storeClasses));
 
         List<StoreClassesTechnician> lists = storeClassesTechnicianService.listClassesTechnicianDetail(technicianId, month);
-        //替换排班的class
-        for (StoreClassesTechnician item : lists) {
-            item.setClasses(map.get(item.getClasses()));
-        }
+        lists.forEach(item -> item.setClassesManage(classesMap.get(item.getClasses())));
         return lists;
     }
 }