@tencent-rtc/mcp 1.0.0

package/@tencent/polaris/dist/limiter/index.js

@@ -0,0 +1,149 @@
+"use strict";
+var _a, _b;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Limiter = void 0;
+const process_1 = require("process");
+const errors_1 = require("../errors");
+const plugins_1 = require("../plugins");
+const console_1 = require("../plugins/logging/console");
+const memory_1 = require("../plugins/registry/memory");
+const unirate_1 = require("../plugins/shaping/unirate");
+const warmup_1 = require("../plugins/shaping/warmup");
+const registry_1 = require("../registry");
+const skeleton_1 = require("../skeleton");
+const utils_1 = require("../utils");
+const ratelimit_1 = require("./ratelimit");
+const isRatelimitServicePlugin = ((plugin) => (plugin.type & plugins_1.PluginType.RatelimitService) !== 0);
+const kRatelimit = Symbol("kRatelimit");
+const kRegistry = Symbol("kRegistry");
+const kCurrentInstances = Symbol("kCurrentInstances");
+const KMaxInstances = Symbol("KMaxInstances");
+class Limiter extends skeleton_1.SkeletonClass {
+    /**
+     * Create Limiter
+     * @param plugins 插件
+     * @param options 配置参数（可选）
+     */
+    constructor(plugins, options = {}) {
+        Limiter[kCurrentInstances] += 1;
+        if (Limiter[kCurrentInstances] >= Limiter[KMaxInstances]) {
+            (0, skeleton_1.emitMaxInstancesExceededWarning)("Consumer", Limiter[KMaxInstances]);
+        }
+        // #region plugins
+        const pluginList = {
+            [plugins_1.PluginType.RatelimitService]: plugins[plugins_1.PluginType.RatelimitService],
+            [plugins_1.PluginType.TrafficShaping]: (plugins[plugins_1.PluginType.TrafficShaping] || []).concat(new unirate_1.UnirateTrafficShaping(), new warmup_1.WarmUpTrafficShaping()),
+            [plugins_1.PluginType.NamingService]: plugins[plugins_1.PluginType.NamingService],
+            [plugins_1.PluginType.LocalRegistry]: plugins[plugins_1.PluginType.LocalRegistry] || new memory_1.MemoryOnlyRegistry(),
+            [plugins_1.PluginType.TraceLogging]: plugins[plugins_1.PluginType.TraceLogging] || [new console_1.ConsoleTraceLogging()],
+            [plugins_1.PluginType.StatReporter]: plugins[plugins_1.PluginType.StatReporter] || []
+        };
+        const naming = pluginList[plugins_1.PluginType.NamingService];
+        /**
+         * 在某些特殊场景下，为了避免死循环与内存开销，
+         * `NamingService` 与 `StatReporter` 插件实现在一个对象内
+         */
+        if ((0, skeleton_1.isStatReporterPlugin)(naming)) {
+            pluginList[plugins_1.PluginType.StatReporter].push(naming);
+        }
+        /**
+         * 在 `RatelimitService`（为空）没有配置情况下，
+         * 查找 `NamingService` 插件是否实现此功能
+         */
+        if (typeof pluginList[plugins_1.PluginType.RatelimitService] === "undefined") {
+            if (isRatelimitServicePlugin(naming)) {
+                pluginList[plugins_1.PluginType.RatelimitService] = naming;
+            }
+            else {
+                throw new errors_1.ConfigError("RatelimitService plugin not provided");
+            }
+        }
+        super(pluginList, options);
+        // #endregion
+        this[kRegistry] = new registry_1.LocalRegistry(this[skeleton_1.kLogger], naming, this.plugins[plugins_1.PluginType.LocalRegistry], this.plugins[plugins_1.PluginType.StatReporter], this.plugins[plugins_1.PluginType.RatelimitService], options);
+        this[kRatelimit] = new ratelimit_1.Ratelimit(this[skeleton_1.kGlobal], this[skeleton_1.kLogger], this[kRegistry], this.plugins[plugins_1.PluginType.RatelimitService], this.plugins[plugins_1.PluginType.TrafficShaping], options);
+    }
+    /**
+     * 当前可实例化的最大次数
+     */
+    static get maxInstances() {
+        return Limiter[KMaxInstances];
+    }
+    /**
+     * 设置可实例化的最大次数
+     * @param n 最大次数
+     */
+    static setMaxInstances(n) {
+        if (typeof n !== "number" || n < 0 || Number.isNaN(n)) {
+            throw new errors_1.ArgumentError("|n| must be a non-negative number");
+        }
+        Limiter[KMaxInstances] = n;
+    }
+    /**
+     * 销毁（释放）所占资源
+     *
+     * @description
+     * dispose order:
+     *  Ratelimit -> Registry -> [plguins]
+     *
+     * @param sync 是否同步释放（可选）
+     */
+    dispose(sync = false) {
+        if (this[skeleton_1.kDisposed]) {
+            return;
+        }
+        this[kRatelimit].dispose();
+        this[kRegistry].dispose();
+        super.dispose(sync);
+        Limiter[kCurrentInstances] -= 1;
+        if (Limiter[kCurrentInstances] < 0) {
+            (0, utils_1.UNREACHABLE)();
+        }
+    }
+    /**
+     * 通过条件获取配额分配对象
+     * @param namespace 命名空间
+     * @param service 服务名
+     * @param amount 请求的配额数
+     * @param cluster 集群名（可选）
+     * @param labels 标签集合（可选）
+     * @param id 唯一 ID（可选）
+     */
+    async acquire(namespace, service, amount, cluster = "", labels = {}, id) {
+        const startTime = (0, process_1.uptime)();
+        let err;
+        try {
+            this[skeleton_1.kMaybeAlreadyDisposed]();
+            const rule = await this[kRatelimit].query(namespace, service, cluster, labels, id);
+            return {
+                quotas: this[kRatelimit].acquire(namespace, service, amount, rule),
+                id: rule === null || rule === void 0 ? void 0 : rule.id
+            };
+        }
+        catch (e) {
+            err = e;
+        }
+        finally {
+            this[skeleton_1.kAPICollector]("Limiter.acquire", ((0, process_1.uptime)() - startTime) * utils_1.kSeconds, err);
+        }
+        throw err;
+    }
+}
+exports.Limiter = Limiter;
+_a = Limiter, _b = kCurrentInstances;
+/**
+ * note:
+ * 由于类包含专用标识符时，`super` 调用必须是构造函数中的第一个语句。
+ * 所以这里采用 `Symbol` 来实现 __私有静态属性__
+ */
+Limiter[_b] = 0;
+(() => {
+    let maxInstances = 3;
+    const { POLARIS_LIMITER_INSTANCE_LIMIT } = process.env;
+    if (POLARIS_LIMITER_INSTANCE_LIMIT !== undefined) {
+        maxInstances = parseInt(POLARIS_LIMITER_INSTANCE_LIMIT, 10);
+        maxInstances = (maxInstances === 0 /** Unlimited */ ? Infinity : maxInstances);
+    }
+    _a.setMaxInstances(maxInstances);
+})();
+//# sourceMappingURL=index.js.map

