feat: extend TopicPartitionSession to inherit from EventEmitter and emit stop/end events

Signed-off-by: Vladislav Polyakov <[email protected]>

Author: polRk

packages/topic/src/partition-session.ts

@@ -1,4 +1,11 @@
-export class TopicPartitionSession {
+import { EventEmitter } from "node:stream"
+
+type TopicPartitionSessionEmitterMap = {
+	"stop": []
+	"end": []
+}
+
+export class TopicPartitionSession extends EventEmitter<TopicPartitionSessionEmitterMap> {
 	/**
 	 * Partition session identifier.
 	 */
@@ -35,6 +42,8 @@ export class TopicPartitionSession {
 	 * @param topicPath - The path of the topic.
 	 */
 	constructor(partitionSessionId: bigint, partitionId: bigint, topicPath: string) {
+		super();
+
 		this.partitionSessionId = partitionSessionId;
 		this.partitionId = partitionId;
 		this.topicPath = topicPath;
@@ -49,10 +58,12 @@ export class TopicPartitionSession {
 	}
 
 	stop(): void {
+		this.emit("stop");
 		this.#stopped = true;
 	}
 
 	end(): void {
+		this.emit("end");
 		this.#ended = true;
 	}
 }