Go：如何从切片中删除元素并在内存中修改它

Question

我正在尝试编写一个简单的服务器/客户端聊天程序以用于学习目的，但我被卡住了。我想让Leave 函数删除它传递的指针并更新结构中的切片，以便指针不再存在。但它不起作用。

例如：Input,Output

type Room struct {
    Name     string
    Visitors []*net.Conn
} 


func (r *Room) Leave(pc *net.Conn) {
    for i, pv := range r.Visitors {
        //found the connection we want to remove
        if pc == pv {
            fmt.Printf("Before %v\n",r.Visitors)
            r.Visitors = append(r.Visitors[:i], r.Visitors[i+1:]...)
            fmt.Printf("Before %v\n",r.Visitors)                
            return
        }
    }
}

Answer 1

删除逻辑是正确的（您的输出也证实了这一点）。问题是多个 goroutine 之间缺乏同步（你的问题没有说明这一点）。

您说（在您的评论中）您希望它首先使用 1 个 goroutine，然后再处理同步。但是你的代码已经使用了多个 goroutine，所以你不能这么奢侈：

//Let a goroutine Handle the connection
go handleConnection(conn)

多个 goroutine 正在读取和写入 Room.Visitors 切片，因此您别无选择，只能同步访问它。

一个例子是使用sync.RWLock:

utils.go:

type Room struct {
    mu       sync.RWLock
    Name     string
    Visitors []net.Conn
}


func (r *Room) Leave(c net.Conn) {
    r.mu.Lock()
    defer r.mu.Unlock()
    for i, v := range r.Visitors {
        //found the connection we want to remove
        if c == v {
            fmt.Printf("Before remove %v\n", r.Visitors)
            r.Visitors = append(r.Visitors[:i], r.Visitors[i+1:]...)
            fmt.Printf("After remove %v\n", r.Visitors)
            return
        }
    }
}

此外，当任何其他代码触及Room.Visitors 时，也要锁定代码（如果您只是阅读它，可以使用RWMutex.RLock()）。

同样，您需要同步对多个 goroutine 读取/更改的所有变量的访问。

还要考虑将删除后释放的元素归零，否则底层数组仍将保留该值，从而阻止垃圾收集器正确释放它使用的内存。有关这方面的更多信息，请参阅Does go garbage collect parts of slices?

Answer 2

您正在使用指向 Room 类型中的接口 (Visitors []*net.Conn) 的指针。您永远不需要指向接口值的指针。接口是值就像一个通用的内容持有者，它在内存中的表示不同于实现该接口的结构。

您应该简单地将Visitors 类型声明为接口：

type Room struct {
    Name     string
    Visitors []net.Conn // should not be a pointer to interface value
} 


func (r *Room) Leave(pc net.Conn) { // same thing as above
    for i, pv := range r.Visitors {
        // found the connection we want to remove
        // in the comparison below actual concrete types are being compared.
        // both these concrete types must be comparable (they can't be slices for example
        if pc == pv {
            r.Visitors = append(r.Visitors[:i], r.Visitors[i+1:]...)
            return
        }
    }
}

请注意，上面的比较 (pc == pv) 并不是那么微不足道。在这里阅读：https://golang.org/ref/spec#Comparison_operators

另外，参考这个问题：Why can't I assign a *Struct to an *Interface?