首页 > 解决方案 > 如何取消延期结单

问题描述

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

标签: gomutex

解决方案


您不能取消延迟功能。

您可以使用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,则可以将其与错误一起返回。


推荐阅读