给查询出来的实训按照状态排序

pull/1/head
xiaoCJ 2 years ago
parent 1111dfbb0c
commit bef0b984a1

@ -18,10 +18,7 @@ import org.springframework.util.StringUtils;
import java.time.LocalDateTime; import java.time.LocalDateTime;
import java.time.ZoneId; import java.time.ZoneId;
import java.util.ArrayList; import java.util.*;
import java.util.Date;
import java.util.List;
import java.util.Map;
@Service @Service
public class TrainingService { public class TrainingService {
@ -125,8 +122,12 @@ public class TrainingService {
for (Member member : members) { for (Member member : members) {
String trainingId = member.getTrainingId(); String trainingId = member.getTrainingId();
Training training = trainingMapper.selectByPrimaryKey(trainingId); Training training = trainingMapper.selectByPrimaryKey(trainingId);
trainingList.add(training); if (training != null) {
trainingList.add(training);
}
} }
List<String> statusOrder = Arrays.asList("ONGOING", "NOT_STARTED", "FINISHED");
trainingList.sort(Comparator.comparingInt(training -> statusOrder.indexOf(training.getStatus())));
return new PageInfo<>(trainingList); return new PageInfo<>(trainingList);
} }
return new PageInfo<>(); return new PageInfo<>();

Loading…
Cancel
Save