简体   繁体   中英

How to return a value in a Go function that panics?

My Go function is expected to return a value, but it may panic when calling a library function. I can use recover() to capture this in a deferred call, but how can I return a value in this case?

func MyFunc() string{
    defer func() {
        if err := recover(); err != nil {
            // What do I do to make MyFunc() return a value in case of panic?
        }
    }()
    SomeFuncThatMayPanic()
    return "Normal Return Value"
    // How can I return "ERROR" in case of panic?
}

You can use named result parameters . Name your return values, and in the deferred function when panic was detected, you can change the values of the return "variables". The changed new values will be returned.

Example:

func main() {
    fmt.Println("Returned:", MyFunc())
}

func MyFunc() (ret string) {
    defer func() {
        if r := recover(); r != nil {
            ret = fmt.Sprintf("was panic, recovered value: %v", r)
        }
    }()
    panic("test")
    return "Normal Return Value"
}

Output (try it on the Go Playground ):

Returned: was panic, recovered value: test

This is mentioned in the Spec: Defer statements :

For instance, if the deferred function is a function literal and the surrounding function has named result parameters that are in scope within the literal, the deferred function may access and modify the result parameters before they are returned.

It is also mentioned in the blog post Defer, Panic and Recover :

Deferred functions may read and assign to the returning function's named return values.

And also in Effective Go: Recover :

If doParse panics, the recovery block will set the return value to nil —deferred functions can modify named return values.

You can use named parameter like this:

func main() {
    fmt.Println("Returned:", MyFunc())
}

func MyFunc() (err error) {
    defer func() {
        if r := recover(); r != nil {
            err=convertRecoverToError(r)
        }
    }()
    panic("test")
    return nil
}

func ConvertRecoverToError(r interface{}) error {
    switch x := r.(type) {
    case string:
        return errors.New(x)
    case error:
        return x
    default:
        return errors.New(fmt.Sprint(x))
    }
}

Using the example in icza's answer, and if you wonder what happens if you recover from the panic but don't assign the value for the named return , like this:

func main() {
    fmt.Println("Returned:", MyFunc()) // false
}

func MyFunc() (ret bool) {
    defer func() {
        if r := recover(); r != nil {
        }
    }()
    panic("test")
    return true
}

The function will return the zero value of the specified return type.

Run the example in the Go Playground

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM