67 lines
2.5 KiB
JavaScript
67 lines
2.5 KiB
JavaScript
|
import fs from 'node:fs';
|
|||
|
|
import path from 'node:path';
|
|||
|
|
import process from 'node:process';
|
|||
|
|
|
|||
|
|
import { HeartbeatProcessor } from '../../src/processor/heartbeatProcessor.js';
|
|||
|
|
|
|||
|
|
function usageAndExit(code = 1) {
|
|||
|
|
console.log(`\n用法：\n node scripts/kafka/decodeMessage.js --base64 <str>\n node scripts/kafka/decodeMessage.js --hex <str>\n node scripts/kafka/decodeMessage.js --file <path> [--encoding base64|hex|raw]\n\n说明：\n- 用于验证 Kafka message.value 的反向解码结果（对端为 JSON + UTF-8 bytes）\n- 会尝试：UTF-8 JSON / base64 -> (gzip|deflate|raw deflate|brotli) 循环解压（兼容但对端当前未用）\n`);
|
|||
|
|
process.exit(code);
|
|||
|
|
}
|
|||
|
|
function parseArgs(argv) {
|
|||
|
|
const args = {};
|
|||
|
|
for (let i = 2; i < argv.length; i++) {
|
|||
|
|
const a = argv[i];
|
|||
|
|
if (a === '--base64') args.base64 = argv[++i];
|
|||
|
|
else if (a === '--hex') args.hex = argv[++i];
|
|||
|
|
else if (a === '--file') args.file = argv[++i];
|
|||
|
|
else if (a === '--encoding') args.encoding = argv[++i];
|
|||
|
|
else if (a === '--help' || a === '-h') args.help = true;
|
|||
|
|
else args._ = [...(args._ ?? []), a];
|
|||
|
|
}
|
|||
|
|
return args;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const args = parseArgs(process.argv);
|
|||
|
|
if (args.help) usageAndExit(0);
|
|||
|
|
|
|||
|
|
const processor = new HeartbeatProcessor(
|
|||
|
|
{ batchSize: 9999, batchTimeout: 1000 },
|
|||
|
|
{ insertHeartbeatEvents: async () => {} }
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
let buf;
|
|||
|
|
if (args.base64) {
|
|||
|
|
buf = Buffer.from(String(args.base64).trim(), 'base64');
|
|||
|
|
} else if (args.hex) {
|
|||
|
|
buf = Buffer.from(String(args.hex).trim().replace(/\s+/g, ''), 'hex');
|
|||
|
|
} else if (args.file) {
|
|||
|
|
const p = path.resolve(process.cwd(), args.file);
|
|||
|
|
const raw = fs.readFileSync(p);
|
|||
|
|
const enc = (args.encoding ?? 'raw').toLowerCase();
|
|||
|
|
if (enc === 'raw') buf = raw;
|
|||
|
|
else if (enc === 'base64') buf = Buffer.from(raw.toString('utf8').trim(), 'base64');
|
|||
|
|
else if (enc === 'hex') buf = Buffer.from(raw.toString('utf8').trim().replace(/\s+/g, ''), 'hex');
|
|||
|
|
else {
|
|||
|
|
console.error('未知 encoding:', enc);
|
|||
|
|
usageAndExit(1);
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
usageAndExit(1);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
const obj = processor.decodeToObject(buf);
|
|||
|
|
const items = Array.isArray(obj) ? obj : [obj];
|
|||
|
|
|
|||
|
|
console.log('[decode] ok; items:', items.length);
|
|||
|
|
console.log(JSON.stringify(obj, null, 2));
|
|||
|
|
|
|||
|
|
const normalized = items.map((x) => processor.normalizeHeartbeat(processor.unwrapPayload(x)));
|
|||
|
|
const validCount = normalized.filter((x) => processor.validateData(x)).length;
|
|||
|
|
console.log('[normalize] valid (required fields present):', validCount, '/', items.length);
|
|||
|
|
} catch (err) {
|
|||
|
|
console.error('[decode] failed:', err);
|
|||
|
|
process.exitCode = 1;
|
|||
|
|
}
|