[Java] 코드 개선을 위한 Collection 사용

👊  코드 개선을 위한 Collection 사용

목차

1. Collection의 장단점
2. 실습 (1)
3. 실습 (2)

1. Collection의 장단점

위와 같이 클래스 A와 B가 있을 때, 간결성과 가독성면에서는 B가 우수합니다.

그러나, 힙 메모리 사용량이나 함수 실행 속도 면에서는 A가 더 우수합니다.

📌 따라서, 장/단점이 있기 때문에 어느 방식을 적용할지는 요구사항, 메모리 사용량, 가독성, CPU 연산량 등 다각도로 고려해야합니다.

2. 실습 (1)

📌switch ~ case ~ 문을 통해 작성한 현재 달의 날 수 출력 프로그램을 Collection을 사용해 개선해 봅시다.

개선 전

import java.util.*;
import java.text.SimpleDateFormat;

public class 코드개선을_위한_Collection_사용_윤년 {
    public static void main(String[] args) {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM");
        String date = sdf.format(new Date());
        int year = Integer.parseInt(date.substring(0,4));
        int month = Integer.parseInt(date.substring(5));
        int day= 0;

        switch(month) {
        case 1:
        case 3:
        case 5:
        case 7:
        case 8:
        case 12:
            day = 31;
            break;
        case 4:
        case 6:
        case 9:
        case 11:
            day = 30;
            break;
        case 2:
            day = (year % 4) == 0 && (year % 100) != 0 || (year % 400) == 0 ? 29 : 28;
        break;
        }

        System.out.println(day + " days for " + year + "-" + month);
    }
}

개선 후

import java.util.*;
import java.text.SimpleDateFormat;

public class 코드개선을_위한_Collection_사용_윤년 {
    public static void main(String[] args) {
        int[][] days = {{0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31},
            {0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31}};

        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM");
        String date = sdf.format(new Date());
        int year = Integer.parseInt(date.substring(0,4));
        int month = Integer.parseInt(date.substring(5));
        int idx = (year % 4) == 0 && (year % 100) != 0 || (year % 400) == 0 ? 0 : 1; 

        System.out.println(days[idx][month] + " days for " + year + "-" + month);
    }
}

코드가 간결해진 것을 확인 할 수 있습니다.

3. 실습 (2)

📌 안드로이드 코드에서 인텐트로 받아온 mode 값과 파라미터로 넘어오는 result에 따라 처리가 달라지는 코드를 Collection을 사용해 개선해 봅시다.

개선 전

중복 코드가 많이 보이고, 직관적이지 못합니다.

public void returnResult(boolean result) {
  String mode = getIntent().getExtras().getString("mode");
  Intent intent = new Intent();
  switch(mode) {
    case "register": {
      // setResult(resultCode, Intent)
            setResult(1000, intent);
      if (result) {
        // intent.putExtra(name, value);
        intent.putExtra("result", "true"); 
      } else {
        intent.putExtra("result", "false");  
      }
            finish();
    }
    case "auth": {
      setResult(4000, intent);
      if (result) {
          intent.putExtra("result","true");
      } else {
          intent.putExtra("result","true");
      }
      finish();
    }
  }
}

위 코드에서는 mode에 따라 resultCode가 달라지고,

파라미터인 result 값(boolean)에 따라 value가 달라집니다. 이러한 점과 Collection을 활용해 코드를 개선해 봅시다!

개선 후

public void returnResult(boolean result) {
    Intent intent = new Intent();
      String intentValue = result ? "true" : "false";
    HashMap<String, Integer> resultCode = new HashMap<String, Integer>() 
    {{
          // put(mode, resultCode)
         put("register", 1000);
         put("auth", 4000);
    }};

    setResult(resultCode.get(mode), intent);
    intent.putExtra("result", intentValue);
    finish();
}

HashMap을 사용해 mode를 키 값으로, resultCode를 밸류 값으로 갖는 구조를 만들었습니다. resultCode.get(mode) 를 사용하면 되기 때문에 switch문을 사용할 필요가 없어졌습니다!

또한, 개선 전 코드에서는 파라미터 result에 따라 value값이 달라졌기 때문에 물음표 함수를 통해 String 값(intentValue)을 초기화해두어 변수로 사용하였습니다. 이로 인해 if ~ else 문을 사용할 필요가 없어졌습니다!

mode의 종류가 많아질수록 코드의 양은 천차만별로 차이가 나게 될 것입니다.

적절한 Collection 사용은 많은 분기 처리가 필요할 때 유용하게 쓰일 것 같습니다.