[문자열 계산기] main 메서드에서는 큰 흐름만 보고싶다.

 

상황

문자열 계산기를 만드는 프로젝트에서 기능 구현까지 끝난 상황.

 

실행 결과 정상적으로 기능이 동작한다

 

문제

기능은 동작하지만 main 메서드에 모든 구현 코드가 들어가 있다. 그래서 큰 흐름이 한 눈에 안들어와서 불편했다.

코드의 기능별로 메서드로 뽑아내면 가독성과 유지보수에 편리할 거 같아서 리팩토링을 해야겠다.

아래의 코드는 입력과 출력 기능을 하는 코드까지만 메서드로 추출했고 계산 기능은 아직 메서드로 뽑아내지 못했다. 계산 기능이 아무래도 더 코드가 많아 여전히 main 메서드가 한 눈에 안들어온다.

public static void main(String[] args) {
    String[] temp = input();

    int result = Integer.parseInt(temp[0]);
    for (int i=1; i<temp.length; i += 2) {
      String operator = temp[i];
      int second = Integer.parseInt(temp[i+1]);

      if (operator.equals("+")) {
        result = result + second;
      } else if (operator.equals("-")) {
        result = result - second;
      } else if (operator.equals("*")) {
        result = result * second;
      } else if (operator.equals("/")) {
        result = result / second;
      } else {
        System.out.println("사칙연산이 아닙니다.");
        exit(0);
      }
    }
    print(result);
}

해결

계산 로직까지 메서드로 추출하니 한결 main 메서드에서 흐름을 파악하기 쉬워졌다.

public static void main(String[] args) {
    String[] temp = input(); // 입력 기능

    int result = Integer.parseInt(temp[0]);
    for (int i=1; i<temp.length; i += 2) {
      String operator = temp[i];
      int second = Integer.parseInt(temp[i+1]);
      result = calculate(result, operator, second); // 계산 기능
    }

    print(result); // 출력 기능
}

static String[] input() {
    System.out.print("계산할 수를 입력하세요: ");
    Scanner in = new Scanner(System.in);
    return in.nextLine().split(" ");
  }

static int calculate(int first, String operator, int second) {
    int result = 0;
    if (operator.equals("+")) {
      result = first + second;
    } else if (operator.equals("-")) {
      result = first - second;
    } else if (operator.equals("*")) {
      result = first * second;
    } else if (operator.equals("/")) {
      result = first / second;
    } else {
      System.out.println("사칙연산이 아닙니다.");
      exit(0);
    }
    return result;
}

static void print(int result) {
    System.out.println("결과값: " + result);
}