diagnosis/src/main/java/edu/math/diagnosis/service/SubjectAbilityService.java

package edu.math.diagnosis.service;

import edu.math.diagnosis.dao.SubjectAbilityRepo;
import edu.math.diagnosis.entity.SubjectAbility;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@Service
public class SubjectAbilityService {

    @Resource
    private SubjectAbilityRepo subjectAbilityRepo;

    public SubjectAbility add(SubjectAbility ability) {
        return subjectAbilityRepo.save(ability);
    }

    public List<SubjectAbility> list(Long subjectId) {
        return subjectAbilityRepo.findAllBySubjectId(subjectId);
    }

    public Map<String, String> map(Long subjectId) {
        List<SubjectAbility> abilities = list(subjectId);
        // Map<code,Name>
        return abilities.stream().collect(Collectors.toMap(SubjectAbility::getCode, SubjectAbility::getName));
    }
}