Extensions on built-in types in Kotlin - Mini Project: Build & Apply

Extensions on built-in types in Kotlin

📖 Scenario: You are working on a Kotlin project where you want to add new useful functions to existing types without changing their original code. This is common when you want to make your code cleaner and easier to read.

🎯 Goal: You will create an extension function on the String type to count vowels, then use it in a program to show how many vowels a word has.

📋 What You'll Learn

Create an extension function on the String type called countVowels

The countVowels function returns the number of vowels in the string

Use the extension function on a sample string

Print the result

💡 Why This Matters

🌍 Real World

Extension functions let you add new abilities to existing types like strings or numbers. This makes your code cleaner and easier to read, like adding new tools to your toolbox without changing the tools themselves.

💼 Career

Many Kotlin projects use extension functions to improve code quality and maintainability. Knowing how to write and use them is important for Kotlin developers in app development and backend services.

1

Create a sample string

Create a variable called word and set it to the string "Kotlin".

Kotlin

// Create a variable called word and set it to "Kotlin"
// Your code here

Hint

Use val word = "Kotlin" to create the string variable.

2

Create an extension function to count vowels

Create an extension function on String called countVowels that returns an Int. It should count the vowels a, e, i, o, u in the string ignoring case.

Kotlin

val word = "Kotlin"

// Create an extension function countVowels on String
// Your code here

Hint

Use fun String.countVowels(): Int and inside use this.count { it.lowercaseChar() in "aeiou" }.

3

Use the extension function on the string

Create a variable called vowelCount and set it to the result of calling countVowels() on the variable word.

Kotlin

val word = "Kotlin"

fun String.countVowels(): Int {
    return this.count { it.lowercaseChar() in "aeiou" }
}

// Call countVowels on word and store in vowelCount
// Your code here

Hint

Use val vowelCount = word.countVowels() to call the extension function.

4

Print the vowel count

Print the text "Number of vowels in Kotlin: " followed by the value of vowelCount using println.

Kotlin

val word = "Kotlin"

fun String.countVowels(): Int {
    return this.count { it.lowercaseChar() in "aeiou" }
}

val vowelCount = word.countVowels()

// Print the number of vowels in word
// Your code here

Hint

Use println("Number of vowels in Kotlin: $vowelCount") to show the result.

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