Go语言：仅在函数返回错误时清理创建目录的最优方案

问题场景 #

我有一个Go函数，开头会创建目录，之后执行其他操作，代码示例如下：

func someFunc(path string) (string, error) {
    err := os.Mkdir(path, os.ModePerm)
    if err != nil {
        return "", err
    }

    err = doSomething()
    if err != nil {
        return "", err
    }

    err = doSomethingElse()
    if err != nil {
        return "", err
    }

    return path, nil
}

希望该函数在所有出现错误的情况下，删除之前创建的目录。直接在每个错误分支调用os.RemoveAll代码冗余且不够优雅；用defer的话，无错误时也会执行删除，不符合需求。

简洁实现方案 #

最优雅的方式是用defer配合一个成功标记变量，仅在函数执行失败时触发目录删除：

func someFunc(path string) (string, error) {
    err := os.Mkdir(path, os.ModePerm)
    if err != nil {
        return "", err
    }

    success := false
    defer func() {
        if !success {
            // 忽略删除错误，避免覆盖原业务错误
            _ = os.RemoveAll(path)
        }
    }()

    err = doSomething()
    if err != nil {
        return "", err
    }

    err = doSomethingElse()
    if err != nil {
        return "", err
    }

    success = true
    return path, nil
}

核心逻辑说明 #

• 初始化success为false，默认函数执行失败。
• defer注册的匿名函数会在函数返回前执行，通过success的值判断是否需要清理目录。
• 仅当所有业务操作都成功完成后，才将success设为true，跳过删除逻辑。
• 删除目录时忽略返回的错误，防止原业务错误被覆盖，确保调用者能拿到真实的失败原因。

内容的提问来源于stack exchange，提问作者adrians