[Go] 함수 고급(가변 인수, defer, 람다함수, 함수타입 변수)

 

 

들어가며

Modern Go 기본 편에서 함수를 다뤘지만, 이걸로는 부족합니다. Go에서는 더욱 강력한 함수 기능이 있기 때문에 간단하게 살펴보겠습니다. 

 

---

1. 가변 인수 함수 (Variadic Function)

• 함수의 매개변수를 개수 제한 없이 받을 수 있음.
• ...타입 형식으로 정의.
• 내부적으로는 slice로 처리됨.

package main

import "fmt"

func sum(nums ...int) int {
    total := 0
    for _, n := range nums {
        total += n
    }
    return total
}

func main() {
    fmt.Println(sum(1, 2))          // 3
    fmt.Println(sum(1, 2, 3, 4, 5)) // 15
}

---

2. defer 지연 실행

• 함수가 리턴되기 직전에 실행됨.
• 보통 리소스 해제, 파일 닫기, unlock 등에 사용.
• defer는 스택처럼 동작 → 나중에 등록된 것이 먼저 실행됨 (LIFO).

package main

import "fmt"

func main() {
    defer fmt.Println("마지막 실행됨")
    defer fmt.Println("이것도 지연 실행됨")

    fmt.Println("메인 함수 시작")
}

출력:

메인 함수 시작
이것도 지연 실행됨
마지막 실행됨

 보통 함수 수명 주기 내에서 끝내야 하는 작업을 defer로 활용. File close, Mutex 잠금 해제 등 처럼 리소스를 정리하는 데 유용하게 사용됨.

func writeLog(w io.Writer, message string) {
    fmt.Fprintf(w, "[Log] %s\n", message)
}

func main() {
    f, _ := os.Create("app.log")
    defer f.Close()

    writeLog(f, "File log")        // 파일에 출력
    writeLog(os.Stdout, "Console") // 콘솔에 출력
}

 

---

3. 함수 타입 변수 (Function Type Variable)

• 함수도 변수에 담을 수 있고, 인자/리턴으로 주고받을 수 있음.

package main

import "fmt"

func add(a, b int) int {
    return a + b
}

func main() {
    // 함수 타입 변수
    var op func(int, int) int
    op = add

    fmt.Println(op(3, 4)) // 7
}

---

4. 함수 리터럴 (Function Literal, 익명 함수)

• 이름이 없는 함수 정의 방식.
• 보통 함수 변수에 바로 대입하거나 즉시 실행.
• 클로저(closure)도 지원 → 외부 변수 캡처 가능.

package main

import "fmt"

func main() {
    // 변수에 담아서 사용
    mul := func(a, b int) int {
        return a * b
    }
    fmt.Println(mul(3, 4)) // 12

    // 즉시 실행 함수
    result := func(msg string) string {
        return "Hello " + msg
    }("Go")
    fmt.Println(result) // Hello Go

    // 클로저 예시
    counter := 0
    inc := func() int {
        counter++
        return counter
    }
    fmt.Println(inc()) // 1
    fmt.Println(inc()) // 2
}