File tree Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Original file line number Diff line number Diff line change @@ -331,11 +331,16 @@ func (t *TcpTransport) Listen(laddr ma.Multiaddr) (transport.Listener, error) {
331
331
list = t .upgrader .GateMaListener (mal )
332
332
}
333
333
334
+ // Always wrap the listener with tcpGatedMaListener to apply TCP-specific configurations
335
+ tcpList := & tcpGatedMaListener {list , 0 }
336
+
334
337
if t .enableMetrics {
335
- // TODO: Fix this: The tcpListener wrapping should happen on both enableMetrics and disabledMetrics path
336
- list = newTracingListener (& tcpGatedMaListener { list , 0 }, t .metricsCollector )
338
+ // Wrap with tracing listener if metrics are enabled
339
+ return t . upgrader . UpgradeGatedMaListener ( t , newTracingListener (tcpList , t .metricsCollector )), nil
337
340
}
338
- return t .upgrader .UpgradeGatedMaListener (t , list ), nil
341
+
342
+ // Regular path without metrics
343
+ return t .upgrader .UpgradeGatedMaListener (t , tcpList ), nil
339
344
}
340
345
341
346
// Protocols returns the list of terminal protocols this transport can dial.
You can’t perform that action at this time.
0 commit comments