package/@tencent/polaris/dist/limiter/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/limiter/index.ts"],"names":[],"mappings":";;;;AAAA,qCAAiC;AAEjC,sCAAuD;AAIvD,wCAAoF;AACpF,wDAAiE;AACjE,uDAAgE;AAChE,wDAAmE;AACnE,sDAAiE;AACjE,0CAAkE;AAClE,0CAAsK;AACtK,oCAA8D;AAC9D,2CAAiE;AA8BjE,MAAM,wBAAwB,GAAG,CAC/B,CAAC,MAAc,EAAoC,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,GAAG,oBAAU,CAAC,gBAAgB,CAAC,KAAK,CAAC,CACxG,CAAC;AAEF,MAAM,UAAU,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;AACxC,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;AACtC,MAAM,iBAAiB,GAAG,MAAM,CAAC,mBAAmB,CAAC,CAAC;AACtD,MAAM,aAAa,GAAG,MAAM,CAAC,eAAe,CAAC,CAAC;AAE9C,MAAa,OAAQ,SAAQ,wBAAgC;IA8B3D;;;;OAIG;IACH,YAAY,OAAkC,EAAE,UAAmC,EAAE;QACnF,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAChC,IAAI,OAAO,CAAC,iBAAiB,CAAC,IAAI,OAAO,CAAC,aAAa,CAAC,EAAE;YACxD,IAAA,0CAA+B,EAAC,UAAU,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC;SACrE;QAED,kBAAkB;QAClB,MAAM,UAAU,GAAG;YACjB,CAAC,oBAAU,CAAC,gBAAgB,CAAC,EAAE,OAAO,CAAC,oBAAU,CAAC,gBAAgB,CAAC;YACnE,CAAC,oBAAU,CAAC,cAAc,CAAC,EAAE,CAAC,OAAO,CAAC,oBAAU,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,+BAAqB,EAAE,EAAE,IAAI,6BAAoB,EAAE,CAAC;YACvI,CAAC,oBAAU,CAAC,aAAa,CAAC,EAAE,OAAO,CAAC,oBAAU,CAAC,aAAa,CAAC;YAC7D,CAAC,oBAAU,CAAC,aAAa,CAAC,EAAE,OAAO,CAAC,oBAAU,CAAC,aAAa,CAAC,IAAI,IAAI,2BAAkB,EAAE;YACzF,CAAC,oBAAU,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC,oBAAU,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,6BAAmB,EAAE,CAAC;YAC1F,CAAC,oBAAU,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC,oBAAU,CAAC,YAAY,CAAC,IAAI,EAAE;SAClE,CAAC;QAEF,MAAM,MAAM,GAAG,UAAU,CAAC,oBAAU,CAAC,aAAa,CAAC,CAAC;QAEpD;;;WAGG;QACH,IAAI,IAAA,+BAAoB,EAAC,MAAM,CAAC,EAAE;YAChC,UAAU,CAAC,oBAAU,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;SAClD;QAED;;;WAGG;QACH,IAAI,OAAO,UAAU,CAAC,oBAAU,CAAC,gBAAgB,CAAC,KAAK,WAAW,EAAE;YAClE,IAAI,wBAAwB,CAAC,MAAM,CAAC,EAAE;gBACpC,UAAU,CAAC,oBAAU,CAAC,gBAAgB,CAAC,GAAG,MAAM,CAAC;aAClD;iBAAM;gBACL,MAAM,IAAI,oBAAW,CAAC,sCAAsC,CAAC,CAAC;aAC/D;SACF;QAED,KAAK,CAAC,UAA+B,EAAE,OAAO,CAAC,CAAC;QAChD,aAAa;QAEb,IAAI,CAAC,SAAS,CAAC,GAAG,IAAI,wBAAa,CACjC,IAAI,CAAC,kBAAO,CAAC,EACb,MAAM,EACN,IAAI,CAAC,OAAO,CAAC,oBAAU,CAAC,aAAa,CAAC,EACtC,IAAI,CAAC,OAAO,CAAC,oBAAU,CAAC,YAAY,CAAC,EACrC,IAAI,CAAC,OAAO,CAAC,oBAAU,CAAC,gBAAgB,CAAC,EACzC,OAAO,CACR,CAAC;QACF,IAAI,CAAC,UAAU,CAAC,GAAG,IAAI,qBAAS,CAC9B,IAAI,CAAC,kBAAO,CAAC,EACb,IAAI,CAAC,kBAAO,CAAC,EACb,IAAI,CAAC,SAAS,CAAC,EACf,IAAI,CAAC,OAAO,CAAC,oBAAU,CAAC,gBAAgB,CAAC,EACzC,IAAI,CAAC,OAAO,CAAC,oBAAU,CAAC,cAAc,CAAC,EACvC,OAAO,CACR,CAAC;IACJ,CAAC;IAnFD;;OAEG;IACH,MAAM,KAAK,YAAY;QACrB,OAAO,OAAO,CAAC,aAAa,CAAC,CAAC;IAChC,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,eAAe,CAAC,CAAS;QAC9B,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE;YACrD,MAAM,IAAI,sBAAa,CAAC,mCAAmC,CAAC,CAAC;SAC9D;QACD,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC;IAqED;;;;;;;;OAQG;IACM,OAAO,CAAC,IAAI,GAAG,KAAK;QAC3B,IAAI,IAAI,CAAC,oBAAS,CAAC,EAAE;YACnB,OAAO;SACR;QAED,IAAI,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,CAAC;QAC3B,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;QAE1B,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAEpB,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAChC,IAAI,OAAO,CAAC,iBAAiB,CAAC,GAAG,CAAC,EAAE;YAClC,IAAA,mBAAW,GAAE,CAAC;SACf;IACH,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,OAAO,CAAC,SAAiB,EAAE,OAAe,EAAE,MAAc,EAAE,OAAO,GAAG,EAAE,EAAE,SAAmB,EAAE,EAAE,EAAW;QAChH,MAAM,SAAS,GAAG,IAAA,gBAAM,GAAE,CAAC;QAE3B,IAAI,GAAsB,CAAC;QAC3B,IAAI;YACF,IAAI,CAAC,gCAAqB,CAAC,EAAE,CAAC;YAC9B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YACnF,OAAO;gBACL,MAAM,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC;gBAClE,EAAE,EAAE,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,EAAE;aACb,CAAC;SACH;QAAC,OAAO,CAAC,EAAE;YACV,GAAG,GAAG,CAAU,CAAC;SAClB;gBAAS;YACR,IAAI,CAAC,wBAAa,CAAC,CAAC,iBAAiB,EAAE,CAAC,IAAA,gBAAM,GAAE,GAAG,SAAS,CAAC,GAAG,gBAAQ,EAAE,GAAG,CAAC,CAAC;SAChF;QAED,MAAM,GAAG,CAAC;IACZ,CAAC;;AAlJH,0BA6JC;mBAvJiB,iBAAiB;AALjC;;;;GAIG;AACY,WAAmB,GAAG,CAAE,CAAA;AA8IvC;IACE,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,MAAM,EAAE,8BAA8B,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC;IACvD,IAAI,8BAA8B,KAAK,SAAS,EAAE;QAChD,YAAY,GAAG,QAAQ,CAAC,8BAA8B,EAAE,EAAE,CAAC,CAAC;QAC5D,YAAY,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;KAChF;IACD,EAAI,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC;AACrC,CAAC,GAAA,CAAA"}

