refactor: refine code

avwo · Dec 2, 2020 · ab7905f · ab7905f
1 parent 1ddbc07
commit ab7905f
Showing 1 changed file with 15 additions and 14 deletions.
diff --git a/lib/tunnel.js b/lib/tunnel.js
@@ -422,31 +422,32 @@ function tunnelProxy(server, proxy) {
           reqSocket.clientPort = req.clientPort;
           reqSocket.globalValue = req.globalValue;
           resSocket.statusCode = resData.statusCode;
-          if (buf) {
-            var _pipe = reqSocket.pipe;
-            reqSocket.pipe = function(stream) {
-              if (buf) {
-                stream.write(buf);
-                buf = null;
-              }
-              return _pipe.apply(this, arguments);
-            };
-          }
           pluginMgr.resolvePipePlugin(reqSocket, function() {
             data.pipe = reqSocket._pipeRule;
             if (reqSocket._pipePluginPorts) {
               reqSocket.inspectFrames = data.inspect = true;
               reqSocket.customParser = false;
             }
+            var connHandler = function() {
+              if (buf) {
+                var _pipe = reqSocket.pipe;
+                reqSocket.pipe = function(stream) {
+                  if (buf) {
+                    stream.write(buf);
+                    buf = null;
+                  }
+                  return _pipe.apply(this, arguments);
+                };
+              }
+              socketMgr.handleConnect(reqSocket, resSocket);
+            };
             var handleEstablished = function() {
               if (useTunnelPolicy) {
                 sendEstablished(200, function() {
-                  setTimeout(function() {
-                    socketMgr.handleConnect(reqSocket, resSocket);
-                  }, 16);
+                  setTimeout(connHandler, 16);
                 });
               } else {
-                socketMgr.handleConnect(reqSocket, resSocket);
+                connHandler();
                 sendEstablished();
               }
             };