如何初始化Go结构体中的成员

Question

我是 Golang 新手，所以其中的分配让我发疯：

import "sync"

type SyncMap struct {
        lock *sync.RWMutex
        hm map[string]string
}
func (m *SyncMap) Put (k, v string) {
        m.lock.Lock()
        defer m.lock.Unlock()

        m.hm[k] = v, true
}

后来，我只是调用：

sm := new(SyncMap)
sm.Put("Test, "Test")

此时我遇到了 nil 指针恐慌。

我已经通过使用另一个函数并在 new() 之后调用它来解决这个问题：

func (m *SyncMap) Init() {
        m.hm = make(map[string]string)
        m.lock = new(sync.RWMutex)
}

但我想知道是否可以摆脱这个样板初始化？

Answer 1

你只需要一个构造函数。一个常见的使用模式是，

func NewSyncMap() *SyncMap {
    return &SyncMap{hm: make(map[string]string)}
}

如果结构中有更多字段，启动一个 goroutine 作为后端，或者注册一个终结器，一切都可以在此构造函数中完成。

func NewSyncMap() *SyncMap {
    sm := SyncMap{
        hm: make(map[string]string),
        foo: "Bar",
    }

    runtime.SetFinalizer(sm, (*SyncMap).stop)

    go sm.backend()

    return &sm
}

Answer 2

由于互斥体未初始化，“Mue”的解决方案不起作用。以下修改有效：

package main

import "sync"

type SyncMap struct {
        lock *sync.RWMutex
        hm map[string]string
}

func NewSyncMap() *SyncMap {
        return &SyncMap{lock: new(sync.RWMutex), hm: make(map[string]string)}
}

func (m *SyncMap) Put (k, v string) {
        m.lock.Lock()
        defer m.lock.Unlock()
        m.hm[k] = v
}

func main() {
    sm := NewSyncMap()
    sm.Put("Test", "Test")
}

http://play.golang.org/p/n-jQKWtEy5

Answer 3

执事抓得好。 Mue 可能正在考虑更常见的模式，将锁作为值而不是指针包含在内。由于互斥体的零值是一种随时可用的未锁定互斥体，因此它不需要初始化，并且包含 1 作为值是很常见的。为了进一步简化，您可以通过省略字段名称来嵌入它。然后，您的结构体获取互斥体的方法集。请参阅此工作示例，http://play.golang.org/p/faO9six-Qx。我还去掉了 defer 的使用。在某种程度上，这是一个偏好和编码风格的问题，但由于它确实有很小的开销，所以我倾向于不在小函数中使用它，特别是在没有条件代码的情况下。