@autofleet/rabbit 4.1.1 → 5.0.0

package/src/lib/celery.ts

@@ -1,89 +0,0 @@
-import { randomUUID } from 'node:crypto';
-import logger from '../logger';
-// Environment configuration
-const config = {
-  host: process.env.RABBITMQ_SERVICE_HOST || 'localhost',
-  username: process.env.RABBITMQ_USERNAME || 'guest',
-  password: process.env.RABBITMQ_PASSWORD || 'guest',
-} as const;
-
-// Type definitions
-interface TaskData {
-    [key: string]: any;
-}
-
-interface TaskMessage {
-    task: string;
-    id: string;
-    args: TaskData[];
-}
-
-interface PublishPayload {
-    properties: {
-        // eslint-disable-next-line camelcase
-        delivery_mode: number;
-        // eslint-disable-next-line camelcase
-        content_type: string;
-    };
-    // eslint-disable-next-line camelcase
-    routing_key: string;
-    payload: string;
-    // eslint-disable-next-line camelcase
-    payload_encoding: string;
-}
-
-interface PublishResponse {
-    routed: boolean;
-}
-
-interface SendTaskOptions {
-    taskName: string;
-    queueName: string;
-}
-
-async function sendCeleryTaskViaHttp(
-  data: TaskData,
-  { taskName, queueName }: SendTaskOptions,
-): Promise<void> {
-  const apiUrl = `http://${config.host}:15672/api/exchanges/%2f/amq.default/publish`;
-
-  const message: TaskMessage = {
-    task: taskName,
-    id: randomUUID(),
-    args: [data],
-  };
-
-  const payload: PublishPayload = {
-    properties: {
-      delivery_mode: 2,
-      content_type: 'application/json',
-    },
-    routing_key: queueName,
-    payload: JSON.stringify(message),
-    payload_encoding: 'string',
-  };
-
-  try {
-    const response = await fetch(apiUrl, {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-        Authorization: `Basic ${Buffer.from(`${config.username}:${config.password}`).toString('base64')}`,
-      },
-      body: JSON.stringify(payload),
-    });
-
-    if (response.ok) {
-      const result: PublishResponse = await response.json();
-      logger.info('Successfully published message:', result);
-    } else {
-      logger.error(`Failed to publish message. Status code: ${response.status}`);
-      logger.error(`Response: ${await response.text()}`);
-    }
-  } catch (error) {
-    logger.error('Error sending request:', error instanceof Error ? error.message : String(error));
-    throw error;
-  }
-}
-
-export { sendCeleryTaskViaHttp, TaskData, SendTaskOptions };

package/src/lib/consts.ts

@@ -1,32 +0,0 @@
-import type { BackoffOptions } from 'exponential-backoff';
-import type { ConsumeOptions } from './types';
-
-export const DEFAULT_DEAD_TTL_TWO_DAYS = 60_000 * 60 * 12;
-export const DEFAULT_LOCK_TIMEOUT = 1000 * 5;
-export const RETRY_HEADER = 'x-retry-count';
-export const TRACING_HEADER = 'x-trace-id';
-export const USER_TRACING_HEADER = 'x-af-user-id';
-export const AUTOMATION_ID_HEADER = 'x-af-automation-id';
-export const USER_OBJECT = 'userObject';
-export const DEFAULT_USE_CONSUME_WITH_LOCK = false;
-export const DEFAULT_OPTIONS = {
-  limit: 1,
-  retries: 1,
-  deadMessageTtl: DEFAULT_DEAD_TTL_TWO_DAYS,
-  lockTimeout: DEFAULT_LOCK_TIMEOUT,
-  useConsumeWithLock: DEFAULT_USE_CONSUME_WITH_LOCK,
-  auditContext: null,
-  enableRabbitTrace: false,
-} satisfies ConsumeOptions;
-
-export const ASSERT_VHOST_EXPONENTIAL_BACKOFF_DEV_ENV_CONFIG = {
-  startingDelay: 500,
-  timeMultiple: 4,
-  numOfAttempts: 5,
-} satisfies BackoffOptions;
-
-export const ASSERT_VHOST_EXPONENTIAL_BACKOFF_PROD_CONFIG = {
-  startingDelay: 1,
-  timeMultiple: 1,
-  numOfAttempts: 5,
-} satisfies BackoffOptions;

package/src/lib/rabbitError.ts