package/@tencent/polaris/dist/limiter/ratelimit.d.ts

@@ -0,0 +1,27 @@
+import { GlobalOptions } from "../global";
+import { Logger } from "../logging";
+import { Metadata } from "../metadata";
+import { RatelimitServicePlugin, TrafficShapingPlugin } from "../plugins";
+import { LocalRegistry } from "../registry";
+import { RatelimitRule } from "../rules";
+import { GlobalRatelimiterOptions, LocalRatelimiterOptions } from "./ratelimiter";
+declare const kDefaultOptions: LocalRatelimiterOptions | GlobalRatelimiterOptions;
+export declare type RatelimitOptions = typeof kDefaultOptions;
+export declare type Quota = Promise<(() => Promise<void>) | void>;
+export declare class Ratelimit {
+    private readonly global;
+    private readonly logger;
+    private readonly registry;
+    private readonly ratelimit;
+    private readonly shaping;
+    private readonly limiters;
+    private readonly options;
+    private disposed;
+    constructor(global: GlobalOptions, logger: Logger, registry: LocalRegistry, ratelimit: RatelimitServicePlugin, shaping: TrafficShapingPlugin[], options?: Partial<RatelimitOptions>);
+    dispose(): void;
+    query(namespace: string, service: string, cluster: string | undefined, labels: Metadata, id?: string): Promise<RatelimitRule | undefined>;
+    acquire(namespace: string, service: string, amount: number, rule?: RatelimitRule): Quota[];
+    private allocation;
+    private produce;
+}
+export {};

