Merge branch 'master' into jordan/split-epoch-tests

onflow · Aug 16, 2023 · 7824795 · 7824795
2 parents 7579c8e + 99233ab
commit 7824795
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/network/p2p/middleware/middleware.go b/network/p2p/middleware/middleware.go
@@ -818,7 +818,7 @@ func (m *Middleware) IsConnected(nodeID flow.Identifier) (bool, error) {
 // unicastMaxMsgSize returns the max permissible size for a unicast message
 func unicastMaxMsgSize(messageType string) int {
 	switch messageType {
-	case "*messages.ChunkDataResponse":
+	case "*messages.ChunkDataResponse", "messages.ChunkDataResponse":
 		return LargeMsgMaxUnicastMsgSize
 	default:
 		return DefaultMaxUnicastMsgSize
@@ -843,7 +843,7 @@ func UnicastMaxMsgSizeByCode(payload []byte) (int, error) {
 // unicastMaxMsgDuration returns the max duration to allow for a unicast send to complete
 func (m *Middleware) unicastMaxMsgDuration(messageType string) time.Duration {
 	switch messageType {
-	case "messages.ChunkDataResponse":
+	case "*messages.ChunkDataResponse", "messages.ChunkDataResponse":
 		if LargeMsgUnicastTimeout > m.unicastMessageTimeout {
 			return LargeMsgUnicastTimeout
 		}