@nitra/nats 4.2.3 → 4.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nitra/nats",
-  "version": "4.2.3",
+  "version": "4.3.0",
   "description": "nats helper",
   "keywords": [
     "nats",
@@ -27,8 +27,8 @@
     "@nats-io/jetstream": "^3.3.0",
     "@nats-io/nats-core": "^3.3.0",
     "@nats-io/transport-node": "^3.3.0",
-    "@nitra/check-env": "^4.1.0",
-    "@nitra/pino": "^2.10.0",
+    "@nitra/check-env": "^4.1.1",
+    "@nitra/pino": "^2.12.0",
     "js-yaml": "^4.1.1"
   },
   "engines": {

package/src/index.js

@@ -1,3 +1,4 @@
 export { publish } from './publish.js'
-export { read, finish } from './worker.js'
+export { read, finish } from './read.js'
+export { readM, finishM, nakM } from './read-m.js'
 export { getPendingCount } from './pending-count.js'

package/src/nats.js

@@ -3,9 +3,6 @@ import { connect } from '@nats-io/transport-node'
 import { jetstream, jetstreamManager } from '@nats-io/jetstream'
 import { env } from 'node:process'
 
-// AckPolicy
-// export { AckPolicy } from '@nats-io/jetstream'
-
 let nc
 
 // Експортуємо JetStream контекст
@@ -13,6 +10,23 @@ export let js
 // Експортуємо менеджер JetStream
 export let jsm
 
+/**
+ * Припиняє приймати нові повідомлення для існуючих підписок (дренить підписки).
+ * Дає дочитати/доробити те, що вже “в дорозі”/в черзі (залежить від типу підписки й вашого хендлера).
+ * Флашить (дочікується відправки) вже ініційовані публікації.
+ * Після цього закриває з’єднання.
+ * Для CLI-скриптів це потрібно, інакше процес буде "висіти" через відкритий сокет.
+ * @returns {Promise<void>}
+ */
+export async function closeNats() {
+  // якщо фейковий режим або ще не підключались — нічого закривати
+  if (!nc) return
+
+  await nc.drain()
+
+  return nc.closed()
+}
+
 // якщо задані демо дані, то використовуємо фейковий NATS
 if (env.NATS_FAKE_DATA) {
   // Створюємо фейкові об'єкти для js та jsm

package/src/pending-count.js

@@ -2,6 +2,7 @@ import { jsm } from './nats.js'
 import { log } from '@nitra/pino'
 import { env } from 'node:process'
 import { stream } from './stream.js'
+import { ClosedConnectionError } from '@nats-io/nats-core'
 
 /**
  * Повертає кількість непрочитаних (pending) повідомлень для durable consumer.
@@ -17,7 +18,13 @@ export async function getPendingCount(consumer) {
   try {
     const info = await jsm.consumers.info(stream, consumer)
     return info.num_pending + info.num_ack_pending
-  } catch {
+  } catch (error) {
+    // Перевірити що це ClosedConnectionError
+    if (error instanceof ClosedConnectionError) {
+      log.error('getPendingCount: nats connection already closed')
+      return null
+    }
+
     log.error(`consumer ${consumer} not found`)
     return 0
   }

package/src/read-m.js

@@ -0,0 +1,99 @@
+import { log } from '@nitra/pino'
+import { env, exit } from 'node:process'
+import { js, closeNats } from './nats.js'
+import { stream } from './stream.js'
+
+// Кешуємо результат перевірки фейкових даних
+let FAKE_DATA
+if (env.NATS_FAKE_DATA) {
+  try {
+    FAKE_DATA = JSON.parse(env.NATS_FAKE_DATA)
+  } catch (error) {
+    log.error('Failed to parse fake data', { error, data: env.NATS_FAKE_DATA })
+    throw new Error(`Invalid JSON in fake data: ${error.message}`)
+  }
+}
+
+// state variables
+let msgs = []
+
+/**
+ * Зчитує N повідомлень з JetStream для вказаного consumer-а.
+ * @param {string} consumer - durable_name для consumer(за замовчуванням durable_name = subject)
+ * @param {number} count - кількість повідомлень, яку треба зчитати
+ * @returns {Promise<object[]>} - масив декодованих даних повідомлень
+ */
+export async function readM(consumer, count = 1) {
+  if (!Number.isFinite(count) || count <= 0) {
+    throw new Error(`Invalid count: ${count}. Must be a positive number.`)
+  }
+
+  // скидаємо попередній стан (якщо readM викликають повторно без finishM/nakM)
+  msgs = []
+
+  // якщо задані демо дані, то повертаємо їх
+  if (FAKE_DATA) {
+    log.info('env.NATS_FAKE_DATA:', env.NATS_FAKE_DATA)
+
+    if (Array.isArray(FAKE_DATA)) return FAKE_DATA.slice(0, count)
+
+    // якщо передали не масив — повторюємо один і той самий payload N разів
+    return Array.from({ length: count }, () => FAKE_DATA)
+  }
+
+  const consumerObj = await js.consumers.get(stream, consumer)
+
+  // час очікування (в мс), протягом якого fetch() буде чекати повідомлення перед тим, як завершити ітератор без повідомлень.
+  const iter = await consumerObj.fetch({ max_messages: count, expires: 2000 })
+
+  const decodedMessages = []
+  for await (const msg of iter) {
+    msgs.push(msg)
+
+    let decoded
+    try {
+      decoded = JSON.parse(msg.data)
+    } catch (error) {
+      log.error('Failed to parse message data', { error, data: msg.data })
+      msg.ack() // НЕ повертаємо повідомлення назад в чергу при помилці парсингу, бо зациклиться
+      throw new Error(`Invalid JSON in message: ${error.message}`)
+    }
+
+    decodedMessages.push(decoded)
+  }
+
+  if (!decodedMessages.length) {
+    log.info(`${consumer} - no msg...`)
+    exit(0) // якщо не було жодного повідомлення
+  }
+
+  if (decodedMessages.length < count) {
+    log.info(`${consumer} - read ${decodedMessages.length}/${count} msg(s)`)
+  }
+
+  return decodedMessages
+}
+
+/**
+ * Підтверджує (ack) всі прочитані повідомлення.
+ * Якщо не викликати, повідомлення будуть повернуті у чергу (nak) після таймауту.
+ * @returns {Promise<void>}
+ */
+export const finishM = async () => {
+  // oxlint-disable-next-line require-await
+  for (const msg of msgs) msg?.ack()
+  msgs = []
+  return closeNats()
+}
+
+/**
+ * Повертає всі прочитані повідомлення назад в чергу (nak).
+ * @returns {Promise<void>}
+ */
+export const nakM = async () => {
+  // oxlint-disable-next-line require-await
+  for (const msg of msgs) msg?.nak()
+  msgs = []
+
+  return closeNats()
+}

package/src/{worker.js → read.js}

@@ -1,6 +1,6 @@
 import { log } from '@nitra/pino'
 import { env, exit } from 'node:process'
-import { js } from './nats.js'
+import { js, closeNats } from './nats.js'
 import { stream } from './stream.js'
 
 // Кешуємо результат перевірки фейкових даних
@@ -57,7 +57,8 @@ export async function read(consumer) {
 
   const consumerObj = await js.consumers.get(stream, consumer)
 
-  const iter = await consumerObj.fetch({ max_messages: 1, expires: 1000 })
+  // час очікування (в мс), протягом якого fetch() буде чекати повідомлення перед тим, як завершити ітератор без повідомлень.
+  const iter = await consumerObj.fetch({ max_messages: 1, expires: 2000 })
   for await (const msgI of iter) {
     msg = msgI
 
@@ -80,6 +81,7 @@ export async function read(consumer) {
   }
 
   log.info(`${consumer} - no msg...`)
+  await closeNats()
   exit(0) // якщо не було жодного повідомлення
 }
 
@@ -89,8 +91,8 @@ export async function read(consumer) {
  * @returns {Promise<void>}
  */
 export const finish = async () => {
-  // oxlint-disable-next-line require-await
-  return msg?.ack()
+  await msg?.ack()
+  return closeNats()
 }
 
 /**
@@ -98,6 +100,6 @@ export const finish = async () => {
  * @returns {Promise<void>}
  */
 export const nak = async () => {
-  // oxlint-disable-next-line require-await
-  return msg?.nak()
+  await msg?.nak()
+  return closeNats()
 }