Merge pull request #1401 from tulios/fix-message-size-type

Differentiate KafkaMessage type between Record and Message
tulios · Jun 28, 2022 · 48b8bc7 · 48b8bc7
2 parents 196105c + e1b016e
commit 48b8bc7
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/types/index.d.ts b/types/index.d.ts
@@ -631,16 +631,28 @@ export type Broker = {
   }): Promise<any>
 }
 
-export type KafkaMessage = {
+interface MessageSetEntry {
   key: Buffer | null
   value: Buffer | null
   timestamp: string
+  attributes: number
+  offset: string
   size: number
+  headers?: never
+}
+
+interface RecordBatchEntry {
+  key: Buffer | null
+  value: Buffer | null
+  timestamp: string
   attributes: number
   offset: string
-  headers?: IHeaders
+  headers: IHeaders
+  size?: never
 }
 
+export type KafkaMessage = MessageSetEntry | RecordBatchEntry
+
 export interface ProducerRecord {
   topic: string
   messages: Message[]