7e0b47311d
Rather than passing functions to construct complex objects such as Dialer and QUICDialer, pass interface implementations. Ensure that a nil implementation does not cause harm. Make Saver implement the correct interface either directly or indirectly. We need to implement the correct interface indirectly for TCP conns (or connected UDP sockets) because we have two distinct use cases inside netx: observing just the connect event and observing just the I/O events. With this change, the construction of composed Dialers and QUICDialers is greatly simplified and more obvious. Part of https://github.com/ooni/probe/issues/2121
36 lines
707 B
Go
36 lines
707 B
Go
package tracex
|
|
|
|
//
|
|
// Saver implementation
|
|
//
|
|
|
|
import "sync"
|
|
|
|
// The Saver saves a trace. The zero value of this type
|
|
// is valid and can be used without initialization.
|
|
type Saver struct {
|
|
// ops contains the saved events.
|
|
ops []Event
|
|
|
|
// mu provides mutual exclusion.
|
|
mu sync.Mutex
|
|
}
|
|
|
|
// Read reads and returns events inside the trace. It advances
|
|
// the read pointer so you won't see such events again.
|
|
func (s *Saver) Read() []Event {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
v := s.ops
|
|
s.ops = nil
|
|
return v
|
|
}
|
|
|
|
// Write adds the given event to the trace. A subsequent call
|
|
// to Read will read this event.
|
|
func (s *Saver) Write(ev Event) {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
s.ops = append(s.ops, ev)
|
|
}
|