Kotlin’s most powerful feature (compared to Java) are probably Extension Functions.
They allow you to add new functionality to an existing class, without using inheritance or reflection.
You can define your own extension functions and Kotlin ships with a lot of pre-defined ones.
There are e.g. a lot extensions for collections, but also those that extend every class in your program. That’s right - every single class is extended by the functions in Standard.kt
.
When I started with Kotlin, I found them quite confusing. With this post, I hope to un-confuse them a little bit.
Before we get to the details, some basics.
An Extension Function is a function that adds new functionality to an existing class. It’s functionally equivalent to a static utility function in Java, but we can call it from the object itself.
In Java, you might have something like this:
class Util {
public static final boolean isNumeric(String receiver) {
return reveiver.matches("\\d+");
}
}
...
String myString = ...;
if(Util.isNumeric(myString)) ...
By using an extension function, Kotlin allows us to call the isNumeric
method directly on the receiving object:
fun String.isNumeric(): Boolean {
return this.matches("\\d+".toRegex())
}
...
val myString = ...
if(myString.isNumeric()) ...
Pretty cool, right? You might agree that using extension functions increases the code’s readability.
The example shows how an extension function is defined. We use the fun
keyword, followed by the type we want to extend (in this case String
). Then a .
and the name of the function. That’s it!
The extension function will always receive an implicit parameter this
, which is the object we called the extension function on (in this case, myString
).
Let’s take a look at Kotlin’s standard extension functions.
run
public inline fun <T, R> T.run(block: T.() -> R): R = block()
This looks a bit messy in the beginning. But it’s actually quite simple.
run
is a generic extension function on any type T
, that executes another extension function on this type (extension functions as parameters are symbolized as T.()
) and returns the result of that function.
Consider this example:
val generator = PasswordGenerator()
generator.seed = "someString"
generator.hash = {s -> someHash(s)}
generator.hashRepititions = 1000
val password: Password = generator.generate()
Someone didn’t quite think through the design of this password generator class. Its constructor does nothing, but it needs a lot of initialization. To use this class, I need to introduce a variable generator
, set all necessary parameters and use generate
to generate the actual password. This is cool if I want to generate multiple passwords with the same generator, but if I throwaway the generator variable anyway, I can save some typing by using run
:
val password: Password = PasswordGenerator().run {
seed = "someString"
hash = {s -> someHash(s)}
hashRepetitions = 1000
generate()
}
Lambdas in Kotlin implicitly return the result of the last line. That’s why I can omit the temporary variable and store the password directly. Because an extension function is passed to run
I can also access the password generator’s properties like seed
or hash
directly.
This way “the things that belong together” stay together.
It’s one line more, I give you that. But it’s still less code with less redundancy I had to write.
apply
public inline fun <T> T.apply(block: T.() -> Unit): T { block(); return this }
This function looks almost like run
, but there is a slight difference. apply
will always return this
, which is the receiver of the extension function.
That means that we can use apply
for builder-style initialization, even if the designer of the class didn’t implement it with a builder in mind.
If we stick to the password generator example:
val generator = PasswordGenerator().apply {
seed = "someString"
hash = {s -> someHash(s)}
hashRepetitions = 1000
}
val password = generator.generate()
This is particularly useful if you need an object with the same settings more than once.
It can also be used to avoid init{}
blocks during the initialization of a class.
Instead of:
class Message(message: String, signature: String) {
val body = MessageBody()
init {
body.text = message + "\n" + signature
}
}
You can write:
class Message(message: String, signature: String) {
val body = MessageBody().apply {
text = message + "\n" + signature
}
}
let
public inline fun <T, R> T.let(block: (T) -> R): R = block(this)
let
is particularly useful to use it instead of a null
check:
val fruitBasket = ...
val result = apple?.let {
fruitBasket.add(it)
}
The apple (it
) will only be added to the basket if it’s not null
.
Notice that let
will return the result of the last line in the lambda (which is the result of add
).
Just like run
, let
also helps to keep the scope of your variables small. Instead of this
you have to use it
or a custom variable name to reference it:
val fruitBasket = ...
appleTree.pick()?.let {
fruitBasket.add(it)
}
also
public inline fun <T> T.also(block: (T) -> Unit): T { block(this); return this }
Kotlin 1.1 introduced this new extension function called also
. It fills the gap between let
and apply
.
Just like apply
, it will always return its receiver. But instead of an extension function it takes a normal function as an argument.
class FruitBasket {
private var weight = 0
fun addFrom(appleTree: AppleTree) {
val apple = appleTree.pick().also { apple ->
this.weight += apple.weight
add(apple)
}
...
}
...
fun add(fruit: Fruit) = ...
}
I renamed the implicit it
to an explicit apple
this time. Assuming the function appleTree.pick()
returns an apple, the weight of the whole basket increases.
Notice that both the apple and the basket have a weight
property. If I had used apply
, it would not be possible* to access the basket’s weight
. Since apply
takes an extension function, this
would refer to the apple and not the basket. With also
this is possible.
takeIf and takeUnless
public inline fun <T> T.takeIf(predicate: (T) -> Boolean): T? = if (predicate(this)) this else null
public inline fun <T> T.takeUnless(predicate: (T) -> Boolean): T? = if (!predicate(this)) this else null
The last two functions I want to mention have also been added to Kotlin with 1.1.
takeIf
does exactly that - it takes the receiver if it satisfies a condition.
takeUnless
inverts the condition and takes this
only if the condition is not satisfied.
For example:
val redApple = apple.takeIf { it.color == RED }
val otherApple = apple.takeUnless { it.color == RED }
Those two methods are the functional equivalent to the filter
function to collections, but they operate on a single variable.
Summary
All extension functions can be used to avoid temporary variables/Re-scope a variable (or the result of a function). Many of the extension functions in Standard.kt
can be used interchangeable. Pick the ones that match your style and that offer the best readability in your context.
The table below shows a common usage for each of the presented functions.
Function | Argument | Returns | Usage examples |
---|---|---|---|
run |
this |
any result | make an object a builder for another object |
apply |
this |
this |
make an object its own builder |
let |
it |
any result | null checks |
also |
it |
this |
You need apply but don’t want to shadow this |
takeIf |
it |
it or null |
shorthand for if(predicate(it)) it else null |
takeUnless |
it |
it or null |
shorthand for if(!predicate(it)) it else null |
* You still can by using this@FruitBasket
, but do you want to do this?