🐞 fix(web-server): 修改proxy解决mid-web内存泄漏
This commit is contained in:
50
web-server/app/middleware/egg-proxy.ts
Normal file
50
web-server/app/middleware/egg-proxy.ts
Normal file
@@ -0,0 +1,50 @@
|
||||
import c2k = require('koa-connect');
|
||||
import { createProxyMiddleware, Options as ContextOptions } from 'http-proxy-middleware';
|
||||
import * as micromatch from 'micromatch';
|
||||
import * as isGlob from 'is-glob';
|
||||
|
||||
function match(context: string, path: string): boolean {
|
||||
// single path
|
||||
if (!isGlob(context)) {
|
||||
return path.indexOf(context) === 0;
|
||||
} else {
|
||||
// single glob path
|
||||
const matches = micromatch([path], context);
|
||||
return matches?.length > 0;
|
||||
}
|
||||
}
|
||||
|
||||
export interface Options {
|
||||
[contextName: string]: ContextOptions;
|
||||
}
|
||||
|
||||
export default function (options: Options) {
|
||||
return async (ctx, next) => {
|
||||
for (const context of Object.keys(options)) {
|
||||
if (match(context, ctx.path)) {
|
||||
const contextOptions: ContextOptions = options[context];
|
||||
const { onProxyReq } = contextOptions;
|
||||
await c2k(
|
||||
createProxyMiddleware(context, {
|
||||
...contextOptions,
|
||||
onProxyReq(proxyReq, req, res) {
|
||||
if (onProxyReq && typeof onProxyReq === 'function') {
|
||||
onProxyReq(proxyReq, req, res);
|
||||
}
|
||||
// reset rawBody after bodyparser
|
||||
const { rawBody, body: requestBody } = ctx.request;
|
||||
if (requestBody && rawBody) {
|
||||
proxyReq.setHeader('Content-Length', Buffer.byteLength(rawBody));
|
||||
proxyReq.write(rawBody);
|
||||
proxyReq.end();
|
||||
}
|
||||
return proxyReq;
|
||||
},
|
||||
}) as any
|
||||
)(ctx, next);
|
||||
}
|
||||
}
|
||||
await next();
|
||||
};
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user