如何在Golang中自动锁定/解锁

tjvv9vkg  于 2023-08-01  发布在  Go
关注(0)|答案(1)|浏览(123)

我有一个结构体,其中有许多“公共”方法,我需要保持线程安全。

package main

import "sync"

type Test struct {
    sync.Mutex
    privateInt    int
    privateString string
    privateMap    map[string]interface{}
}

func (p *Test) A() {
    p.Lock()
    defer p.Unlock()

    // do something that changes the fields...
}

func (p *Test) B() {
    p.Lock()
    defer p.Unlock()

    // do something that changes the fields...
}

func (p *Test) C() {
    p.Lock()
    defer p.Unlock()

    // do something that changes the fields...
}

func (p *Test) D() {
    p.Lock()
    defer p.Unlock()

    // do something that changes the fields...
}

// and so on

字符串
如果结构体有很多方法,我必须检查并确认每个方法都执行锁。看起来有点傻。

arknldoa

arknldoa1#

我想到了一个类似数据库的方法。我用另一个结构体TestProvider Package 了结构体Test,并且在使用Test之前必须调用Transaction()来获取指针。

package main

import "sync"

type Test struct {
    privateInt    int
    privateString string
    privateMap    map[string]interface{}
}

func (p *Test) A() {
    // do something that changes the fields...
}

func (p *Test) B() {
    // do something that changes the fields...
}

func (p *Test) C() {
    // do something that changes the fields...
}

func (p *Test) D() {
    // do something that changes the fields...
}

// and so on

type TestProvider struct {
    sync.Mutex
    test *Test
}

func (p *TestProvider) Transaction(callback func(test *Test)) {
    p.Lock()
    defer p.Unlock()
    callback(p.test)
}

func NewTestProvider(test *Test) *TestProvider {
    return &TestProvider{
        test: test,
    }
}

func main() {
    p := NewTestProvider(&Test{})
    go p.Transaction(func(test *Test) {
        test.A()
        test.B()
    })
    go p.Transaction(func(test *Test) {
        test.C()
        test.D()
    })
}

字符串
这很好,但我认为可能有更好的方法。

相关问题