mirror of
https://github.com/status-im/status-go.git
synced 2025-01-09 14:16:21 +00:00
0babdad17b
* chore: upgrade go-waku to v0.5 * chore: add println and logs to check what's being stored in the enr, and preemptively delete the multiaddr field (#3219) * feat: add wakuv2 test (#3218)
43 lines
748 B
Go
43 lines
748 B
Go
package webtransport
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/quic-go/quic-go"
|
|
)
|
|
|
|
type closeFunc func()
|
|
|
|
// The streamsMap manages the streams of a single QUIC connection.
|
|
// Note that several WebTransport sessions can share one QUIC connection.
|
|
type streamsMap struct {
|
|
mx sync.Mutex
|
|
m map[quic.StreamID]closeFunc
|
|
}
|
|
|
|
func newStreamsMap() *streamsMap {
|
|
return &streamsMap{m: make(map[quic.StreamID]closeFunc)}
|
|
}
|
|
|
|
func (s *streamsMap) AddStream(id quic.StreamID, close closeFunc) {
|
|
s.mx.Lock()
|
|
s.m[id] = close
|
|
s.mx.Unlock()
|
|
}
|
|
|
|
func (s *streamsMap) RemoveStream(id quic.StreamID) {
|
|
s.mx.Lock()
|
|
delete(s.m, id)
|
|
s.mx.Unlock()
|
|
}
|
|
|
|
func (s *streamsMap) CloseSession() {
|
|
s.mx.Lock()
|
|
defer s.mx.Unlock()
|
|
|
|
for _, cl := range s.m {
|
|
cl()
|
|
}
|
|
s.m = nil
|
|
}
|