[Java] 제네릭을 활용한 성적관리 프로그램

수정중

package grademanagementsystem;
 
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.Iterator;
import java.util.Scanner;
import java.util.function.UnaryOperator;
 
public class GradeMagSystem {
    // 필드
    ArrayList<Student> list;
    Student stu;    
    
    // 생성자
    GradeMagSystem() {
        // ArrayList 생성        
        list = new ArrayList<>();
        stu = null;
    }
    
    // 1: 국영수 점수를 입력하면 총점, 평균을 자동 계산 후 내림차순으로 정렬
    public void input() {
        Scanner sc = new Scanner(System.in);
        
        stu = new Student();
                
        System.out.println("──────────────────");
        System.out.println("학생 정보를 입력합니다.");
        System.out.println("이름을 입력하세요.");
        System.out.println("──────────────────");
        System.out.print("　──> ");
        stu.setName(sc.next());
        
        System.out.println("──────────────────");
        System.out.println("국어 성적을 입력하세요.");
        System.out.println("──────────────────");
        System.out.print("　──> ");
        stu.setKor(sc.nextInt());
        
        System.out.println("──────────────────");
        System.out.println("영어 성적을 입력하세요.");
        System.out.println("──────────────────");
        System.out.print("　──> ");
        stu.setEng(sc.nextInt());
        
        System.out.println("──────────────────");
        System.out.println("수학 성적을 입력하세요.");
        System.out.println("──────────────────");
        System.out.print("　──> ");
        stu.setMath(sc.nextInt());
        
        // 총점, 평균 계산
        stu.setTotal(stu.getKor() + stu.getEng() + stu.getMath());
        stu.setAvg(stu.getTotal() / 3.f);
        
        // 국영수, 총점, 평균까지 담긴 데이터 삽입
        list.add(stu);
        
        System.out.println("──────────────────");
        System.out.println("학생 정보 등록 완료!");
        
        Descend Descend = new Descend();
        Collections.sort(list, Descend);
    }
    
    // 2: 학생 전체 출력
    public void showAllInfo() {
        // 학생 데이터가 1도 없는 경우
        if(list.isEmpty()) {
            emptyStudentMessage();
            return;
        }
        System.out.println("NAME /  KOR  /  ENG  /  MATH  / TOTAL / AVERAGE");
        for (Student stu : list)
            System.out.println(stu);
    }
    
    // 3번 기능 - 검색
    public void search() {
        // 학생 데이터가 1도 없는 경우
        if(list.isEmpty()) {
            emptyStudentMessage();
            return;
        }
 
        Scanner sc = new Scanner(System.in);
        
        System.out.println("──────────────────");
        System.out.println("학생 이름으로 검색합니다.");
        System.out.println("찾고자 하는 이름을 입력하세요.");
        System.out.println("──────────────────");
        System.out.print("　──> ");
        String wantToName = sc.next();
 
        // 찾는 이름이 있으면 그 학생에 대한 정보를 출력한다.
        showTargetInfo(wantToName);
    }
    
    public int showTargetInfo(String wantToName) {
        Iterator<Student> it = list.iterator();
        
        while(it.hasNext()) {
            Student target = it.next();
            String name = target.getName();
 
            if(name.equals(wantToName)) {
                int idx = list.indexOf(target);
 
                // 출력
                System.out.println("──────────────────");
                System.out.println("NAME /  KOR  /  ENG  /  MATH  / TOTAL / AVERAGE");
                System.out.println(list.get(idx));
                
                return idx;
                
            } else if(list.contains(it)) {
                System.out.println("──────────────────");
                System.out.println("찾고자 하는 이름이 없습니다.");
            }
        }
        
        return -1;
    }
    
