登录
首页 >  Golang >  Go问答

goroutine 上的“选择性”互斥

来源:stackoverflow

时间:2024-04-29 21:21:34 395浏览 收藏

各位小伙伴们,大家好呀!看看今天我又给各位带来了什么文章?本文标题《goroutine 上的“选择性”互斥》,很明显是关于Golang的文章哈哈哈,其中内容主要会涉及到等等,如果能帮到你,觉得很不错的话,欢迎各位多多点评和分享!

问题内容

我是 Go 新手,我想实现一种自定义互斥机制,其中每个应用程序用户一次可以执行一个 goroutine。 为了简化起见,将 U1 和 U2 视为应用程序用户,以及 F1(userId)、F2(userId) 和 F3(userId) 三个不同的 goroutine,它们从数据库中读取/写入仅与给定用户相关的记录强>。我希望,如果U1调用(F1,F2,F3),(F2,F3)直到F1结束才执行,那么执行F2或F3(按调用顺序将是最佳解决方案),最后执行剩余的一个被执行。 U2 不受 U1 锁的影响,但她如前所述受其自己的锁控制。你会如何实施这个?是否有内置原语?谢谢


解决方案


使用 sync.Mutex 进行互斥。为每个活动用户使用一个互斥体。

我假设用户是通过某种 id 来识别的。我们将该类型称为 userid。使用由 userid 键控的映射来存储每个用户互斥锁。

只有活动用户才需要互斥体,而不是所有潜在用户。使用整数计数器来确定用户是否处于活动状态。

上述地图和计数器也应该受到互斥锁的保护。

代码如下:

type userID string // replace userID type with id type for your application

// userLock is the per user lock
type userLock struct {
    sync.Mutex

    // n is number of goroutines between lock and unlock
    n int
}

var (
    // mu protects locks map and n field in userLock
    mu    sync.Mutex

    // locks is map of active locks
    locks = map[userID]*userLock{}
)


// lockUser locks the user specified by id. If the lock is
// already in use, the calling goroutine blocks until
// the lock is available.
func lockUser(id userID) {
    mu.Lock()
    l := locks[id]
    if l == nil {
        l = &userLock{}
        locks[id] = l
    }
    l.n++
    mu.Unlock()
    l.Lock()
}

// unlockUser unlocks the user specified by id. It 
// is a run-time error if the user is not locked on
// entry unlockUser.
func unlockUser(id userID) {
    mu.Lock()
    l := locks[id]
    if l == nil {
        panic("unlock without lock")
    }
    l.n--
    if l.n == 0 {
        delete(locks, id)
    }
    mu.Unlock()
    l.Unlock()
}

文中关于的知识介绍,希望对你的学习有所帮助!若是受益匪浅,那就动动鼠标收藏这篇《goroutine 上的“选择性”互斥》文章吧,也可关注golang学习网公众号了解相关技术文章。

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>