@@ -1,6 +0,0 @@
-export default class RabbitError extends Error {
-  constructor(message: string) {
-    super(message);
-    this.name = 'RabbitError';
-  }
-}

package/src/lib/redis.ts

@@ -1,11 +0,0 @@
-import { createClient } from 'redis';
-
-export type RedisConfig = {
-  host: string;
-  port: number | undefined;
-  prefix?: string;
-}
-
-const getRedisInstance = (config: RedisConfig): ReturnType<typeof createClient> => createClient({ socket: config });
-
-export default getRedisInstance;

package/src/lib/types.ts

@@ -1,77 +0,0 @@
-import type { AmqpConnectionManager } from 'amqp-connection-manager';
-import type { ConsumeMessage, Options, Replies } from 'amqplib';
-
-export interface ExchangesCache {
-  [key: string]: Replies.AssertExchange | undefined;
-}
-
-export interface QueuesCache {
-  [key: string]: Replies.AssertQueue | undefined;
-}
-
-export interface QueueSetupPromisesDictionary {
-  [key: string]: Promise<Replies.AssertQueue> | undefined;
-}
-
-export interface AssertExchangePromisesDictionary {
-  [key: string]: Promise<Replies.AssertExchange> | undefined;
-}
-
-export type CustomMessageHeaders = {
-  redisTimestampValidationKey?: string;
-}
-export type ConsumeMessageOrNull = ConsumeMessage | null;
-
-export interface ConsumeOptions {
-  retries?: number;
-  deadMessageTtl?: number;
-  messageTtl?: number;
-  limit?: number;
-  lockTimeout?: number;
-  useConsumeWithLock?: boolean;
-  auditContext?: any;
-  enableRabbitTrace?: boolean;
-  // TODO: [QUORUM-PHASE-3] Redundant because all the queues are quorum queues
-  isQuorumQueue?: boolean;
-}
-
-export interface NackOptions {
-  skipRetry?: boolean;
-}
-
-export type Message = Omit<ConsumeMessage, 'content'> & { content: any; };
-
-export type CallbackFunction = (msg: Message, ack: () => Promise<void>, nack: (ignored: ConsumeMessageOrNull, nackOptions?: NackOptions) => Promise<void>) => Promise<void>;
-
-export type newChannelOpts = {
-  name?: string;
-  onClose?: null | ((args: any | null) => void);
-};
-export type assertChannelOpts = {
-  channelName?: string;
-  force?: boolean;
-}
-export type AfConsumer = {
-  queue: string;
-  callback: CallbackFunction;
-  options: ConsumeOptions | undefined;
-}
-
-const HA_PROMOTE_ON_FAILURE = 'ha-promote-on-failure';
-
-const HA_PROMOTE_ON_SHUTDOWN = 'ha-promote-on-shutdown';
-
-export const CONSUMER_DEFAULT_OPTIONS: Options.Consume = {
-  arguments: {
-    [HA_PROMOTE_ON_FAILURE]: 'always',
-    [HA_PROMOTE_ON_SHUTDOWN]: 'always',
-  },
-};
-
-export type ConnectionData = {
-  amqpConnection: AmqpConnectionManager | null;
-  creatingConnection: boolean;
-  connectionCreatedEventName: string;
-  connectionFailedEventName: string;
-  blockReconnect: boolean;
-};

package/src/lib/utils.ts