package/@tencent/polaris/dist/limiter/ratelimit.js

@@ -0,0 +1,134 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Ratelimit = void 0;
+const errors_1 = require("../errors");
+const metadata_1 = require("../metadata");
+const plugins_1 = require("../plugins");
+const rules_1 = require("../rules");
+const utils_1 = require("../utils");
+const ratelimiter_1 = require("./ratelimiter");
+const kDefaultOptions = {
+    outdatedPeriod: 2,
+    idlePeriod: 10,
+    instanceUpdateTime: 5 * utils_1.kSeconds,
+    updatePercent: 0.8
+};
+class Ratelimit {
+    constructor(global, logger, registry, ratelimit, shaping, options) {
+        this.global = global;
+        this.logger = logger;
+        this.registry = registry;
+        this.ratelimit = ratelimit;
+        this.shaping = shaping;
+        this.limiters = new WeakMap();
+        this.disposed = false;
+        this.options = Object.assign(Object.assign({}, kDefaultOptions), options);
+    }
+    dispose() {
+        this.disposed = true;
+    }
+    async query(namespace, service, cluster = "", labels, id) {
+        if (this.disposed) {
+            throw new errors_1.StateError("Already disposed");
+        }
+        const rules = await this.registry.fetch(plugins_1.RegistryCategory.Ratelimit, namespace, service);
+        let selectedRule;
+        if (typeof id === "string" && id !== "") { /** fast case */
+            selectedRule = rules.find(rule => rule.id === id);
+        }
+        if (typeof selectedRule === "undefined") {
+            const { globalVariables } = this.global;
+            // eslint-disable-next-line @typescript-eslint/prefer-for-of
+            for (let i = 0; i < rules.length; i += 1) { /** slow case */
+                const rule = rules[i];
+                if (rule.cluster === cluster
+                    && (0, metadata_1.isMetadataMatch)(labels, rule.labels, globalVariables)
+                    && (selectedRule === undefined || rule.priority > selectedRule.priority)) {
+                    selectedRule = rule;
+                }
+            }
+        }
+        return selectedRule;
+    }
+    acquire(namespace, service, amount, rule) {
+        const quotas = [];
+        if (this.disposed) {
+            const err = new errors_1.StateError("Already disposed");
+            for (let i = 0; i < amount; i += 1) {
+                quotas.push(Promise.reject(err));
+            }
+            return quotas;
+        }
+        /**
+         * 当规则不存在时，默认放通所有配额
+         */
+        if (typeof rule === "undefined") {
+            for (let i = 0; i < amount; i += 1) {
+                quotas.push(Promise.resolve());
+            }
+            return quotas;
+        }
+        let limiter;
+        if (!this.limiters.has(rule)) {
+            switch (rule.type) {
+                case rules_1.LimitType.Local: {
+                    limiter = new ratelimiter_1.LocalRatelimiter(rule, this.options);
+                    break;
+                }
+                case rules_1.LimitType.Global: {
+                    limiter = new ratelimiter_1.GlobalRatelimiter(namespace, service, rule, this.logger, this.registry, this.ratelimit, this.options);
+                    break;
+                }
+                default: {
+                    (0, utils_1.UNREACHABLE)();
+                }
+            }
+            this.limiters.set(rule, limiter);
+        }
+        let shaping;
+        if (rule.action !== "") {
+            shaping = this.shaping.find(plugin => plugin.name === rule.action);
+            if (typeof shaping === "undefined") {
+                throw new errors_1.PluginError(`action(${rule.action}) implement not found`);
+            }
+        }
+        for (let i = 0; i < amount; i += 1) {
+            let quota = Promise.resolve();
+            if (rule.enable) {
+                if (typeof shaping !== "undefined") {
+                    quota = quota
+                        .then(() => limiter.getPartition())
+                        .then(partition => shaping.inFlow(rule, partition));
+                }
+                quota = quota
+                    .then(() => this.allocation(rule))
+                    .then(result => this.produce(rule, result));
+            }
+            quotas.push(quota);
+        }
+        return quotas;
+    }
+    allocation(rule) {
+        const limiter = this.limiters.get(rule);
+        if (typeof limiter === "undefined") {
+            throw new errors_1.StateError("current rate limit rule has been changed, all allocations are rejected");
+        }
+        return limiter.consume();
+    }
+    produce(rule, result) {
+        if (!result) {
+            throw new errors_1.StateError("no available quota");
+        }
+        if (rule.resource === rules_1.LimitResource.Concurrency) {
+            return () => {
+                const limiter = this.limiters.get(rule);
+                if (typeof limiter === "undefined") {
+                    throw new errors_1.StateError("current rate limit rule has been changed, all releases are rejected");
+                }
+                return limiter.return();
+            };
+        }
+    }
+}
+exports.Ratelimit = Ratelimit;
+//# sourceMappingURL=ratelimit.js.map

