1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
| @Override public List<OneSubject> getOneTwoSubject() { QueryWrapper<EduSubject> wrapperOne = new QueryWrapper<>(); wrapperOne.eq("parent_id","0"); List<EduSubject> oneSubjectList = baseMapper.selectList(wrapperOne);
QueryWrapper<EduSubject> wrapperTwo = new QueryWrapper<>(); wrapperOne.ne("parent_id","0"); List<EduSubject> twoSubjectList = baseMapper.selectList(wrapperTwo);
List<OneSubject> finalSubjectList = new ArrayList<>(); for (int i = 0; i <oneSubjectList.size() ; i++) { EduSubject eduSubject = oneSubjectList.get(i); OneSubject oneSubject = new OneSubject(); BeanUtils.copyProperties(eduSubject,oneSubject); finalSubjectList.add(oneSubject); List<TwoSubject> twoFinalSubjectList = new ArrayList<>(); for (int j = 0; j < twoSubjectList.size(); j++) { EduSubject eduSubject2 = twoSubjectList.get(j); if (eduSubject2.getParentId().equals(eduSubject.getId())) { TwoSubject twoSubject = new TwoSubject(); BeanUtils.copyProperties(eduSubject2,twoSubject); twoFinalSubjectList.add(twoSubject); } } oneSubject.setChildren(twoFinalSubjectList); } return finalSubjectList; }
|