@wildix/xbees-conversations-utils 1.2.0 → 1.2.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (70) hide show
  1. package/dist-cjs/normalization.js +0 -2
  2. package/dist-es/normalization.js +0 -2
  3. package/dist-types/types.d.ts +0 -1
  4. package/package.json +1 -1
  5. package/dist-cjs/rateLimits/RedisCooldown.js +0 -66
  6. package/dist-cjs/rateLimits/constants.js +0 -43
  7. package/dist-cjs/rateLimits/createRateLimitedStreamProxy.js +0 -92
  8. package/dist-cjs/rateLimits/exception/createRateLimitExceededException.js +0 -21
  9. package/dist-cjs/rateLimits/exception/processStreamRateLimitException.js +0 -47
  10. package/dist-cjs/rateLimits/executeStreamRequestWithRateLimitHandling.js +0 -94
  11. package/dist-cjs/rateLimits/generatedMethodNames.js +0 -320
  12. package/dist-cjs/rateLimits/helpers/RedisCooldown.js +0 -174
  13. package/dist-cjs/rateLimits/helpers/createRateLimitExceededException.js +0 -32
  14. package/dist-cjs/rateLimits/helpers/executeWithRateLimitHandling.js +0 -144
  15. package/dist-cjs/rateLimits/helpers/headerUtils.js +0 -40
  16. package/dist-cjs/rateLimits/helpers/isStreamChannelLike.js +0 -23
  17. package/dist-cjs/rateLimits/helpers/processStreamBudgetHeaders.js +0 -93
  18. package/dist-cjs/rateLimits/helpers/processStreamRateLimitException.js +0 -63
  19. package/dist-cjs/rateLimits/helpers/processStreamRateLimitHeaders.js +0 -50
  20. package/dist-cjs/rateLimits/helpers/rateLimitSnapshot.js +0 -31
  21. package/dist-cjs/rateLimits/helpers/registerStreamBudgetInterceptors.js +0 -48
  22. package/dist-cjs/rateLimits/helpers/registerStreamInterceptors.js +0 -58
  23. package/dist-cjs/rateLimits/helpers/splitCallArgsAndOptions.js +0 -21
  24. package/dist-cjs/rateLimits/index.js +0 -7
  25. package/dist-cjs/rateLimits/types.js +0 -3
  26. package/dist-cjs/rateLimits/withStreamRateLimitOptions.js +0 -10
  27. package/dist-es/rateLimits/RedisCooldown.js +0 -62
  28. package/dist-es/rateLimits/constants.js +0 -33
  29. package/dist-es/rateLimits/createRateLimitedStreamProxy.js +0 -88
  30. package/dist-es/rateLimits/exception/createRateLimitExceededException.js +0 -17
  31. package/dist-es/rateLimits/exception/processStreamRateLimitException.js +0 -37
  32. package/dist-es/rateLimits/executeStreamRequestWithRateLimitHandling.js +0 -93
  33. package/dist-es/rateLimits/generatedMethodNames.js +0 -317
  34. package/dist-es/rateLimits/helpers/RedisCooldown.js +0 -170
  35. package/dist-es/rateLimits/helpers/createRateLimitExceededException.js +0 -28
  36. package/dist-es/rateLimits/helpers/executeWithRateLimitHandling.js +0 -140
  37. package/dist-es/rateLimits/helpers/headerUtils.js +0 -34
  38. package/dist-es/rateLimits/helpers/isStreamChannelLike.js +0 -19
  39. package/dist-es/rateLimits/helpers/processStreamBudgetHeaders.js +0 -89
  40. package/dist-es/rateLimits/helpers/processStreamRateLimitException.js +0 -59
  41. package/dist-es/rateLimits/helpers/processStreamRateLimitHeaders.js +0 -46
  42. package/dist-es/rateLimits/helpers/rateLimitSnapshot.js +0 -26
  43. package/dist-es/rateLimits/helpers/registerStreamBudgetInterceptors.js +0 -44
  44. package/dist-es/rateLimits/helpers/registerStreamInterceptors.js +0 -54
  45. package/dist-es/rateLimits/helpers/splitCallArgsAndOptions.js +0 -17
  46. package/dist-es/rateLimits/index.js +0 -4
  47. package/dist-es/rateLimits/types.js +0 -1
  48. package/dist-es/rateLimits/withStreamRateLimitOptions.js +0 -6
  49. package/dist-types/rateLimits/RedisCooldown.d.ts +0 -7
  50. package/dist-types/rateLimits/constants.d.ts +0 -29
  51. package/dist-types/rateLimits/createRateLimitedStreamProxy.d.ts +0 -11
  52. package/dist-types/rateLimits/exception/createRateLimitExceededException.d.ts +0 -12
  53. package/dist-types/rateLimits/exception/processStreamRateLimitException.d.ts +0 -8
  54. package/dist-types/rateLimits/executeStreamRequestWithRateLimitHandling.d.ts +0 -8
  55. package/dist-types/rateLimits/generatedMethodNames.d.ts +0 -11
  56. package/dist-types/rateLimits/helpers/RedisCooldown.d.ts +0 -11
  57. package/dist-types/rateLimits/helpers/createRateLimitExceededException.d.ts +0 -10
  58. package/dist-types/rateLimits/helpers/executeWithRateLimitHandling.d.ts +0 -2
  59. package/dist-types/rateLimits/helpers/headerUtils.d.ts +0 -4
  60. package/dist-types/rateLimits/helpers/isStreamChannelLike.d.ts +0 -6
  61. package/dist-types/rateLimits/helpers/processStreamBudgetHeaders.d.ts +0 -9
  62. package/dist-types/rateLimits/helpers/processStreamRateLimitException.d.ts +0 -3
  63. package/dist-types/rateLimits/helpers/processStreamRateLimitHeaders.d.ts +0 -6
  64. package/dist-types/rateLimits/helpers/rateLimitSnapshot.d.ts +0 -3
  65. package/dist-types/rateLimits/helpers/registerStreamBudgetInterceptors.d.ts +0 -2
  66. package/dist-types/rateLimits/helpers/registerStreamInterceptors.d.ts +0 -2
  67. package/dist-types/rateLimits/helpers/splitCallArgsAndOptions.d.ts +0 -6
  68. package/dist-types/rateLimits/index.d.ts +0 -4
  69. package/dist-types/rateLimits/types.d.ts +0 -68
  70. package/dist-types/rateLimits/withStreamRateLimitOptions.d.ts +0 -6
