Extensions on built-in types in Kotlin

Introduction

Extensions let you add new functions to existing types like String or Int without changing their code. This helps you write cleaner and easier-to-read programs.

You want to add a helpful function to a type like String without creating a new class.

You need to reuse a small piece of code that works on built-in types in many places.

You want to make your code more readable by calling functions directly on values.

You want to keep your code organized by grouping related functions with the types they work on.

Syntax

Kotlin

fun TypeName.newFunctionName(params): ReturnType {
    // function body
}

Replace TypeName with the built-in type you want to extend, like String or Int.

The function can be called directly on any value of that type.

Examples

Adds an exclamation mark to the end of a String.

Kotlin

fun String.addExclamation(): String {
    return this + "!"
}

Checks if an integer is even.

Kotlin

fun Int.isEven(): Boolean {
    return this % 2 == 0
}

Returns the square of a Double using a short syntax.

Kotlin

fun Double.square(): Double = this * this

Sample Program

This program adds two extension functions: one to make a string uppercase and add an exclamation, and one to check if an integer is positive. It then uses them and prints the results.

Kotlin

fun String.shout(): String {
    return this.uppercase() + "!"
}

fun Int.isPositive(): Boolean {
    return this > 0
}

fun main() {
    val word = "hello"
    println(word.shout())

    val number = 10
    println("Is $number positive? ${number.isPositive()}")
}

Important Notes

Extension functions do not actually change the original class; they are just a convenient way to call new functions on existing types.

You can add extension properties, but they do not store data in the instance (they use getters/setters).

If a member function and an extension function have the same signature, the member function is called.

Summary

Extensions let you add new functions to built-in types without changing their code.

They make your code cleaner and easier to read by calling functions directly on values.

Use the syntax fun TypeName.functionName() to create an extension function.

Practice

1. What is the main purpose of extension functions in Kotlin?

easy

A. To create new classes that inherit from built-in types

B. To add new functions to existing types without modifying their source code

C. To override existing functions in built-in types

D. To convert one type to another automatically

Solution

1. Step 1: Understand extension functions concept

   Extension functions allow adding new functions to existing types without changing their original code.

2. Step 2: Compare options with concept

   Only To add new functions to existing types without modifying their source code correctly describes this purpose. Options B and C talk about inheritance and overriding, which are different concepts. To convert one type to another automatically is unrelated.

3. Final Answer:

   To add new functions to existing types without modifying their source code -> Option B

4. Quick Check:

   Extension functions add new behavior without changing original code [OK]

Hint: Remember: extensions add functions, not change classes [OK]

Common Mistakes:

• Confusing extension functions with inheritance
• Thinking extensions override existing methods
• Believing extensions convert types automatically

2. Which of the following is the correct syntax to define an extension function isEven for Int in Kotlin?

easy

A. fun Int.isEven(): Boolean { return this % 2 == 0 }

B. fun isEven(): Boolean { return this % 2 == 0 }

C. fun Int.isEven: Boolean = this % 2 == 0

D. fun isEven(Int): Boolean = Int % 2 == 0

Solution

1. Step 1: Recall extension function syntax

   Extension functions are declared as fun TypeName.functionName(). Here, Int.isEven() is correct.

2. Step 2: Check each option

   fun Int.isEven(): Boolean { return this % 2 == 0 } matches the correct syntax. fun isEven(): Boolean { return this % 2 == 0 } misses the receiver type. fun Int.isEven: Boolean = this % 2 == 0 misses parentheses. fun isEven(Int): Boolean = Int % 2 == 0 has wrong parameter syntax.

3. Final Answer:

   fun Int.isEven(): Boolean { return this % 2 == 0 } -> Option A

4. Quick Check:

   Extension syntax = fun TypeName.functionName() [OK]

Hint: Extension syntax always starts with fun TypeName.functionName() [OK]

Common Mistakes:

• Omitting the receiver type before function name
• Missing parentheses after function name
• Using parameter instead of receiver for extension

3. What will be the output of this Kotlin code?

fun String.addExclamation() = this + "!"

fun main() {
    val word = "Hello"
    println(word.addExclamation())
}

medium

A. Hello

B. Hello!!

C. Compilation error

D. Hello!

Solution

1. Step 1: Understand the extension function behavior

   The extension function addExclamation adds an exclamation mark to the string.

2. Step 2: Trace the main function output

   Calling word.addExclamation() returns "Hello!" which is printed.

3. Final Answer:

   Hello! -> Option D

4. Quick Check:

   Extension adds "!" so output is "Hello!" [OK]

Hint: Extension functions act like normal methods on the type [OK]

Common Mistakes:

• Expecting original string unchanged
• Thinking extension adds multiple exclamations
• Assuming syntax error due to extension

4. Identify the error in this Kotlin extension function and fix it:

fun Int.isPositive {
    return this > 0
}

medium

A. Missing parentheses after function name; should be fun Int.isPositive()

B. Extension functions cannot return Boolean

C. The receiver type Int is invalid for extensions

D. The return statement is not allowed in extension functions

Solution

1. Step 1: Check function declaration syntax

   Extension functions must have parentheses after the function name, even if no parameters.

2. Step 2: Identify the fix

   Add parentheses: fun Int.isPositive(). Other options are incorrect because Boolean return is allowed, Int is valid receiver, and return statements are allowed.

3. Final Answer:

   Missing parentheses after function name; should be fun Int.isPositive() -> Option A

4. Quick Check:

   Extension functions need () after name [OK]

Hint: Always add () after extension function name, even if no parameters [OK]

Common Mistakes:

• Omitting parentheses after function name
• Thinking extensions can't return values
• Confusing receiver type validity

5. You want to create an extension function wordCount for String that returns the number of words separated by spaces. Which implementation correctly handles multiple spaces and empty strings?

hard

A. fun String.wordCount() = this.split(" ").size

B. fun String.wordCount() = this.count { it == ' ' }

C. fun String.wordCount() = this.trim().split(" ").filter { it.isNotEmpty() }.size

D. fun String.wordCount() = this.length

Solution

1. Step 1: Understand the problem requirements

   We need to count words separated by spaces, ignoring extra spaces and empty strings.

2. Step 2: Analyze each option

   fun String.wordCount() = this.split(" ").size splits by space but counts empty strings if multiple spaces exist. fun String.wordCount() = this.trim().split(" ").filter { it.isNotEmpty() }.size trims, splits, filters out empty strings, then counts, correctly handling multiple spaces and empty strings. fun String.wordCount() = this.length returns string length, unrelated. fun String.wordCount() = this.count { it == ' ' } counts spaces, not words.

3. Final Answer:

   fun String.wordCount() = this.trim().split(" ").filter { it.isNotEmpty() }.size -> Option C

4. Quick Check:

   Trim, split, filter empty = correct word count [OK]

Hint: Trim and filter empty strings after split to count words correctly [OK]

Common Mistakes:

• Counting empty strings as words
• Using string length instead of word count
• Counting spaces instead of words