groupmq-plus 1.1.2 → 1.1.3

package/dist/index.cjs

@@ -2765,9 +2765,122 @@ function sleep(ms) {
 	return new Promise((r) => setTimeout(r, ms));
 }
 
+//#endregion
+//#region src/strategies/priority-strategy.ts
+var PriorityStrategy = class {
+	constructor(options = {}) {
+		this.cache = /* @__PURE__ */ new Map();
+		this.overrides = /* @__PURE__ */ new Map();
+		this.algorithmConfig = options.algorithm ?? { type: "weighted-random" };
+		this.defaultPriority = options.defaultPriority ?? 1;
+		this.cacheTtlMs = options.cacheTtlMs ?? 5e3;
+		this.onGetPriority = options.onGetPriority;
+	}
+	/**
+	* 手动覆盖某个组的优先级（优先级高于 Redis 配置和 getPriority）
+	* 主要用于测试或临时调整
+	*/
+	setPriority(groupId, priority) {
+		this.overrides.set(groupId, priority);
+	}
+	/**
+	* 清除手动覆盖，恢复使用 Redis 配置或 getPriority
+	*/
+	clearPriority(groupId) {
+		this.overrides.delete(groupId);
+	}
+	/**
+	* 获取组的基础优先级
+	* 优先级来源顺序：overrides > cache > onGetPriority(config) 或 config.priority
+	*/
+	async resolvePriority(queue, groupId) {
+		const override = this.overrides.get(groupId);
+		if (override !== void 0) return override;
+		const now = Date.now();
+		const cached = this.cache.get(groupId);
+		if (cached && cached.expiresAt > now) return cached.priority;
+		const config = await queue.getGroupConfig(groupId);
+		let priority;
+		if (this.onGetPriority) priority = await this.onGetPriority(groupId, config);
+		else priority = config.priority !== 1 ? config.priority : this.defaultPriority;
+		if (this.cacheTtlMs > 0) this.cache.set(groupId, {
+			priority,
+			expiresAt: now + this.cacheTtlMs
+		});
+		return priority;
+	}
+	/**
+	* 严格优先级算法：总是返回优先级最高的组
+	*/
+	selectByStrict(groups) {
+		groups.sort((a, b) => b.priority - a.priority);
+		return groups[0].groupId;
+	}
+	/**
+	* 加权随机算法：根据优先级计算概率选择
+	* 使用 minWeightRatio 确保低优先级组也有机会
+	*/
+	selectByWeightedRandom(groups) {
+		if (groups.length === 1) return groups[0].groupId;
+		const minWeightRatio = this.algorithmConfig.minWeightRatio ?? .1;
+		const minWeight = Math.max(...groups.map((g) => g.priority)) * minWeightRatio;
+		const weights = groups.map((g) => ({
+			groupId: g.groupId,
+			weight: Math.max(g.priority, minWeight)
+		}));
+		const totalWeight = weights.reduce((sum, w) => sum + w.weight, 0);
+		let random = Math.random() * totalWeight;
+		for (const w of weights) {
+			random -= w.weight;
+			if (random <= 0) return w.groupId;
+		}
+		return groups[0].groupId;
+	}
+	/**
+	* 时间衰减算法：等待时间越长，优先级加成越高
+	*/
+	selectByAging(groups) {
+		const now = Date.now();
+		const intervalMs = this.algorithmConfig.intervalMs ?? 6e4;
+		const adjustedGroups = groups.map((g) => {
+			let ageBonus = 0;
+			if (g.oldestTimestamp) {
+				const waitTime = now - g.oldestTimestamp;
+				ageBonus = Math.floor(waitTime / intervalMs);
+			}
+			return {
+				groupId: g.groupId,
+				adjustedPriority: g.priority + ageBonus
+			};
+		});
+		adjustedGroups.sort((a, b) => b.adjustedPriority - a.adjustedPriority);
+		return adjustedGroups[0].groupId;
+	}
+	async getNextGroup(queue) {
+		const readyGroups = await queue.getReadyGroups(0, 100);
+		if (readyGroups.length === 0) return null;
+		const groupInfos = await Promise.all(readyGroups.map(async (groupId) => {
+			const priority = await this.resolvePriority(queue, groupId);
+			const info = {
+				groupId,
+				priority
+			};
+			if (this.algorithmConfig.type === "aging") info.oldestTimestamp = await queue.getGroupOldestTimestamp(groupId);
+			return info;
+		}));
+		switch (this.algorithmConfig.type) {
+			case "strict": return this.selectByStrict(groupInfos);
+			case "aging": return this.selectByAging(groupInfos);
+			case "weighted-random":
+			default: return this.selectByWeightedRandom(groupInfos);
+		}
+	}
+};
+
 //#endregion
 exports.BullBoardGroupMQAdapter = BullBoardGroupMQAdapter;
 exports.Job = Job;
+exports.PriorityStrategy = PriorityStrategy;
 exports.Queue = Queue;
 exports.UnrecoverableError = UnrecoverableError;
 exports.Worker = Worker;