@@ -1,42 +0,0 @@
-import type { ConfirmChannel, Replies } from 'amqplib';
-import type { ChannelWrapper } from 'amqp-connection-manager';
-import type { BackoffOptions } from 'exponential-backoff';
-import { ASSERT_VHOST_EXPONENTIAL_BACKOFF_PROD_CONFIG, ASSERT_VHOST_EXPONENTIAL_BACKOFF_DEV_ENV_CONFIG } from './consts';
-
-const { PROJECT_ID } = process.env;
-
-export const assertExchangeFanout = async (
-  c: ChannelWrapper | ConfirmChannel,
-  exchangeName: string,
-): Promise<Replies.AssertExchange> => c.assertExchange(exchangeName, 'fanout');
-
-export const rand = (): number => Math.floor(Math.random() * 100_000);
-
-interface PromiseWithResolvers<T> {
-  promise: Promise<T>;
-  resolve: (value: T | PromiseLike<T>) => void;
-  reject: (reason?: any) => void;
-}
-interface PromiseCls extends PromiseConstructor {
-  withResolvers<T>(): PromiseWithResolvers<T>;
-}
-/** This is polyfill for `Promise.withResolvers` which exists only on Node v22 and onwards */
-export const createDeferredPromise = <T = void>(): PromiseWithResolvers<T> => {
-  if ((Promise as PromiseCls).withResolvers) {
-    return (Promise as PromiseCls).withResolvers<T>();
-  }
-  let resolve!: PromiseWithResolvers<T>['resolve'];
-  let reject!: PromiseWithResolvers<T>['reject'];
-  const promise = new Promise<T>((res, rej) => {
-    resolve = res;
-    reject = rej;
-  });
-  return { promise, resolve, reject };
-};
-
-const isDevEnv = (): boolean => ['af-experiment-manager', 'dev1-experiment-manager'].includes(PROJECT_ID || '');
-
-export const getAssertVhostExponentialBackoffConfig = (): BackoffOptions => (
-  isDevEnv()
-    ? ASSERT_VHOST_EXPONENTIAL_BACKOFF_DEV_ENV_CONFIG
-    : ASSERT_VHOST_EXPONENTIAL_BACKOFF_PROD_CONFIG);

package/src/logger.ts

@@ -1,5 +0,0 @@
-import Logger from '@autofleet/logger';
-
-const logger = Logger();
-
-export default logger;

package/src/mock/index.ts

@@ -1,25 +0,0 @@
-// eslint-disable-next-line import/no-unresolved
-import 'jest';
-import type { IAfRabbitMq } from '../index';
-
-class RabbitMq implements IAfRabbitMq {
-  ack: any = jest.fn();
-
-  nack: any = jest.fn();
-
-  assertChannel: any = jest.fn();
-
-  assertExchange: any = jest.fn();
-
-  assertQueue: any = jest.fn();
-
-  consume: any = jest.fn();
-
-  consumeFromExchange: any = jest.fn();
-
-  publish: any = jest.fn();
-
-  sendToQueue: any = jest.fn();
-}
-
-export default RabbitMq;

package/src/mock/vitest.ts

@@ -1,24 +0,0 @@
-import { vi } from 'vitest';
-import { IAfRabbitMq } from '../index';
-
-class RabbitMq implements IAfRabbitMq {
-  ack = vi.fn();
-
-  nack = vi.fn();
-
-  assertChannel = vi.fn();
-
-  assertExchange = vi.fn();
-
-  assertQueue = vi.fn();
-
-  consume = vi.fn();
-
-  consumeFromExchange = vi.fn();
-
-  publish = vi.fn();
-
-  sendToQueue = vi.fn();
-}
-
-export default RabbitMq;

package/src/redis-lock.d.ts

@@ -1,6 +0,0 @@
-declare module 'redis-lock' {
-  import type { createClient } from 'redis';
-
-  declare function redisLock(client: ReturnType<typeof createClient>, retryDelay?: number): (lockName: string, timeout?: number) => Promise<() => Promise<void>>;
-  export = redisLock;
-}

package/tsconfig.build.json

@@ -1,5 +0,0 @@
-{
-  "extends": "./tsconfig.json",
-  "include": ["src"],
-  "exclude": ["node_modules", "dist", "src/example.ts", "**/*.test.ts"],
-}

package/tsconfig.json

@@ -1,16 +0,0 @@
-{
-  "compilerOptions": {
-    "target": "es2020",
-    "module": "commonjs",
-    "declaration": true,
-    "outDir": "./dist",
-    "strict": true,
-    "esModuleInterop": true,
-    "experimentalDecorators": true,
-    "emitDecoratorMetadata": true,
-    "skipLibCheck": true,
-    "sourceMap": true,
-  },
-  "exclude": ["node_modules"],
-  "include": ["src/", "vitest.config.ts"],
-}

package/vitest.config.ts

@@ -1,17 +0,0 @@
-// eslint-disable-next-line import/no-unresolved
-import { defineConfig } from 'vitest/config';
-
-export default defineConfig({
-  test: {
-    coverage: {
-      provider: 'v8',
-      include: ['src/**/*.ts'],
-      exclude: ['src/example.ts', 'src/mock/*.ts'],
-      reporter: 'text',
-      thresholds: {
-        // TO DO: Increase the coverage threshold back to 75 in phase 3
-        lines: 65,
-      },
-    },
-  },
-});