fix: no stats on multistream proto dial

Author: dryajov

src/dial.js

@@ -218,6 +218,7 @@ class Dialer {
       }
 
       if (err) {
+        log('muxer upgrade failed with error', err)
         // couldn't upgrade to Muxer, it is ok, use the existing connection
         return callback(null, connection)
       }
@@ -257,7 +258,7 @@ class Dialer {
       //  - add incomming new streams to connHandler
       const nextMuxer = (key) => {
         log('selecting %s', key)
-        selectSafe(msDialer, key, (err, conn) => {
+        selectSafe(msDialer, key, (err, _conn) => {
           if (err) {
             if (muxers.length === 0) {
               return callback(new Error('could not upgrade to stream muxing'))
@@ -266,6 +267,9 @@ class Dialer {
             return nextMuxer(muxers.shift())
           }
 
+          // observe muxed connections
+          const conn = observeConnection(null, key, _conn, this.switch.observer)
+
           const muxedConn = this.switch.muxers[key].dialer(conn)
           this.switch.muxedConns[b58Id] = {
             muxer: muxedConn
@@ -404,7 +408,8 @@ class Dialer {
         return callback(err)
       }
 
-      selectSafe(msDialer, this.protocol, (err, conn) => {
+      selectSafe(msDialer, this.protocol, (err, _conn) => {
+        const conn = observeConnection(null, this.protocol, _conn, this.switch.observer)
         callback(err, conn)
       }, callback)
     }, callback)

src/protocol-muxer.js

@@ -3,9 +3,15 @@
 const multistream = require('multistream-select')
 const observeConn = require('./observe-connection')
 
+const debug = require('debug')
+const log = debug('libp2p:switch:protocol-muxer')
+
 module.exports = function protocolMuxer (protocols, observer) {
   return (transport) => (_parentConn) => {
-    const parentConn = observeConn(transport, null, _parentConn, observer)
+    const parentConn = transport
+      ? observeConn(transport, null, _parentConn, observer)
+      : _parentConn
+
     const ms = new multistream.Listener()
 
     Object.keys(protocols).forEach((protocol) => {
@@ -14,6 +20,7 @@ module.exports = function protocolMuxer (protocols, observer) {
       }
 
       const handler = (protocolName, _conn) => {
+        log(`registering handler with protocol ${protocolName}`)
         const protocol = protocols[protocolName]
         if (protocol) {
           const handlerFunc = protocol && protocol.handlerFunc