mirror of
https://github.com/tailscale/tailscale.git
synced 2025-10-05 04:21:01 +02:00
206d98e84b
In the earlier http2 package migration (1d93bdce20ddd2, #17394) I had removed Direct.Close's tracking of the connPool, thinking it wasn't necessary. Some tests (in another repo) are strict and like it to tear down the world and wait, to check for leaked goroutines. And they caught this letting some goroutines idle past Close, even if they'd eventually close down on their own. This restores the connPool accounting and the aggressife close. Updates #17305 Updates #17394 Change-Id: I5fed283a179ff7c3e2be104836bbe58b05130cc7 Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
37 lines
835 B
Go
37 lines
835 B
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package set
|
|
|
|
// HandleSet is a set of T.
|
|
//
|
|
// It is not safe for concurrent use.
|
|
type HandleSet[T any] map[Handle]T
|
|
|
|
// Handle is an opaque comparable value that's used as the map key in a
|
|
// HandleSet.
|
|
type Handle struct {
|
|
v *byte
|
|
}
|
|
|
|
// NewHandle returns a new handle value.
|
|
func NewHandle() Handle {
|
|
return Handle{new(byte)}
|
|
}
|
|
|
|
// Add adds the element (map value) e to the set.
|
|
//
|
|
// It returns a new handle (map key) with which e can be removed, using a map
|
|
// delete or the [HandleSet.Delete] method.
|
|
func (s *HandleSet[T]) Add(e T) Handle {
|
|
h := NewHandle()
|
|
if *s == nil {
|
|
*s = make(HandleSet[T])
|
|
}
|
|
(*s)[h] = e
|
|
return h
|
|
}
|
|
|
|
// Delete removes the element with handle h from the set.
|
|
func (s HandleSet[T]) Delete(h Handle) { delete(s, h) }
|