package/@tencent/polaris/dist/limiter/ratelimit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ratelimit.js","sourceRoot":"","sources":["../../src/limiter/ratelimit.ts"],"names":[],"mappings":";;;AAAA,sCAAoD;AAGpD,0CAAwD;AACxD,wCAA4F;AAE5F,oCAAmE;AACnE,oCAAiD;AACjD,+CAAoI;AAEpI,MAAM,eAAe,GAAuD;IAC1E,cAAc,EAAE,CAAC;IACjB,UAAU,EAAE,EAAE;IACd,kBAAkB,EAAE,CAAC,GAAG,gBAAQ;IAChC,aAAa,EAAE,GAAG;CACnB,CAAC;AAMF,MAAa,SAAS;IAKpB,YACmB,MAAqB,EACrB,MAAc,EACd,QAAuB,EACvB,SAAiC,EACjC,OAA+B,EAChD,OAAmC;QALlB,WAAM,GAAN,MAAM,CAAe;QACrB,WAAM,GAAN,MAAM,CAAQ;QACd,aAAQ,GAAR,QAAQ,CAAe;QACvB,cAAS,GAAT,SAAS,CAAwB;QACjC,YAAO,GAAP,OAAO,CAAwB;QATjC,aAAQ,GAAG,IAAI,OAAO,EAA8B,CAAC;QAE9D,aAAQ,GAAG,KAAK,CAAC;QAUvB,IAAI,CAAC,OAAO,GAAG,gCACV,eAAe,GACf,OAAO,CACkB,CAAC;IACjC,CAAC;IAED,OAAO;QACL,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,SAAiB,EAAE,OAAe,EAAE,OAAO,GAAG,EAAE,EAAE,MAAgB,EAAE,EAAW;QACzF,IAAI,IAAI,CAAC,QAAQ,EAAE;YACjB,MAAM,IAAI,mBAAU,CAAC,kBAAkB,CAAC,CAAC;SAC1C;QAED,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,0BAAgB,CAAC,SAAS,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAExF,IAAI,YAAuC,CAAC;QAE5C,IAAI,OAAO,EAAE,KAAK,QAAQ,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,gBAAgB;YACzD,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;SACnD;QAED,IAAI,OAAO,YAAY,KAAK,WAAW,EAAE;YACvC,MAAM,EAAE,eAAe,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;YACxC,4DAA4D;YAC5D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,EAAE,gBAAgB;gBAC1D,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;gBACtB,IACE,IAAI,CAAC,OAAO,KAAK,OAAO;uBACrB,IAAA,0BAAe,EAAC,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,eAAe,CAAC;uBACrD,CAAC,YAAY,KAAK,SAAS,IAAI,IAAI,CAAC,QAAQ,GAAG,YAAY,CAAC,QAAQ,CAAC,EACxE;oBACA,YAAY,GAAG,IAAI,CAAC;iBACrB;aACF;SACF;QAED,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,OAAO,CAAC,SAAiB,EAAE,OAAe,EAAE,MAAc,EAAE,IAAoB;QAC9E,MAAM,MAAM,GAAY,EAAE,CAAC;QAE3B,IAAI,IAAI,CAAC,QAAQ,EAAE;YACjB,MAAM,GAAG,GAAG,IAAI,mBAAU,CAAC,kBAAkB,CAAC,CAAC;YAC/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;gBAClC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;aAClC;YACD,OAAO,MAAM,CAAC;SACf;QAED;;WAEG;QACH,IAAI,OAAO,IAAI,KAAK,WAAW,EAAE;YAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;gBAClC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;aAChC;YACD,OAAO,MAAM,CAAC;SACf;QAED,IAAI,OAAoB,CAAC;QACzB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;YAC5B,QAAQ,IAAI,CAAC,IAAI,EAAE;gBACjB,KAAK,iBAAS,CAAC,KAAK,CAAC,CAAC;oBACpB,OAAO,GAAG,IAAI,8BAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;oBACnD,MAAM;iBACP;gBACD,KAAK,iBAAS,CAAC,MAAM,CAAC,CAAC;oBACrB,OAAO,GAAG,IAAI,+BAAiB,CAC7B,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CACnF,CAAC;oBACF,MAAM;iBACP;gBACD,OAAO,CAAC,CAAC;oBACP,IAAA,mBAAW,GAAE,CAAC;iBACf;aACF;YACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;SAClC;QAED,IAAI,OAAyC,CAAC;QAC9C,IAAI,IAAI,CAAC,MAAM,KAAK,EAAE,EAAE;YACtB,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,MAAM,CAAC,CAAC;YACnE,IAAI,OAAO,OAAO,KAAK,WAAW,EAAE;gBAClC,MAAM,IAAI,oBAAW,CAAC,UAAU,IAAI,CAAC,MAAM,uBAAuB,CAAC,CAAC;aACrE;SACF;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YAClC,IAAI,KAAK,GAAU,OAAO,CAAC,OAAO,EAAE,CAAC;YAErC,IAAI,IAAI,CAAC,MAAM,EAAE;gBACf,IAAI,OAAO,OAAO,KAAK,WAAW,EAAE;oBAClC,KAAK,GAAG,KAAK;yBACV,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;yBAClC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,OAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC;iBACxD;gBACD,KAAK,GAAG,KAAK;qBACV,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;qBACjC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;aAC/C;YAED,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;SACpB;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,UAAU,CAAC,IAAmB;QACpC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,OAAO,OAAO,KAAK,WAAW,EAAE;YAClC,MAAM,IAAI,mBAAU,CAAC,wEAAwE,CAAC,CAAC;SAChG;QAED,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAEO,OAAO,CAAC,IAAmB,EAAE,MAAe;QAClD,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,mBAAU,CAAC,oBAAoB,CAAC,CAAC;SAC5C;QAED,IAAI,IAAI,CAAC,QAAQ,KAAK,qBAAa,CAAC,WAAW,EAAE;YAC/C,OAAO,GAAG,EAAE;gBACV,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBACxC,IAAI,OAAO,OAAO,KAAK,WAAW,EAAE;oBAClC,MAAM,IAAI,mBAAU,CAAC,qEAAqE,CAAC,CAAC;iBAC7F;gBAED,OAAO,OAAO,CAAC,MAAM,EAAE,CAAC;YAC1B,CAAC,CAAC;SACH;IACH,CAAC;CACF;AApJD,8BAoJC"}

