watson/umami
1
0
Fork 0
mirror of https://github.com/umami-software/umami.git synced 2025-12-06 01:18:00 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

Added timeout for Kafka producer.

Browse source
This commit is contained in:
Mike Cao 2025-01-17 19:10:00 -08:00
parent a02a6c76a6
commit 1da81150d5
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/lib/kafka.ts
View file

@ -4,6 +4,9 @@ import { KAFKA, KAFKA_PRODUCER } from 'lib/db';
import * as tls from 'tls';
const log = debug('umami:kafka');
const CONNECT_TIMEOUT = 5000;
const SEND_TIMEOUT = 3000;
const ACKS = 1;
let kafka: Kafka;
let producer: Producer;
@ -31,7 +34,7 @@ function getClient() {
const client: Kafka = new Kafka({
clientId: 'umami',
brokers: brokers,
connectionTimeout: 3000,
connectionTimeout: CONNECT_TIMEOUT,
logLevel: logLevel.ERROR,
...ssl,
});
@ -71,7 +74,8 @@ async function sendMessage(
value: JSON.stringify(message),
},
],
acks: 1,
timeout: SEND_TIMEOUT,
acks: ACKS,
});
}
@ -83,7 +87,8 @@ async function sendMessages(topic: string, messages: { [key: string]: string | n
messages: messages.map(a => {
return { value: JSON.stringify(a) };
}),
acks: 1,
timeout: SEND_TIMEOUT,
acks: ACKS,
});
}