go - 如何取消延期结单

Question

我有以下代码结构，其中我在 A 点 Lock() 并且需要在同一函数的 B 点确定 Unlock()。在 A 点和 B 点之间，我根据需要解锁()的错误有多个返回。使用 defer lock.Unlock() A点解决了万一出现错误锁会被释放的问题。但是，如果执行成功到达 B 点 - 我怎么能取消延迟和 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() ?
}

Answer 1

您不能取消延迟功能。

您可以使用 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 中的代码中，它可以与错误一起返回。