Funkcje jednowyrażeniowe

Czym jest skrócony zapis

Wiele funkcji robi tylko jedną rzecz: liczy wynik z parametrów i go zwraca. Dla takich funkcji pełny zapis z klamrami i return jest niepotrzebnie rozwlekły. Kotlin oferuje wtedy funkcję jednowyrażeniową – krótszy zapis ze znakiem =.

// zapis blokowy (z poprzedniej lekcji)
fun kwadrat(x: Int): Int {
    return x * x
}

// zapis jednowyrażeniowy – to samo działanie
fun kwadrat(x: Int) = x * x

Składnia: = zamiast { return }

Zamiast klamr i słowa return stawiamy znak = i podajemy jedno wyrażenie. Jego wartość staje się wynikiem funkcji.

fun suma(a: Int, b: Int) = a + b
fun szescian(x: Int) = x * x * x
fun powitanie(imie: String) = "Cześć, $imie!"

fun main() {
    println(suma(3, 5))        // 8
    println(szescian(2))         // 8
    println(powitanie("Anna"))    // Cześć, Anna!
}

Pominięcie typu zwracanego

Przy zapisie jednowyrażeniowym Kotlin potrafi sam rozpoznać typ wyniku z wyrażenia po = – tak jak rozpoznawał typ zmiennej z jej wartości. Dlatego typ zwracany zwykle pomijamy.

fun kwadrat(x: Int) = x * x          // wynik to Int
fun polowa(x: Double) = x / 2         // wynik to Double
fun pelnoletni(wiek: Int) = wiek >= 18   // wynik to Boolean

if / when jako ciało funkcji

Pamiętasz, że w Kotlinie if i when są wyrażeniami (mają wartość)? Dzięki temu mogą być całym ciałem funkcji jednowyrażeniowej – to jeden z najwygodniejszych zapisów.

fun wiekszy(a: Int, b: Int) = if (a > b) a else b

println(wiekszy(5, 8))   // 8

fun nazwaDnia(n: Int) = when (n) {
    1 -> "Poniedziałek"
    2 -> "Wtorek"
    3 -> "Środa"
    else -> "Inny dzień"
}

println(nazwaDnia(2))   // Wtorek

Kiedy stosować

Skrócony zapis to wygoda, ale nie do wszystkiego. Prosta zasada:

Częste błędy

❌ Błąd 1: return w funkcji z =

fun kwadrat(x: Int) = return x * x
// Błąd! po = podajemy
// samo wyrażenie,
// bez return

fun kwadrat(x: Int) = x * x

❌ Błąd 2: wiele instrukcji po =

fun opis(x: Int) =
    println("Liczba:")
    println(x)
// druga linia jest już
// POZA funkcją

fun opis(x: Int) {
    println("Liczba:")
    println(x)
}

❌ Błąd 3: brak znaku =

fun kwadrat(x: Int): Int x * x
// Błąd! potrzebny = albo
// klamry { return ... }

fun kwadrat(x: Int): Int = x * x

❌ Błąd 4: zaskakujący wywnioskowany typ

fun polowa(x: Int) = x / 2
// polowa(5) -> 2 (Int!)
// dzielenie całkowite

fun polowa(x: Int) = x / 2.0
// polowa(5) -> 2.5 (Double)

Podsumowanie

• Funkcja jednowyrażeniowa zastępuje { return wyrażenie } zapisem = wyrażenie.
• Po = podajemy jedno wyrażenie – bez klamr i bez return.
• Typ zwracany zwykle pomijamy – Kotlin rozpoznaje go z wyrażenia (inferencja).
• Przy złożonym wyrażeniu warto typ jednak zapisać dla czytelności.
• if i when (jako wyrażenia) świetnie sprawdzają się jako ciało skróconej funkcji.
• Wiele instrukcji, pętla lub kilka zmiennych → wróć do zapisu blokowego { ... }.

Zadania do wykonania

Zadania wykonaj w Kotlin Playground (play.kotlinlang.org).