fix: split map check/create to guard nil (#781)

wolf31o2 · web-flow · commit e899da9ab65b · 2024-11-09T12:21:38.000-05:00
Signed-off-by: Chris Gianelloni &lt;wolf31o2@blinklabs.io&gt;
diff --git a/muxer/muxer.go b/muxer/muxer.go
@@ -160,6 +160,8 @@ func (m *Muxer) RegisterProtocol(
 	m.protocolReceiversMutex.Lock()
 	if _, ok := m.protocolSenders[protocolId]; !ok {
 		m.protocolSenders[protocolId] = make(map[ProtocolRole]chan *Segment)
+	}
+	if _, ok := m.protocolReceivers[protocolId]; !ok {
 		m.protocolReceivers[protocolId] = make(map[ProtocolRole]chan *Segment)
 	}
 	m.protocolSenders[protocolId][protocolRole] = senderChan

Original file line number	Diff line number	Diff line change
`@@ -160,6 +160,8 @@ func (m *Muxer) RegisterProtocol(`
`160`	`160`	`m.protocolReceiversMutex.Lock()`
`161`	`161`	`if _, ok := m.protocolSenders[protocolId]; !ok {`
`162`	`162`	`m.protocolSenders[protocolId] = make(map[ProtocolRole]chan *Segment)`
	`163`	`+ }`
	`164`	`+ if _, ok := m.protocolReceivers[protocolId]; !ok {`
`163`	`165`	`m.protocolReceivers[protocolId] = make(map[ProtocolRole]chan *Segment)`
`164`	`166`	`}`
`165`	`167`	`m.protocolSenders[protocolId][protocolRole] = senderChan`