Extension functions in Android Kotlin

Introduction

Extension functions let you add new actions to existing classes without changing their code. This helps keep your code clean and easy to read.

You want to add a helper function to a class from a library you can't change.

You want to make your code more readable by adding meaningful functions to common types.

You want to organize utility functions close to the types they work with.

You want to avoid creating subclasses just to add small functions.

You want to keep your code simple and avoid cluttering classes with extra methods.

Syntax

Android Kotlin

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

Use the keyword fun followed by the class name you want to extend.

Inside the function, this refers to the instance of the class.

Examples

Adds an exclamation mark to any String.

Android Kotlin

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

Checks if an integer is even.

Android Kotlin

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

Calculates the sum of all integers in a list.

Android Kotlin

fun List<Int>.sumAll(): Int {
    var sum = 0
    for (num in this) {
        sum += num
    }
    return sum
}

Sample App

This program adds a shout function to the String class. It makes the text uppercase and adds an exclamation mark. Then it prints the shouted greeting.

Android Kotlin

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

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

Important Notes

Extension functions do not actually modify the original class; they just look like they do.

If a class already has a function with the same name and parameters, the class's own function is called instead of the extension.

Use extension functions to keep your code organized and easy to understand.

Summary

Extension functions add new functions to existing classes without changing them.

They make your code cleaner and easier to read.

You can call them like regular functions on the class instances.

Practice

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

easy

A. To override existing functions in a class

B. To create a new subclass from an existing class

C. To add new functions to existing classes without modifying them

D. To delete functions from a class

Solution

1. Step 1: Understand extension functions

   Extension functions let you add new functions to classes without changing their original code.

2. Step 2: Compare options

   Only To add new functions to existing classes without modifying them correctly describes this behavior; others describe subclassing, overriding, or deleting, which are different concepts.

3. Final Answer:

   To add new functions to existing classes without modifying them -> Option C

4. Quick Check:

   Extension function = Add new functions [OK]

Hint: Extension functions add features without changing original class [OK]

Common Mistakes:

• Confusing extension functions with inheritance
• Thinking extension functions override existing methods
• Believing extension functions can remove class methods

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

easy

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

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

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

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

Solution

1. Step 1: Check extension function syntax

   Extension functions are declared with fun ClassName.functionName(), so fun Int.isEven() is correct.

2. Step 2: Validate return type and body

   fun Int.isEven(): Boolean { return this % 2 == 0 } correctly uses parentheses and return type with a proper function body.

3. Final Answer:

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

4. Quick Check:

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

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

Common Mistakes:

• Omitting the receiver type before function name
• Missing parentheses after function name
• Incorrect return type declaration

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

fun String.addHello() = "Hello, $this!"

fun main() {
  val name = "Anna"
  println(name.addHello())
}

medium

A. Hello, Anna!

B. AnnaHello, !

C. Hello, !

D. Compilation error

Solution

1. Step 1: Understand the extension function

   The function addHello() adds "Hello, " before the string and an exclamation mark after.

2. Step 2: Apply function to variable

   Calling name.addHello() where name is "Anna" returns "Hello, Anna!".

3. Final Answer:

   Hello, Anna! -> Option A

4. Quick Check:

   Extension adds prefix and suffix correctly [OK]

Hint: Extension functions run like normal methods on instances [OK]

Common Mistakes:

• Misreading string interpolation syntax
• Assuming extension modifies original string
• Expecting output without exclamation mark

4. Identify the error in this Kotlin extension function code:

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

medium

A. Function must declare a return type Boolean

B. The function body should not use return

C. The keyword 'this' cannot be used in extension functions

D. Extension function cannot be declared on Int

Solution

1. Step 1: Check function return type

   The function returns a Boolean expression but does not declare a return type, which is required.

2. Step 2: Confirm correct syntax

   Adding : Boolean after the function name fixes the error.

3. Final Answer:

   Function must declare a return type Boolean -> Option A

4. Quick Check:

   Return type needed for non-Unit functions [OK]

Hint: Declare return type when function returns a value [OK]

Common Mistakes:

• Omitting return type for Boolean-returning functions
• Thinking 'this' is disallowed in extensions
• Confusing Unit functions with value-returning ones

5. You want to add an extension function toTitleCase() to String that capitalizes the first letter of each word. Which implementation correctly achieves this?

hard

A. fun String.toTitleCase() = this.toUpperCase()

B. fun String.toTitleCase() = this.lowercase()

C. fun String.toTitleCase() = this.replaceFirstChar { it.uppercase() }

D. fun String.toTitleCase() = split(" ").joinToString(" ") { it.capitalize() }

Solution

1. Step 1: Understand title case requirement

   Title case means capitalizing the first letter of each word, not the whole string or just the first letter.

2. Step 2: Analyze options

   fun String.toTitleCase() = split(" ").joinToString(" ") { it.capitalize() } splits the string by spaces, capitalizes each word, then joins them back, correctly implementing title case.

3. Final Answer:

   fun String.toTitleCase() = split(" ").joinToString(" ") { it.capitalize() } -> Option D

4. Quick Check:

   Title case = capitalize each word [OK]

Hint: Split words, capitalize each, then join for title case [OK]

Common Mistakes:

• Using uppercase() on whole string instead of each word
• Capitalizing only first character of entire string
• Ignoring spaces between words