TIL

Lec 08. 코틀린에서 함수를 다루는 방법

1. 함수 선언 문법

접근 지시어 fun 함수 이름(파라미터): 반환타입 {
}

• 두 정수를 받아 더 큰 정수를 반환하는 예제

// 자바의 함수
public int max(int a, int b) {
  if (a > b) {
    return a;
  }
  return b;
}

// 코틀린의 함수
fun max(a: Int, b: Int): Int {
  return if (a > b) {
    a
  } else {
    b
  }
}

• 코틀린에서 {} 대신 =을 사용해 함수를 작성할 수도 있다.

fun max(a: Int, b: Int): Int =
  if (a > b) {
    a
  } else {
    b
  }
}

코틀린 함수의 특징

• fun - 함수를 의미하는 키워드
• 함수는 클래스 안에 있을 수도, 파일 최상단에 있을 수도 있다.
  • 한 파일 안에 여러 함수들이 있을 수도 있다.
• 기본적으로 public이 생략되어 있기에 fun만 작성하면 public이다.
• 함수의 매개변수는 [매개변수명: 타입]으로 작성한다.
• 함수의 반환 타입이 void인 경우 생략 가능
• 함수가 하나의 결과값이라면 블럭 대신 = 사용 가능
• =을 사용하는 경우 타입을 추론할 수 있기에 반환 타입을 생략해도 된다.
  • 한줄로 이어 쓸 수도 있다.
  • block {}을 사용하는 경우 반환 타입이 Unit이 아니라면 타입을 명시해야 한다.

fun max(a: Int, b: Int) = if (a > b) a else b

2. default parameter

• 자바에선 여러 파라미터를 받는 메서드를 오버로딩하여 몇몇 파라미터는 기본값을 사용하도록 구현할 수 있다.
  • 오버로딩을 사용하면 메서드를 여러 개 구현해야 하는 번거로움이 있다.

// 출력할 문자열, 반복 횟수, 개행 여부를 받아 문자열을 반복 출력하는 메서드
public void repeat(String str, int num, boolean useNewLine) {
  // ...
}

public void repeat(String str, int num) {
  repeat(str, num, true);
}

public void repeat(String str) {
  repeat(str, 3, true);
}

• 코틀린에서 default parameter를 사용하면 자바 오버로딩의 단점을 해소할 수 있다.
  • 오버로딩된 자바 함수처럼 기본값이 지정된 파라미터를 넣지 않아도 클라이언트는 메서드를 사용할 수 있다.
• 물론 코틀린에도 오버로딩 기능이 있긴 하다.

fun repeat(
  str: String,
  num: Int = 3,
  useNewLine: Boolean = true
) {
  // ... 
}

3. named argument (parameter)

• 만약 특정 매개변수는 기본값을 사용하고 특정 매개변수는 직접 넣고 싶다면 named argument를 사용할 수 있다.
  • ex) num은 3을 그대로 쓰고 useNewLine은 false를 쓰고 싶은 경우

fun main() {
  repeat("Hello Wiorld", useNewLine = false)
}

• 매개변수 이름을 통해 직접 지정하고 지정되지 않은 매개변수는 기본값을 사용할 수 있다.
• named argument를 통해 builder를 직접 만들지 않고 builder의 장점을 가질 수 있는 것이다.
  • 동일한 타입이 연속해서 존재할 때 명시적으로 파라미터를 주입 가능

fun main() {
  printNameAndGender(
    name = "does",
    gender = "MALE"
  )
}

• 단 코틀린에서 자바 함수를 가져다 사용하는 경우엔 named argument를 사용할 수 없다.

4. 같은 타입의 여러 파라미터 받기 (가변인자)

• 자바처럼 …을 쓰지 않고 vararg 키워드를 사용한다.
• 배열을 가변인자에 넣을 땐 spread 연산자로 *를 붙여야 한다.

fun printAll(vararg strings: String) {
    for (str in strings) {
        println(str)
    }
}

fun main() {
  printAll("A", "B", "C")
  
  val array = arrayOf("A", "B", "C")
  printAll(*array)
}