@@ -209,7 +209,6 @@ function localizeStreamChannel(streamChannel) {
209
209
  const members = Object.values(state.members).filter(({ user_id }) => user_id !== wid);
210
210
  const userId = data?.type === xbees_conversations_client_1.ChannelType.DIRECT && members.length > 0 ? members[0].user_id : null;
211
211
  const unread = state.read[wid]?.unread_messages || 0;
212
- const unreadMark = data?.unreadMarkUsers?.includes(wid) || false;
213
212
  const lastMessage = streamChannel.lastMessage();
214
213
  const normalizedLastMessage = lastMessage ? localizeStreamMessage(lastMessage, normalizedChannel.channelId, wid) : null;
215
214
  const lastMessageAt = normalizedLastMessage ? normalizedLastMessage.createdAt : updatedAt || createdAt;
@@ -224,7 +223,6 @@ function localizeStreamChannel(streamChannel) {
224
223
  updatedAt,
225
224
  isArchived,
226
225
  unread,
227
- unreadMark,
228
226
  isExternal: Boolean(normalizedChannel.external),
229
227
  isBroadcast: Boolean(normalizedChannel.broadcast),
230
228
  isAutoRecord: Boolean(normalizedChannel.autoRecord),
@@ -196,7 +196,6 @@ export function localizeStreamChannel(streamChannel) {
196
196
  const members = Object.values(state.members).filter(({ user_id }) => user_id !== wid);
197
197
  const userId = data?.type === ChannelType.DIRECT && members.length > 0 ? members[0].user_id : null;
198
198
  const unread = state.read[wid]?.unread_messages || 0;
199
- const unreadMark = data?.unreadMarkUsers?.includes(wid) || false;
200
199
  const lastMessage = streamChannel.lastMessage();
201
200
  const normalizedLastMessage = lastMessage ? localizeStreamMessage(lastMessage, normalizedChannel.channelId, wid) : null;
202
201
  const lastMessageAt = normalizedLastMessage ? normalizedLastMessage.createdAt : updatedAt || createdAt;
@@ -211,7 +210,6 @@ export function localizeStreamChannel(streamChannel) {
211
210
  updatedAt,
212
211
  isArchived,
213
212
  unread,
214
- unreadMark,
215
213
  isExternal: Boolean(normalizedChannel.external),
216
214
  isBroadcast: Boolean(normalizedChannel.broadcast),
217
215
  isAutoRecord: Boolean(normalizedChannel.autoRecord),
@@ -31,7 +31,6 @@ export interface LocalizedChannel extends Omit<Channel, 'createdAt' | 'updatedAt
31
31
  lastRead: Date;
32
32
  role: ChannelMemberRole;
33
33
  unread: number;
34
- unreadMark: boolean;
35
34
  updatedAt?: Date;
36
35
  userId: string | null;
37
36
  }
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@wildix/xbees-conversations-utils",
3
- "version": "1.2.0",
3
+ "version": "1.2.1",
4
4
  "description": "",
5
5
  "main": "./dist-cjs/index.js",
6
6
  "module": "./dist-es/index.js",
@@ -1,66 +0,0 @@
1
- 'use strict';
2
- Object.defineProperty(exports, '__esModule', {value: true});
3
- exports.RedisCooldown = void 0;
4
- const constants_1 = require('./constants');
5
-
6
- class RedisCooldown {
7
- static async getRemainingMs(cooldownKey, context) {
8
- const {redis, logger} = context;
9
-
10
- try {
11
- const ttlMs = await redis.pttl(`${constants_1.STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX}${cooldownKey}`);
12
-
13
- if (ttlMs > 0) {
14
- return ttlMs;
15
- }
16
- } catch (error) {
17
- logger.warn('Failed to read stream rate-limit cooldown from Redis', {
18
- cooldownKey,
19
- error,
20
- });
21
- }
22
-
23
- return 0;
24
- }
25
-
26
- static async setRemainingMs(cooldownKey, cooldownMs, context) {
27
- if (cooldownMs <= 0) {
28
- return;
29
- }
30
-
31
- const {redis, logger} = context;
32
- const redisKey = `${constants_1.STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX}${cooldownKey}`;
33
-
34
- try {
35
- await redis.eval(
36
- `
37
- local key = KEYS[1]
38
- local new_ttl = tonumber(ARGV[1])
39
- local current_ttl = redis.call('PTTL', key)
40
-
41
- if current_ttl == -2 then
42
- redis.call('SET', key, '1', 'PX', new_ttl)
43
- return 1
44
- end
45
-
46
- if current_ttl == -1 or current_ttl < new_ttl then
47
- redis.call('PEXPIRE', key, new_ttl)
48
- return 2
49
- end
50
-
51
- return 0
52
- `,
53
- 1,
54
- redisKey,
55
- cooldownMs.toString(),
56
- );
57
- } catch (error) {
58
- logger.warn('Failed to write stream rate-limit cooldown to Redis', {
59
- cooldownKey,
60
- cooldownMs,
61
- error,
62
- });
63
- }
64
- }
65
- }
66
- exports.RedisCooldown = RedisCooldown;
@@ -1,43 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.DEFAULT_OPTIONS = exports.SOFT_THROTTLE_DELAY_HIGH_MS = exports.SOFT_THROTTLE_DELAY_MEDIUM_MS = exports.SOFT_THROTTLE_DELAY_LOW_MS = exports.SOFT_THROTTLE_USAGE_HIGH = exports.SOFT_THROTTLE_USAGE_MEDIUM = exports.SOFT_THROTTLE_USAGE_LOW = exports.BUDGET_COOLDOWN_SLEEP_CHUNK_MS = exports.BUDGET_MINIMAL_DELAY_MAX_MS = exports.BUDGET_MINIMAL_DELAY_MIN_MS = exports.BUDGET_RELAXED_DELAY_MAX_MS = exports.BUDGET_RELAXED_DELAY_MIN_MS = exports.BUDGET_RELAXED_USAGE_THRESHOLD = exports.BUDGET_CLEAR_USAGE_THRESHOLD = exports.BUDGET_DELAY_MAX_MS = exports.BUDGET_DELAY_MIN_MS = exports.BUDGET_THRESHOLD = exports.STREAM_BUDGET_COOLDOWN_KEY = exports.STREAM_RATE_LIMIT_METADATA_KEY_SUFFIX = exports.STREAM_RATE_LIMIT_SOFT_COOLDOWN_KEY = exports.STREAM_RATE_LIMIT_OPTIONS_MARKER = exports.STREAM_BUDGET_COOLDOWN_KEY_PREFIX = exports.STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX = exports.RETRY_AFTER_BUFFER_MS = exports.DEFAULT_JITTER_MS = exports.DEFAULT_RETRY_AFTER_MS = exports.DEFAULT_MAX_DELAY_MS = exports.STREAM_SYNC_METHOD_NAMES = exports.STREAM_SYNC_CHANNEL_RESULT_METHOD_NAMES = exports.STREAM_METHOD_NAMES = exports.STREAM_CHANNEL_RESULT_METHOD_NAMES = exports.CHANNEL_SYNC_PROXY_RESULT_METHOD_NAMES = exports.CHANNEL_SYNC_METHOD_NAMES = exports.CHANNEL_METHOD_NAMES = void 0;
4
- var generatedMethodNames_1 = require("./generatedMethodNames");
5
- Object.defineProperty(exports, "CHANNEL_METHOD_NAMES", { enumerable: true, get: function () { return generatedMethodNames_1.CHANNEL_METHOD_NAMES; } });
6
- Object.defineProperty(exports, "CHANNEL_SYNC_METHOD_NAMES", { enumerable: true, get: function () { return generatedMethodNames_1.CHANNEL_SYNC_METHOD_NAMES; } });
7
- Object.defineProperty(exports, "CHANNEL_SYNC_PROXY_RESULT_METHOD_NAMES", { enumerable: true, get: function () { return generatedMethodNames_1.CHANNEL_SYNC_PROXY_RESULT_METHOD_NAMES; } });
8
- Object.defineProperty(exports, "STREAM_CHANNEL_RESULT_METHOD_NAMES", { enumerable: true, get: function () { return generatedMethodNames_1.STREAM_CHANNEL_RESULT_METHOD_NAMES; } });
9
- Object.defineProperty(exports, "STREAM_METHOD_NAMES", { enumerable: true, get: function () { return generatedMethodNames_1.STREAM_METHOD_NAMES; } });
10
- Object.defineProperty(exports, "STREAM_SYNC_CHANNEL_RESULT_METHOD_NAMES", { enumerable: true, get: function () { return generatedMethodNames_1.STREAM_SYNC_CHANNEL_RESULT_METHOD_NAMES; } });
11
- Object.defineProperty(exports, "STREAM_SYNC_METHOD_NAMES", { enumerable: true, get: function () { return generatedMethodNames_1.STREAM_SYNC_METHOD_NAMES; } });
12
- exports.DEFAULT_MAX_DELAY_MS = 5000;
13
- exports.DEFAULT_RETRY_AFTER_MS = 10000;
14
- exports.DEFAULT_JITTER_MS = 150;
15
- exports.RETRY_AFTER_BUFFER_MS = 1000;
16
- exports.STREAM_RATE_LIMIT_COOLDOWN_KEY_PREFIX = 'stream-rate-limit-cooldown:';
17
- exports.STREAM_BUDGET_COOLDOWN_KEY_PREFIX = 'stream-budget-cooldown:';
18
- exports.STREAM_RATE_LIMIT_OPTIONS_MARKER = '__xbsRateLimitOptions';
19
- exports.STREAM_RATE_LIMIT_SOFT_COOLDOWN_KEY = 'stream-rate-limit-soft-global';
20
- exports.STREAM_RATE_LIMIT_METADATA_KEY_SUFFIX = ':meta';
21
- exports.STREAM_BUDGET_COOLDOWN_KEY = 'stream-budget-global';
22
- exports.BUDGET_THRESHOLD = 0.8;
23
- exports.BUDGET_DELAY_MIN_MS = 30000;
24
- exports.BUDGET_DELAY_MAX_MS = 60000;
25
- exports.BUDGET_CLEAR_USAGE_THRESHOLD = 0.6;
26
- exports.BUDGET_RELAXED_USAGE_THRESHOLD = 0.7;
27
- exports.BUDGET_RELAXED_DELAY_MIN_MS = 5000;
28
- exports.BUDGET_RELAXED_DELAY_MAX_MS = 10000;
29
- exports.BUDGET_MINIMAL_DELAY_MIN_MS = 1000;
30
- exports.BUDGET_MINIMAL_DELAY_MAX_MS = 2000;
31
- exports.BUDGET_COOLDOWN_SLEEP_CHUNK_MS = 3000;
32
- exports.SOFT_THROTTLE_USAGE_LOW = 0.7;
33
- exports.SOFT_THROTTLE_USAGE_MEDIUM = 0.85;
34
- exports.SOFT_THROTTLE_USAGE_HIGH = 0.95;
35
- exports.SOFT_THROTTLE_DELAY_LOW_MS = 500;
36
- exports.SOFT_THROTTLE_DELAY_MEDIUM_MS = 1500;
37
- exports.SOFT_THROTTLE_DELAY_HIGH_MS = 3000;
38
- exports.DEFAULT_OPTIONS = {
39
- maxAttempts: 3,
40
- enableCooldown: true,
41
- maxDelayMs: 5000,
42
- maxRetryableDelayMs: 10000,
43
- };
@@ -1,92 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.createRateLimitedStreamProxy = void 0;
4
- const constants_1 = require("./constants");
5
- const executeWithRateLimitHandling_1 = require("./helpers/executeWithRateLimitHandling");
6
- const isStreamChannelLike_1 = require("./helpers/isStreamChannelLike");
7
- const registerStreamInterceptors_1 = require("./helpers/registerStreamInterceptors");
8
- const splitCallArgsAndOptions_1 = require("./helpers/splitCallArgsAndOptions");
9
- const STREAM_SYNC_METHODS = new Set(constants_1.STREAM_SYNC_METHOD_NAMES);
10
- const STREAM_CHANNEL_RESULT_METHODS = new Set(constants_1.STREAM_CHANNEL_RESULT_METHOD_NAMES);
11
- const STREAM_SYNC_CHANNEL_RESULT_METHODS = new Set(constants_1.STREAM_SYNC_CHANNEL_RESULT_METHOD_NAMES);
12
- const CHANNEL_SYNC_METHODS = new Set(constants_1.CHANNEL_SYNC_METHOD_NAMES);
13
- const CHANNEL_SYNC_PROXY_RESULT_METHODS = new Set(constants_1.CHANNEL_SYNC_PROXY_RESULT_METHOD_NAMES);
14
- function wrapKnownChannelResult(result, dependencies) {
15
- if (Array.isArray(result)) {
16
- return result.map((item) => wrapKnownChannelResult(item, dependencies));
17
- }
18
- if (typeof result !== 'object' || result === null) {
19
- return result;
20
- }
21
- return wrapTarget(result, dependencies);
22
- }
23
- function normalizeStreamCallResult(result, dependencies, wrapAsChannelResult = false) {
24
- if (wrapAsChannelResult) {
25
- return wrapKnownChannelResult(result, dependencies);
26
- }
27
- if (Array.isArray(result)) {
28
- return result.map((item) => normalizeStreamCallResult(item, dependencies));
29
- }
30
- if (!(0, isStreamChannelLike_1.isStreamChannelLike)(result)) {
31
- return result;
32
- }
33
- return wrapTarget(result, dependencies);
34
- }
35
- function createRateLimitedMethodWrapper(target, value, operation, dependencies, wrapAsChannelResult) {
36
- return async (...args) => {
37
- const { callArgs, callOptions } = (0, splitCallArgsAndOptions_1.splitCallArgsAndOptions)(args);
38
- const result = await (0, executeWithRateLimitHandling_1.executeWithRateLimitHandling)(() => Reflect.apply(value, target, callArgs), operation, dependencies.context, {
39
- ...constants_1.DEFAULT_OPTIONS,
40
- ...callOptions,
41
- });
42
- return normalizeStreamCallResult(result, dependencies, wrapAsChannelResult);
43
- };
44
- }
45
- function wrapTarget(target, dependencies) {
46
- const cachedObject = dependencies.cache.get(target);
47
- if (cachedObject) {
48
- return cachedObject;
49
- }
50
- const proxy = new Proxy(target, {
51
- get: (originalTarget, property, receiver) => {
52
- const value = Reflect.get(originalTarget, property, receiver);
53
- if (typeof property !== 'string' || typeof value !== 'function') {
54
- return value;
55
- }
56
- if (STREAM_SYNC_CHANNEL_RESULT_METHODS.has(property)) {
57
- return (...args) => {
58
- const { callArgs } = (0, splitCallArgsAndOptions_1.splitCallArgsAndOptions)(args);
59
- const result = Reflect.apply(value, originalTarget, callArgs);
60
- return normalizeStreamCallResult(result, dependencies, true);
61
- };
62
- }
63
- if (CHANNEL_SYNC_PROXY_RESULT_METHODS.has(property)) {
64
- return (...args) => {
65
- const { callArgs } = (0, splitCallArgsAndOptions_1.splitCallArgsAndOptions)(args);
66
- const result = Reflect.apply(value, originalTarget, callArgs);
67
- if (typeof result === 'object' && result !== null) {
68
- return wrapTarget(result, dependencies);
69
- }
70
- return result;
71
- };
72
- }
73
- if (STREAM_SYNC_METHODS.has(property) || CHANNEL_SYNC_METHODS.has(property)) {
74
- return (...args) => {
75
- const { callArgs } = (0, splitCallArgsAndOptions_1.splitCallArgsAndOptions)(args);
76
- return Reflect.apply(value, originalTarget, callArgs);
77
- };
78
- }
79
- return createRateLimitedMethodWrapper(originalTarget, value, property, dependencies, STREAM_CHANNEL_RESULT_METHODS.has(property));
80
- },
81
- });
82
- dependencies.cache.set(target, proxy);
83
- return proxy;
84
- }
85
- function createRateLimitedStreamProxy(stream, context) {
86
- (0, registerStreamInterceptors_1.registerStreamInterceptors)(stream, context);
87
- return wrapTarget(stream, {
88
- context,
89
- cache: new WeakMap(),
90
- });
91
- }
92
- exports.createRateLimitedStreamProxy = createRateLimitedStreamProxy;
@@ -1,21 +0,0 @@
1
- 'use strict';
2
- Object.defineProperty(exports, '__esModule', {value: true});
3
- exports.createRateLimitExceededException = void 0;
4
- const xbees_conversations_client_1 = require('@wildix/xbees-conversations-client');
5
- const constants_1 = require('../constants');
6
-
7
- function createRateLimitExceededException(options = {}) {
8
- const {rateLimit, message, operation, retryAfterMs = constants_1.DEFAULT_RETRY_AFTER_MS} = options;
9
- const defaultMessage = operation ? `Rate limit exceeded for ${operation}` : 'Rate limit exceeded';
10
-
11
- return new xbees_conversations_client_1.RateLimitExceededException({
12
- $metadata: {},
13
- message: message || defaultMessage,
14
- rateLimitRemaining: rateLimit?.remaining || 0,
15
- retryAfter: rateLimit?.reset ? rateLimit.reset - Date.now() : retryAfterMs,
16
- rateLimit: rateLimit?.limit?.toString() || 'unknown',
17
- rateLimitReset: rateLimit?.reset || Date.now() + retryAfterMs,
18
- });
19
- }
20
-
21
- exports.createRateLimitExceededException = createRateLimitExceededException;
@@ -1,47 +0,0 @@
1
- 'use strict';
2
- Object.defineProperty(exports, '__esModule', {value: true});
3
- exports.processStreamRateLimitException = void 0;
4
- const createRateLimitExceededException_1 = require('./createRateLimitExceededException');
5
-
6
- function isStreamException(error) {
7
- return (
8
- typeof error === 'object' &&
9
- error !== null &&
10
- 'metadata' in error &&
11
- typeof error.metadata === 'object' &&
12
- error.metadata !== null &&
13
- 'responseCode' in error.metadata
14
- );
15
- }
16
-
17
- function isStreamAPIException(error) {
18
- return typeof error === 'object' && !!error.code;
19
- }
20
-
21
- function processStreamRateLimitException(error, logger) {
22
- if (isStreamException(error) && error.metadata?.responseCode === 429) {
23
- const rateLimit = error.metadata?.rateLimit;
24
- logger.warn('Stream rate limit exceeded [StreamException]', {rateLimit});
25
-
26
- return (0, createRateLimitExceededException_1.createRateLimitExceededException)({
27
- rateLimit,
28
- message: error.message || 'Rate limit exceeded',
29
- });
30
- }
31
-
32
- if (isStreamAPIException(error) && (error.code === 429 || error.StatusCode === 429)) {
33
- logger.warn('Stream rate limit exceeded [StreamAPIException]', {code: error.code, StatusCode: error.StatusCode});
34
-
35
- return (0, createRateLimitExceededException_1.createRateLimitExceededException)({
36
- message: error.message || 'Rate limit exceeded',
37
- });
38
- }
39
-
40
- if (typeof error === 'object' && error !== null && 'status' in error && error.status === 429) {
41
- logger.warn('Stream rate limit exceeded [error.status === 429]');
42
-
43
- return (0, createRateLimitExceededException_1.createRateLimitExceededException)({message: 'Rate limit exceeded'});
44
- }
45
- }
46
-
47
- exports.processStreamRateLimitException = processStreamRateLimitException;
@@ -1,94 +0,0 @@
1
- 'use strict';
2
- Object.defineProperty(exports, '__esModule', {value: true});
3
- exports.executeStreamRequestWithRateLimitHandling = void 0;
4
- const promises_1 = require('node:timers/promises');
5
- const constants_1 = require('./constants');
6
- const createRateLimitExceededException_1 = require('./exception/createRateLimitExceededException');
7
- const processStreamRateLimitException_1 = require('./exception/processStreamRateLimitException');
8
- const RedisCooldown_1 = require('./RedisCooldown');
9
-
10
- function calculateRateLimitDelayMs(error, maxDelayMs) {
11
- const retryAfterMs =
12
- typeof error.retryAfter === 'number'
13
- ? error.retryAfter + constants_1.RETRY_AFTER_BUFFER_MS
14
- : constants_1.DEFAULT_RETRY_AFTER_MS;
15
-
16
- if (retryAfterMs > 0) {
17
- const jitterMs = Math.floor(Math.random() * constants_1.DEFAULT_JITTER_MS);
18
-
19
- return Math.min(maxDelayMs, retryAfterMs + jitterMs);
20
- }
21
-
22
- return Math.min(maxDelayMs, constants_1.DEFAULT_FIXED_RATE_LIMIT_DELAY_MS);
23
- }
24
-
25
- async function executeStreamRequestWithRateLimitHandling(execute, operation, context, options = {}) {
26
- const {logger, redis} = context;
27
- const {enableCooldown, maxAttempts, maxDelayMs, maxRetryableDelayMs} = {
28
- ...constants_1.DEFAULT_OPTS_1_ATTEMPT_NO_COOLDOWN,
29
- ...options,
30
- };
31
- let attempt = 1;
32
-
33
- if (enableCooldown) {
34
- const cooldownMs = await RedisCooldown_1.RedisCooldown.getRemainingMs(operation, {redis, logger});
35
-
36
- if (cooldownMs > 0) {
37
- logger.warn('Skipping stream request due to active rate-limit cooldown', {
38
- operation,
39
- cooldown: cooldownMs,
40
- });
41
- throw (0, createRateLimitExceededException_1.createRateLimitExceededException)({
42
- operation,
43
- retryAfterMs: cooldownMs,
44
- });
45
- }
46
- }
47
-
48
- for (;;) {
49
- try {
50
- return await execute();
51
- } catch (error) {
52
- const rateLimitError = (0, processStreamRateLimitException_1.processStreamRateLimitException)(error, logger);
53
-
54
- if (!rateLimitError) {
55
- throw error;
56
- }
57
-
58
- const retryAfterMs = calculateRateLimitDelayMs(rateLimitError, maxDelayMs);
59
-
60
- if (enableCooldown) {
61
- await RedisCooldown_1.RedisCooldown.setRemainingMs(operation, retryAfterMs, {redis, logger});
62
- }
63
-
64
- if (attempt < maxAttempts && retryAfterMs <= maxRetryableDelayMs) {
65
- logger.warn('Retrying stream request after rate limit', {
66
- operation,
67
- attempt,
68
- maxAttempts,
69
- retryAfterMs,
70
- rateLimit: rateLimitError.rateLimit,
71
- rateLimitRemaining: rateLimitError.rateLimitRemaining,
72
- rateLimitReset: rateLimitError.rateLimitReset,
73
- });
74
- await (0, promises_1.setTimeout)(retryAfterMs);
75
- attempt++;
76
- continue;
77
- }
78
-
79
- logger.warn('Propagating stream rate limit w/o retry', {
80
- operation,
81
- attempt,
82
- maxAttempts,
83
- retryAfterMs,
84
- maxRetryableDelayMs,
85
- rateLimit: rateLimitError.rateLimit,
86
- rateLimitRemaining: rateLimitError.rateLimitRemaining,
87
- rateLimitReset: rateLimitError.rateLimitReset,
88
- });
89
- throw rateLimitError;
90
- }
91
- }
92
- }
93
-
94
- exports.executeStreamRequestWithRateLimitHandling = executeStreamRequestWithRateLimitHandling;