1.局部函数嵌套
Kotlin 中提供了函数的嵌套,在函数内部还可以定义新的函数。这样我们可以在函数中嵌套
这些提前的函数,来抽取重复代码。
class User(val id: Int, val name: String, val address: String, val email: 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")
}
if (user.email.isEmpty()) {
throw IllegalArgumentException("Can't save user ${user.id}: empty Email")
}
// save to db ...
}
上面的代码在判断 name、address 等是否为空的处理其实很类似。这时候,我们可以利用
在函数内部嵌套的声明一个通用的判空函数将相同的代码抽取到一起:
/** 利用局部函数抽取相同的逻辑,去除重复的代码* */
fun saveUser2(user: User) {
fun validate(value: String, fildName: String) {
if (value.isEmpty()) {
throw IllegalArgumentException("Can't save user ${user.id}: empty
$fildName")
}
}
validate(user.name, "Name")
validate(user.address, "Address")
validate(user.email, "Email")
}
2.扩展函数嵌套
除了利用嵌套函数去抽取,此时,其实也可以用扩展函数来抽取,如下所示:
/** 利用扩展函数抽取逻辑* */
fun User.validateAll() {
fun validate(value: String, fildName: String) {
if (value.isEmpty()) {
throw IllegalArgumentException("Can't save user $id: empty $fildName")
}
}
validate(name, "Name")
validate(address, "Address")
validate(email, "Email")
}
fun saveUser3(user: User) {
user.validateAll()
// save to db ...
}