(c *gossh.ServerConn, add bool)
| 446 | } |
| 447 | |
| 448 | func (srv *Server) trackConn(c *gossh.ServerConn, add bool) { |
| 449 | srv.mu.Lock() |
| 450 | defer srv.mu.Unlock() |
| 451 | |
| 452 | if srv.conns == nil { |
| 453 | srv.conns = make(map[*gossh.ServerConn]struct{}) |
| 454 | } |
| 455 | if add { |
| 456 | srv.conns[c] = struct{}{} |
| 457 | srv.connWg.Add(1) |
| 458 | } else { |
| 459 | delete(srv.conns, c) |
| 460 | srv.connWg.Done() |
| 461 | } |
| 462 | } |