package/@tencent/polaris/dist/limiter/ratelimiter/base.d.ts

@@ -0,0 +1,17 @@
+import { BasicBucket } from "../bucket";
+export declare const deduction: <T extends BasicBucket>(buckets: T[], tokens: number, preroll: boolean) => boolean;
+export interface Ratelimiter {
+    /**
+     * 消费令牌
+     * @param preroll 是否实际扣减
+     */
+    consume(preroll?: boolean): Promise<boolean>;
+    /**
+     * 放回令牌
+     */
+    return(): Promise<void>;
+    /**
+     * 获得当前分区数
+     */
+    getPartition(): Promise<number>;
+}

package/@tencent/polaris/dist/limiter/ratelimiter/base.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.deduction = void 0;
+const deduction = (buckets, tokens, preroll) => {
+    const isEnough = tokens > 0 ? buckets.every(bucket => bucket.consume(tokens, true)) : true;
+    if (!isEnough || preroll) {
+        return isEnough;
+    }
+    buckets.forEach((bucket) => {
+        bucket.consume(tokens, false);
+    });
+    return true;
+};
+exports.deduction = deduction;
+//# sourceMappingURL=base.js.map

package/@tencent/polaris/dist/limiter/ratelimiter/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../src/limiter/ratelimiter/base.ts"],"names":[],"mappings":";;;AAEO,MAAM,SAAS,GAAG,CAAwB,OAAY,EAAE,MAAc,EAAE,OAAgB,EAAE,EAAE;IACjG,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAC3F,IAAI,CAAC,QAAQ,IAAI,OAAO,EAAE;QACxB,OAAO,QAAQ,CAAC;KACjB;IAED,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;QACzB,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AAXW,QAAA,SAAS,aAWpB"}

