取消推迟提交报告

为了避免在执行过程中出现错误时忘记解锁，可以通过在函数开头使用 defer 关键字来自动解锁。但是，如果执行成功，需要在特定点手动解锁。本文探讨了如何取消 defer 的解锁操作，并提出了两种解决方案。第一种方法使用 sync.once 来确保互斥锁仅被解锁一次。第二种方法建议将锁定部分重构到单独的函数中，以简化代码结构。

我有以下代码结构，其中我在 a 点上 lock() ，并且需要在同一函数中的 b 点上明确 unlock() 。在 a 点和 b 点之间，我有多个基于需要 unlock() 的错误的返回。在a点使用defer lock.unlock()解决了出现错误时锁被释放的问题。但是，如果执行成功到达 b 点 - 我怎样才能取消 defer 和 unlock() 呢？

func foo() {
    ...
    // point A
    lock.Lock()
    defer lock.Unlock()
    ...
    err := bar()
    if err != nil {
        return
    }
    ...
    //point B - need to definetely unlock here
    //lock.Unlock() ?
}

解决方案

您无法取消延迟函数。

您可以使用 sync.once 来确保互斥锁被解锁一次：

func foo() {
    var unlockonce sync.once

    // point a
    lock.lock()
    defer unlockonce.do(lock.unlock)
    ...
    err := bar()
    if err != nil {
        return
    }
    ...
    // point b - need to unlock here
    unlockonce.do(lock.unlock)
}

如果可能，最好重构代码，以便锁定部分保留在单个函数中：

func foolock() error {
    lock.lock()
    defer lock.unlock()
    if err := bar(); err != nil { return err }
    ...
    return nil
}

func foo() {
    if err := foolock(); err != nil { return }
    ... do the bit that doesn't need a lock
}

显然，这里的命名和错误处理很宽松，因为代码是通用的而不是特定的。如果您需要 b 点之前的代码块（现在位于 foolock 中的代码中）的信息，则可以将其与错误一起返回。

无法取消延迟函数。

p>

使用局部变量记录函数相对于锁的状态，并在延迟函数中检查该变量。

lock.Lock()
locked := true

defer func() {
    if locked  {
        lock.Unlock()
    }
}()
...
err := bar()
if err != nil {
    return
}
...

lock.Unlock()
locked = false

...

由于锁通常用在多线程环境中，所以需要注意的是，函数局部变量 locked 只能由单个 goroutine 访问（感谢 rick-777 在评论中指出这一点）。

好了，本文到此结束，带大家了解了《取消推迟提交报告》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！