zishuo/natpass
1
0
Fork 0
mirror of https://github.com/lwch/natpass synced 2025-10-18 02:40:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

修改server端的析构逻辑

Browse Source
This commit is contained in:
lwch
2021-08-26 14:31:30 +08:00
parent ecfbc0d14d
commit 6602b69504
3 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
code/server/handler/handler.go
View File

@@ -185,10 +185,10 @@ func (h *Handler) closeClient(cli *client) {
pair := h.links[t]
h.lockLinks.RUnlock()
if pair[0] != nil {
pair[0].close(t)
pair[0].closeLink(t)
}
if pair[1] != nil {
pair[1].close(t)
pair[1].closeLink(t)
}
h.lockLinks.Lock()
delete(h.links, t)
@@ -197,10 +197,13 @@ func (h *Handler) closeClient(cli *client) {
h.lockClients.RLock()
clients := h.clients[cli.parent.id]
h.lockClients.RUnlock()
clients.close(cli.idx)
if len(clients.data) == 0 {
h.lockClients.Lock()
delete(h.clients, clients.id)
h.lockClients.Unlock()
if clients != nil {
clients.close(cli.idx)
}
}
func (h *Handler) removeClients(id string) {
h.lockClients.Lock()
delete(h.clients, id)
h.lockClients.Unlock()
}