package/@tencent/polaris/dist/limiter/ratelimiter/global.d.ts

@@ -0,0 +1,59 @@
+import { Logger } from "../../logging";
+import { RatelimitServicePlugin } from "../../plugins";
+import { LocalRegistry } from "../../registry";
+import { RatelimitRule } from "../../rules";
+import { Ratelimiter } from "./base";
+export interface GlobalRatelimiterOptions {
+    /**
+     * 远端令牌桶，过期判定周期数
+     */
+    outdatedPeriod: number;
+    /**
+     * 令牌桶空闲回收，判定周期数
+     */
+    idlePeriod: number;
+    /**
+     * 实例数更新时间 (ms)
+     */
+    instanceUpdateTime: number;
+    /**
+     * 使用了百分之多少配额后，启动一次实时上报
+     * 范围 (0,1]
+     */
+    updatePercent: number;
+}
+/**
+ * 全局令牌桶
+ * 对应 `LimitType.Global` 全局限流
+ */
+export declare class GlobalRatelimiter implements Ratelimiter {
+    private readonly namespace;
+    private readonly service;
+    private readonly rule;
+    private readonly logger;
+    private readonly registry;
+    private readonly backend;
+    private readonly options;
+    private readonly buckets;
+    private readonly taskInfo;
+    private indexes?;
+    private wakePromise;
+    private partition;
+    private idlePeriods;
+    private isAccessed;
+    private maxDuration;
+    private status;
+    private readonly updateConfig;
+    constructor(namespace: string, service: string, rule: RatelimitRule, logger: Logger, registry: LocalRegistry, backend: RatelimitServicePlugin, options: GlobalRatelimiterOptions);
+    consume(preroll?: boolean): Promise<boolean>;
+    return(): Promise<void>;
+    getPartition(): Promise<number>;
+    private dispose;
+    private wake;
+    private sleep;
+    private scheduleTask;
+    private updatePartition;
+    private syncQuota;
+    private checkOutdated;
+    private shouldSleep;
+}