All files / redis-smq-api/src/plugins/message-rate/consumer consumer-message-rate-writer.ts

96.87% Statements 31/32
88.88% Branches 8/9
100% Functions 3/3
96.66% Lines 29/30

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 9057x 57x 57x 57x 57x 57x 57x                 57x                                                 24x 24x 24x   24x   24x       24x       24x       24x                     145x 145x 290x 290x 290x 6x 3x 3x 3x   3x 3x 3x       145x        
import { ConsumerAcknowledgedTimeSeries } from './consumer-acknowledged-time-series';
import { ConsumerDeadLetteredTimeSeries } from './consumer-dead-lettered-time-series';
import { QueueAcknowledgedTimeSeries } from './queue-acknowledged-time-series';
import { QueueDeadLetteredTimeSeries } from './queue-dead-lettered-time-series';
import { GlobalAcknowledgedTimeSeries } from './global-acknowledged-time-series';
import { GlobalDeadLetteredTimeSeries } from './global-dead-lettered-time-series';
import { MessageRateWriter } from '../common/message-rate-writer';
import {
  ICallback,
  TQueueParams,
  TRedisClientMulti,
} from 'redis-smq/dist/types';
import { RedisClient } from 'redis-smq/dist/src/common/redis-client/redis-client';
import { IConsumerMessageRateFields } from '../../../../types';
 
export class ConsumerMessageRateWriter extends MessageRateWriter<IConsumerMessageRateFields> {
  protected redisClient: RedisClient;
  protected acknowledgedTimeSeries: ReturnType<
    typeof ConsumerAcknowledgedTimeSeries
  >;
  protected deadLetteredTimeSeries: ReturnType<
    typeof ConsumerDeadLetteredTimeSeries
  >;
  protected queueAcknowledgedRateTimeSeries: ReturnType<
    typeof QueueAcknowledgedTimeSeries
  >;
  protected queueDeadLetteredTimeSeries: ReturnType<
    typeof QueueDeadLetteredTimeSeries
  >;
  protected globalAcknowledgedRateTimeSeries: ReturnType<
    typeof GlobalAcknowledgedTimeSeries
  >;
  protected globalDeadLetteredTimeSeries: ReturnType<
    typeof GlobalDeadLetteredTimeSeries
  >;
  constructor(
    redisClient: RedisClient,
    queue: TQueueParams,
    consumerId: string,
  ) {
    super();
    this.redisClient = redisClient;
    this.globalAcknowledgedRateTimeSeries =
      GlobalAcknowledgedTimeSeries(redisClient);
    this.globalDeadLetteredTimeSeries =
      GlobalDeadLetteredTimeSeries(redisClient);
    this.acknowledgedTimeSeries = ConsumerAcknowledgedTimeSeries(
      redisClient,
      consumerId,
    );
    this.deadLetteredTimeSeries = ConsumerDeadLetteredTimeSeries(
      redisClient,
      consumerId,
    );
    this.queueAcknowledgedRateTimeSeries = QueueAcknowledgedTimeSeries(
      redisClient,
      queue,
    );
    this.queueDeadLetteredTimeSeries = QueueDeadLetteredTimeSeries(
      redisClient,
      queue,
    );
  }
 
  onUpdate(
    ts: number,
    rates: IConsumerMessageRateFields,
    cb: ICallback<void>,
  ): void {
    let multi: TRedisClientMulti | null = null;
    for (const field in rates) {
      multi = multi ?? this.redisClient.multi();
      const value: number = rates[field];
      if (value) {
        if (field === 'acknowledgedRate') {
          this.acknowledgedTimeSeries.add(ts, value, multi);
          this.queueAcknowledgedRateTimeSeries.add(ts, value, multi);
          this.globalAcknowledgedRateTimeSeries.add(ts, value, multi);
        } else {
          this.deadLetteredTimeSeries.add(ts, value, multi);
          this.queueDeadLetteredTimeSeries.add(ts, value, multi);
          this.globalDeadLetteredTimeSeries.add(ts, value, multi);
        }
      }
    }
    if (multi) this.redisClient.execMulti(multi, () => cb());
    else Ecb();
  }
}