67 lines
2.1 KiB
JavaScript
67 lines
2.1 KiB
JavaScript
|
import dotenv from 'dotenv';
|
||
|
|
import kafka from 'kafka-node';
|
||
|
|
|
||
|
|
dotenv.config();
|
||
|
|
|
||
|
|
const probeTs = Number(process.argv[2] || Date.now());
|
||
|
|
const probeRoom = process.argv[3] || `PROBE-${probeTs}`;
|
||
|
|
const probeHotelId = Number(process.argv[4] || 1085);
|
||
|
|
const topic = process.env.KAFKA_TOPIC || process.env.KAFKA_TOPICS || 'blwlog4Nodejs-rcu-register-topic';
|
||
|
|
const kafkaHost = (process.env.KAFKA_BROKERS || '').split(',').map((s) => s.trim()).filter(Boolean).join(',');
|
||
|
|
const saslEnabled = process.env.KAFKA_SASL_ENABLED === 'true';
|
||
|
|
const sslEnabled = process.env.KAFKA_SSL_ENABLED === 'true';
|
||
|
|
|
||
|
|
const kafkaClientOptions = {
|
||
|
|
kafkaHost,
|
||
|
|
clientId: process.env.KAFKA_CLIENT_ID || 'bls-register-producer'
|
||
|
|
};
|
||
|
|
|
||
|
|
if (saslEnabled && process.env.KAFKA_SASL_USERNAME && process.env.KAFKA_SASL_PASSWORD) {
|
||
|
|
kafkaClientOptions.sasl = {
|
||
|
|
mechanism: process.env.KAFKA_SASL_MECHANISM || 'plain',
|
||
|
|
username: process.env.KAFKA_SASL_USERNAME,
|
||
|
|
password: process.env.KAFKA_SASL_PASSWORD
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
if (sslEnabled) {
|
||
|
|
kafkaClientOptions.sslOptions = { rejectUnauthorized: false };
|
||
|
|
}
|
||
|
|
|
||
|
|
const payload = {
|
||
|
|
ts_ms: probeTs,
|
||
|
|
upgrade_ts_ms: probeTs + 1234,
|
||
|
|
hotel_id: Number.isFinite(probeHotelId) ? probeHotelId : 1085,
|
||
|
|
room_id: probeRoom,
|
||
|
|
device_id: `probe-device-${probeTs}`,
|
||
|
|
is_send: 0,
|
||
|
|
udp_raw: 'abc\u0000def',
|
||
|
|
extra: { source: 'probe', note: 'with\\u0000nul' },
|
||
|
|
app_version: 'v1\u0000.2',
|
||
|
|
launcher_version: 'launcher-1',
|
||
|
|
config_version: 'cfg-1'
|
||
|
|
};
|
||
|
|
|
||
|
|
const client = new kafka.KafkaClient(kafkaClientOptions);
|
||
|
|
|
||
|
|
const producer = new kafka.Producer(client);
|
||
|
|
|
||
|
|
producer.on('ready', () => {
|
||
|
|
producer.send(
|
||
|
|
[{ topic, messages: JSON.stringify(payload) }],
|
||
|
|
(err) => {
|
||
|
|
if (err) {
|
||
|
|
console.error(`[probe] publish failed: ${err.message}`);
|
||
|
|
process.exit(1);
|
||
|
|
}
|
||
|
|
console.log(`[probe] published topic=${topic} ts_ms=${probeTs} hotel_id=${payload.hotel_id} room_id=${probeRoom}`);
|
||
|
|
producer.close(() => client.close(() => process.exit(0)));
|
||
|
|
}
|
||
|
|
);
|
||
|
|
});
|
||
|
|
|
||
|
|
producer.on('error', (err) => {
|
||
|
|
console.error(`[probe] producer error: ${err.message}`);
|
||
|
|
process.exit(1);
|
||
|
|
});
|