Collection을 활용한 코드 개선 #1 - 과일 명 리턴

public String getFruit(){

    int random = RandomInt();

    int i = random % 4;

    switch(i){
    
        case 0:
            return "Grape";
        case 1:
            return "Apple";
        case 2:
            return "Orange";
        case 3:
            return "Banana";
    }
}

<개선 전>

 

public String getFruit(){

    String fruits[] = {"Grape", "Apple", "Orange", "Banana"};
    
    int random = RandomInt();
    
    int i = random % fruits.getLength();
    
    return fruits[i];
}

<개선 후>

 

 

비교문에서 코드 패턴이 일정하다면 Collection을 활용하여 코드를 개선할 수 있다.
예제는 랜덤으로 과일 명을 돌려보내는 함수이다. List를 활용하여 코드를 개선할 수 있다.