openide/java/java-tests/testData/inspection/switchExpressionBackwardMigration/afterLastComment.java

// "Replace with old style 'switch' statement" "true"


import java.util.function.BiFunction;

public class AAA {
  public static void main(String[] args) {
    BiFunction<Integer, Integer, Integer> operation = getOperation(1);
    System.out.println(operation.apply(5, 3));
  }

  private static BiFunction<Integer, Integer, Integer> getOperation(int operationCode) {
      switch (operationCode) {
          case 1:
              return (a, b) -> a + b;
          // Addition
          case 2:
              return (a, b) -> a - b;
          // Subtraction
          case 3:
              return (a, b) -> a * b;
          // Multiplication
          case 4:
              return (a, b) -> a / b;
          // Division
          default:
              return (a, b) -> 0; // Default case
      }
  }

}