46 lines
1.7 KiB
TypeScript
46 lines
1.7 KiB
TypeScript
import 'reflect-metadata';
|
|
import { NestFactory } from '@nestjs/core';
|
|
import { createServer } from 'node:http';
|
|
import { MetricsService } from './metrics/metrics.service';
|
|
import { SendWorkerModule } from './send-worker.module';
|
|
|
|
Object.defineProperty(BigInt.prototype, 'toJSON', {
|
|
configurable: true,
|
|
value(this: bigint) {
|
|
const result = Number(this);
|
|
if (!Number.isSafeInteger(result)) throw new RangeError('金额超过 JavaScript 安全整数范围');
|
|
return result;
|
|
},
|
|
});
|
|
|
|
async function bootstrap() {
|
|
if ((process.env.CMPP_PROCESS_ROLE?.trim() || 'worker') !== 'worker') {
|
|
throw new Error('send-worker requires CMPP_PROCESS_ROLE=worker');
|
|
}
|
|
const app = await NestFactory.createApplicationContext(SendWorkerModule);
|
|
app.enableShutdownHooks();
|
|
const metrics = app.get(MetricsService);
|
|
const host = process.env.API_WORKER_METRICS_HOST?.trim() || '127.0.0.1';
|
|
const port = Number(process.env.API_WORKER_METRICS_PORT ?? 9465);
|
|
const metricsServer = createServer((request, response) => {
|
|
if (request.method !== 'GET' || request.url !== '/metrics') {
|
|
response.writeHead(404).end();
|
|
return;
|
|
}
|
|
response.writeHead(200, { 'Content-Type': 'text/plain; version=0.0.4; charset=utf-8', 'Cache-Control': 'no-store' });
|
|
response.end(metrics.render());
|
|
});
|
|
await new Promise<void>((resolve, reject) => {
|
|
metricsServer.once('error', reject);
|
|
metricsServer.listen(port, host, resolve);
|
|
});
|
|
const close = async () => {
|
|
await new Promise<void>((resolve) => metricsServer.close(() => resolve()));
|
|
await app.close();
|
|
};
|
|
process.once('SIGTERM', () => void close());
|
|
process.once('SIGINT', () => void close());
|
|
}
|
|
|
|
void bootstrap();
|