com.jimuwd.xian.registry-proxy 1.0.60 → 1.0.61

package/dist/index.js

@@ -161,7 +161,7 @@ async function fetchFromRegistry(registry, targetUrl, limiter) {
     }
 }
 // 修改后的 writeSuccessfulResponse 函数
-async function writeSuccessfulResponse(registryInfo, targetUrl, res, upstreamResponse, req, proxyInfo, proxyPort, registryInfos) {
+async function writeSuccessfulResponse(registryInfo, targetUrl, resToDownstream, upstreamResponse, downstreamReq, proxyInfo, proxyPort, registryInfos) {
     if (!upstreamResponse.ok)
         throw new Error("Only 2xx upstream response is supported");
     try {
@@ -184,7 +184,7 @@ async function writeSuccessfulResponse(registryInfo, targetUrl, res, upstreamRes
             // JSON 处理逻辑
             const data = await upstreamResponse.json();
             if (data.versions) {
-                const host = req.headers.host || `localhost:${proxyPort}`;
+                const host = downstreamReq.headers.host || `localhost:${proxyPort}`;
                 const baseUrl = `${proxyInfo.https ? 'https' : 'http'}://${host}${proxyInfo.basePath === '/' ? '' : proxyInfo.basePath}`;
                 for (const versionKey in data.versions) {
                     const packageVersion = data.versions[versionKey];
@@ -196,32 +196,46 @@ async function writeSuccessfulResponse(registryInfo, targetUrl, res, upstreamRes
                     }
                 }
             }
-            res.writeHead(upstreamResponse.status, { "content-type": contentType }).end(JSON.stringify(data));
+            resToDownstream.writeHead(upstreamResponse.status, { "content-type": contentType }).end(JSON.stringify(data));
         }
         else {
             // 二进制流处理
             if (!upstreamResponse.body) {
                 logger.error(`Empty response body from ${targetUrl}`);
-                res.writeHead(502).end('Empty Upstream Response');
+                resToDownstream.writeHead(502).end('Empty Upstream Response');
             }
             else {
                 // write back to client
-                res.writeHead(upstreamResponse.status, safeHeaders);
+                resToDownstream.writeHead(upstreamResponse.status, safeHeaders);
                 // stop transfer if client is closed accidentally.
-                // req.on('close', () => upstreamResponse.body?.unpipe());
+                const cleanup = () => {
+                    downstreamReq.off('close', cleanup);
+                    logger.warn("Stop transfer when client is closed accidentally.");
+                    upstreamResponse.body?.unpipe();
+                };
+                downstreamReq.on('close', cleanup);
                 // write back body data (chunked probably)
+                // pipe upstream body to client
+                upstreamResponse.body.pipe(resToDownstream, { end: true });
                 upstreamResponse.body
-                    .on('data', (chunk) => res.write(chunk))
-                    .on('end', () => res.end())
-                    .on('close', () => res.destroy(new Error(`Upstream server ${registryInfo.normalizedRegistryUrl} closed connection while transferring data on url ${targetUrl}`)))
-                    .on('error', (err) => res.destroy(new Error(`Stream error: ${err.message}`, { cause: err, })));
+                    .on('data', (chunk) => logger.info(`chunk transferred size=${chunk.size}`))
+                    .on('close', () => {
+                    const errMsg = `Upstream server ${registryInfo.normalizedRegistryUrl} closed connection while transferring data on url ${targetUrl}`;
+                    logger.error(errMsg);
+                    resToDownstream.destroy(new Error(errMsg));
+                })
+                    .on('error', (err) => {
+                    const errMsg = `Stream error: ${err.message}`;
+                    logger.error(errMsg);
+                    resToDownstream.destroy(new Error(errMsg, { cause: err, }));
+                });
             }
         }
     }
     catch (err) {
         logger.error('Failed to write upstreamResponse:', err);
-        if (!res.headersSent) {
-            res.writeHead(502).end('Internal Server Error');
+        if (!resToDownstream.headersSent) {
+            resToDownstream.writeHead(502).end('Internal Server Error');
         }
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "com.jimuwd.xian.registry-proxy",
-  "version": "1.0.60",
+  "version": "1.0.61",
   "description": "A lightweight npm registry proxy with fallback support",
   "type": "module",
   "main": "dist/index.js",