[Kotlin] 로컬 함수와 확장

중복 코드 제거하기

class User(val id: Int, val name: String, val address: String)

fun saveUser(user: User) {
  if (user.name.isEmpty()) {
    throw IllegalArgumentException("Can't save user ${user.id}: empty Name")
  }

  if (user.address.isEmpty()) {
    throw IllegalArgumentException("Can't save user ${user.id}: empty Address")
  }
}

  위의 코드는 User를 객체를 저장할 때 name, address의 필드를 검사한다. name, address를 검증하는 조건문은 중복된다. 이렇게 반복되는 코드를 제거하면 코드를 깔끔하게 작성할 수 있다.

 

fun saveUser(user: User) {
  fun validate(user: User, value: String, fieldName: String) {
    if (value.isEmpty()) {
      throw IllegalArgumentException("Can't save user ${user.id}: empty $fieldName")
    }
  }

  validate(user, user.name, "Name")
  validate(user, user.address, "Address")
}

  문자열을 검증하는 코드를 로컬 함수로 만들 수 있다. 위의 코드처럼 saveUser 함수 안의 validate가 로컬 함수이다. saveUser 함수 안에서 validate를 호출함으로써 필드 값을 검증한다.

 

fun saveUser(user: User) {
  fun validate(value: String, fieldName: String) {
    if (value.isEmpty()) {
      throw IllegalArgumentException("Can't save user ${user.id}: empty $fieldName")
    }
  }

  validate(user.name, "Name")
  validate(user.address, "Address")
}

  로컬 함수는 바깥 함수의 모든 파라미터와 변수를 사용할 수 있다. 따라서 user 필드를 전달하지 않아도 바로 사용할 수 있고 파라미터를 제거하면 더 깔끔한 함수가 된다.

 

fun User.validateBeforeSave() {
  fun validate(value: String, fieldName: String) {
    if (value.isEmpty()) {
      throw IllegalArgumentException("Can't save user ${id}: empty $fieldName")
    }
  }

  validate(name, "Name")
  validate(address, "Address")
}

fun saveUser(user: User) {
  user.validateBeforeSave()
  
  // ...
}

  validate 부분을 확장 함수로도 분리할 수 있다. 이렇게 확장 함수로 분리하면 [객체.필드]처럼 점(.)을 사용하지 않고도 프로퍼티에 접근할 수 있다.

 

 

함수 중첩시 유의 사항

  함수를 중첩했을 때 깊이가 깊어지면 코드를 파악하기 쉽지 않다. 따라서 함수를 중첩할 때는 한 단계 정도만 중첩 시키는 것을 권장한다.

 

 

참고

• Kotlin IN ACTION