    // 4: 학생 데이터 삭제
    public void deleteInfo() {
        // 학생 데이터가 1도 없는 경우
        if(list.isEmpty()) {
            emptyStudentMessage();
            return;
        }
        
        Scanner sc = new Scanner(System.in);
        System.out.println("──────────────────");
        System.out.println("학생 정보를 삭제합니다.");
        System.out.println("지우고자 하는 이름을 입력하세요.");
        System.out.println("──────────────────");
        System.out.print("　──> ");
        String wantToName = sc.next();
        
        int idx = showTargetInfo(wantToName);
        
        System.out.println("──────────────────");
        System.out.println("현재 " + wantToName + "의 데이터가 존재합니다.");
        System.out.println("정말 삭제하시겠습니까?");
        System.out.println("1. 네" + "\t" + "2. 아니오");
        System.out.println("──────────────────");
        System.out.print("　──> ");
        int button = sc.nextInt();
        
        switch(button)
        {
        case 1:
            list.remove(idx);
            
            System.out.println("──────────────────");
            System.out.println("삭제 완료!");
        case 2:
            System.out.println("──────────────────");
            System.out.println("메인 화면으로 돌아갑니다.");
            return;
        default:
            break;
        }
    }    // end of deleteInfo()
    
    
    // 학생 성적 수정
    public void correctGrade() {
        // 학생 데이터가 1도 없는 경우
        if (list.isEmpty()) {
            emptyStudentMessage();
            return;
        }
        
        Scanner sc = new Scanner(System.in);
        int temp = 0;
        
        System.out.println("──────────────────");
        System.out.println("학생 이름으로 검색합니다.");
        System.out.println("찾고자 하는 이름을 입력하세요.");
        System.out.println("──────────────────");
        System.out.print("　──> ");
        String wantToName = sc.next();
 
        int idx = showTargetInfo(wantToName);
 
        System.out.println("──────────────────");
        System.out.println("현재 " + wantToName + "의 데이터가 다음과 같습니다.");
        System.out.println("어떤 과목의 성적을 수정하시겠습니까?");
        System.out.println("1. 국어" + "\t" + "2. 영어" + "\t" + "3. 수학");
        System.out.println("──────────────────");
        System.out.print("　──> ");
        int button = sc.nextInt();
 
        switch (button) {
        case 1:
            System.out.println("──────────────────");
            System.out.println("국어 성적을 수정합니다.");
            System.out.println("수정할 점수를 입력하세요.");
            System.out.println("──────────────────");
            System.out.print("　──> ");
            temp = sc.nextInt();
            list.get(idx).setKor(temp);
 
            break;
 
        case 2:
            System.out.println("──────────────────");
            System.out.println("영어 성적을 수정합니다.");
            System.out.println("수정할 점수를 입력하세요.");
            System.out.println("──────────────────");
            System.out.print("　──> ");
            temp = sc.nextInt();
            list.get(idx).setEng(temp);
            
            break;
 
        case 3:
            System.out.println("──────────────────");
            System.out.println("수학 성적을 수정합니다.");
            System.out.println("수정할 점수를 입력하세요.");
            System.out.println("──────────────────");
            System.out.print("　──> ");
            temp = sc.nextInt();
            list.get(idx).setMath(temp);
 
            break;
            
        default:
            System.out.println("──────────────────");
            System.out.println("잘못된 번호를 입력하였습니다.");
            System.out.println("다시 시도해주세요.");
            
            break;
        }
        
        // 총점, 평균 계산
        list.get(idx).setTotal(list.get(idx).getKor() + list.get(idx).getEng() + list.get(idx).getMath());
        list.get(idx).setAvg(list.get(idx).getTotal() / 3.f);
        
        Descend Descend = new Descend();
        Collections.sort(list, Descend);
        
        System.out.println("──────────────────");
        System.out.println("수정이 완료되었습니다!");
        System.out.println("확인해보삼ㅎㅎ");
    }
    
    public void emptyStudentMessage() {
        System.out.println("──────────────────");
        System.out.println("현재 등록된 학생이 1도 없습니다.");
        System.out.println("메인 화면으로 돌아갑니다.");
    }
}
 
//내림차순 정렬 메소드
class Descend implements Comparator<Student> {
    @Override
    public int compare(Student o1, Student o2) {
        return o2.compareTo(o1);
    }
}