Add proxy event logger state propagate

This commit is contained in:
Shelikhoo 2021-12-20 13:45:53 +00:00
parent e4305a4d2b
commit f12cfe6a9f
No known key found for this signature in database
GPG key ID: C4D5E79D22B25316

View file

@ -115,6 +115,7 @@ type SnowflakeProxy struct {
NATProbeURL string NATProbeURL string
// NATTypeMeasurementInterval is time before NAT type is retested // NATTypeMeasurementInterval is time before NAT type is retested
NATTypeMeasurementInterval time.Duration NATTypeMeasurementInterval time.Duration
EventDispatcher event.SnowflakeEventDispatcher
shutdown chan struct{} shutdown chan struct{}
} }
@ -340,7 +341,7 @@ func (sf *SnowflakeProxy) makePeerConnectionFromOffer(sdp *webrtc.SessionDescrip
close(dataChan) close(dataChan)
pr, pw := io.Pipe() pr, pw := io.Pipe()
conn := &webRTCConn{pc: pc, dc: dc, pr: pr} conn := &webRTCConn{pc: pc, dc: dc, pr: pr, eventLogger: sf.EventDispatcher}
conn.bytesLogger = newBytesSyncLogger() conn.bytesLogger = newBytesSyncLogger()
dc.OnOpen(func() { dc.OnOpen(func() {
@ -524,6 +525,9 @@ func (sf *SnowflakeProxy) Start() error {
if sf.NATProbeURL == "" { if sf.NATProbeURL == "" {
sf.NATProbeURL = DefaultNATProbeURL sf.NATProbeURL = DefaultNATProbeURL
} }
if sf.EventDispatcher == nil {
sf.EventDispatcher = event.NewSnowflakeEventDispatcher()
}
broker, err = newSignalingServer(sf.BrokerURL, sf.KeepLocalAddresses) broker, err = newSignalingServer(sf.BrokerURL, sf.KeepLocalAddresses)
if err != nil { if err != nil {