use generic messageinfo type

open-telemetry · Apr 23, 2024 · c7fd550 · c7fd550
1 parent f4ff11c
commit c7fd550
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/plugins/node/instrumentation-kafkajs/src/types.ts b/plugins/node/instrumentation-kafkajs/src/types.ts
@@ -24,17 +24,17 @@ export interface KafkajsMessage {
   timestamp?: string;
 }
 
-export interface MessageInfo {
+export interface MessageInfo<T = KafkajsMessage> {
   topic: string;
-  message: KafkajsMessage;
+  message: T;
 }
 
-export interface KafkaProducerCustomAttributeFunction {
-  (span: Span, info: MessageInfo): void;
+export interface KafkaProducerCustomAttributeFunction<T = KafkajsMessage> {
+  (span: Span, info: MessageInfo<T>): void;
 }
 
-export interface KafkaConsumerCustomAttributeFunction {
-  (span: Span, info: MessageInfo): void;
+export interface KafkaConsumerCustomAttributeFunction<T = KafkajsMessage> {
+  (span: Span, info: MessageInfo<T>): void;
 }
 
 export interface KafkaJsInstrumentationConfig extends InstrumentationConfig {