@tencent-rtc/mcp 1.0.0

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

@@ -0,0 +1,397 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GlobalRatelimiter = void 0;
+const process_1 = require("process");
+const errors_1 = require("../../errors");
+const plugins_1 = require("../../plugins");
+const rules_1 = require("../../rules");
+const utils_1 = require("../../utils");
+const bucket_1 = require("../bucket");
+const base_1 = require("./base");
+const friendlyTaskType = (task) => {
+    switch (task) {
+        case 3 /* CheckOutdated */:
+            return "CheckOutdated";
+        case 4 /* ShouldSleep */:
+            return "ShouldSleep";
+        case 2 /* SyncQuota */:
+            return "SyncQuota";
+        case 1 /* UpdatePartition */:
+            return "UpdatePartition";
+        default:
+            (0, utils_1.UNREACHABLE)();
+    }
+};
+var GlobalStatus;
+(function (GlobalStatus) {
+    /** 释放 */
+    GlobalStatus[GlobalStatus["Dispose"] = 0] = "Dispose";
+    /** 休眠 */
+    GlobalStatus[GlobalStatus["Sleep"] = 1] = "Sleep";
+    /** 正常 */
+    GlobalStatus[GlobalStatus["Normal"] = 2] = "Normal";
+    /** 已过期 */
+    GlobalStatus[GlobalStatus["Outdated"] = 3] = "Outdated";
+})(GlobalStatus || (GlobalStatus = {}));
+/**
+ * 全局令牌桶
+ * 对应 `LimitType.Global` 全局限流
+ */
+class GlobalRatelimiter {
+    constructor(namespace, service, rule, logger, registry, backend, options) {
+        var _a, _b, _c, _d;
+        this.namespace = namespace;
+        this.service = service;
+        this.rule = rule;
+        this.logger = logger;
+        this.registry = registry;
+        this.backend = backend;
+        this.options = options;
+        this.buckets = Object.create(null);
+        this.taskInfo = Object.create(null);
+        this.partition = 0;
+        this.idlePeriods = 0;
+        this.isAccessed = false;
+        this.status = GlobalStatus.Sleep;
+        if (rule.amounts.length === 0) {
+            throw new errors_1.ArgumentError("no valid |QuotaConfig|");
+        }
+        let durations = [];
+        rule.amounts.forEach(({ duration }) => {
+            if (!(duration >= 1 * utils_1.kSeconds)) {
+                throw new errors_1.ArgumentError(`duration(${duration}ms) must be larger than or equal to 1s`);
+            }
+            durations.push(~~(duration / utils_1.kSeconds));
+        });
+        this.maxDuration = Math.max(...durations) * utils_1.kSeconds;
+        let interval = (_b = (_a = this.rule.report) === null || _a === void 0 ? void 0 : _a.interval) !== null && _b !== void 0 ? _b : 0;
+        if (interval > 0) {
+            durations.push(~~(interval / utils_1.kSeconds));
+        }
+        durations = durations.filter(duration => duration > 0);
+        if (durations.length === 0) {
+            (0, utils_1.UNREACHABLE)();
+        }
+        interval = ((0, utils_1.GreatestCommonDivisor)(...durations) * utils_1.kSeconds) / 10;
+        if (Number.isNaN(interval)) {
+            (0, utils_1.UNREACHABLE)();
+        }
+        let percent = (_d = (_c = this.rule.report) === null || _c === void 0 ? void 0 : _c.percent) !== null && _d !== void 0 ? _d : 0;
+        if (!(percent > 0)) {
+            percent = this.options.updatePercent;
+        }
+        this.updateConfig = {
+            interval,
+            percent
+        };
+    }
+    async consume(preroll = false) {
+        if (this.status === GlobalStatus.Dispose) {
+            return false;
+        }
+        this.isAccessed = true;
+        if (this.status === GlobalStatus.Sleep) {
+            if (typeof this.wakePromise === "undefined") {
+                this.wakePromise = this.wake();
+            }
+            await this.wakePromise;
+            return this.consume(preroll);
+        }
+        if (typeof this.indexes === "undefined") {
+            (0, utils_1.UNREACHABLE)();
+        }
+        const { local, remote, allocated } = this.indexes;
+        let result;
+        switch (this.status) {
+            case GlobalStatus.Normal: {
+                result = (0, base_1.deduction)(remote, 1, true);
+                break;
+            }
+            case GlobalStatus.Outdated: {
+                result = (0, base_1.deduction)(local, 1, true);
+                break;
+            }
+            default: {
+                (0, utils_1.UNREACHABLE)();
+            }
+        }
+        if (result && !preroll) {
+            (0, base_1.deduction)(local, 1, false);
+            (0, base_1.deduction)(remote, 1, false);
+            (0, base_1.deduction)(allocated, -1, false);
+            if (allocated.some(({ remainingTokens }, index) => {
+                const { partitionTokens } = local[index];
+                return (remainingTokens / partitionTokens) > this.updateConfig.percent;
+            })) {
+                /*
+                 * note:
+                 *  异步同步远端配额，主流程尽快返回给调用者
+                 */
+                // eslint-disable-next-line @typescript-eslint/unbound-method
+                this.scheduleTask(2 /* SyncQuota */, this.syncQuota, this.updateConfig.interval, true)
+                    .catch(err => this.logger.error("[GlobalRatelimiter] [consume], call SyncQuota task", err));
+            }
+        }
+        return result;
+    }
+    async return() {
+        if (this.status === GlobalStatus.Dispose) {
+            return;
+        }
+        this.isAccessed = true;
+        if (this.status === GlobalStatus.Sleep) {
+            if (typeof this.wakePromise === "undefined") {
+                this.wakePromise = this.wake();
+            }
+            await this.wakePromise;
+            return this.return();
+        }
+        if (typeof this.indexes === "undefined") {
+            (0, utils_1.UNREACHABLE)();
+        }
+        const { local, remote, allocated } = this.indexes;
+        if (!(0, base_1.deduction)(allocated, 1, false)) {
+            (0, utils_1.UNREACHABLE)();
+        }
+        (0, base_1.deduction)(local, -1, false);
+        (0, base_1.deduction)(remote, -1, false);
+    }
+    async getPartition() {
+        this.isAccessed = true;
+        if (this.status === GlobalStatus.Sleep) {
+            if (typeof this.wakePromise === "undefined") {
+                this.wakePromise = this.wake();
+            }
+            await this.wakePromise;
+            return this.getPartition();
+        }
+        return this.partition;
+    }
+    dispose() {
+        this.sleep();
+        this.status = GlobalStatus.Dispose;
+    }
+    // #region Wake-sleep
+    async wake() {
+        try {
+            // eslint-disable-next-line @typescript-eslint/unbound-method
+            await this.scheduleTask(1 /* UpdatePartition */, this.updatePartition, this.options.instanceUpdateTime, true);
+        }
+        catch (e) {
+            this.wakePromise = undefined;
+            throw e;
+        }
+        /**
+         * 仅当第一次调用（桶列表为空）时才构建，
+         * 唤醒时直接使用现有桶
+         */
+        if (typeof this.indexes === "undefined") {
+            const localIndex = [];
+            const remoteIndex = [];
+            const allocatedIndex = [];
+            this.indexes = {
+                local: localIndex,
+                remote: remoteIndex,
+                allocated: allocatedIndex
+            };
+            // eslint-disable-next-line @typescript-eslint/naming-convention
+            let Bucket;
+            switch (this.rule.resource) {
+                case rules_1.LimitResource.Concurrency: {
+                    Bucket = bucket_1.BasicBucket;
+                    break;
+                }
+                case rules_1.LimitResource.QPS: {
+                    Bucket = bucket_1.InternalBucket;
+                    break;
+                }
+                default: {
+                    (0, utils_1.UNREACHABLE)();
+                }
+            }
+            this.rule.amounts.forEach(({ amount, duration }) => {
+                const local = new Bucket(amount, this.partition, false, duration, this.options);
+                const allocated = new Bucket(0, 1, true, duration, this.options);
+                const remote = new bucket_1.ExternalBucket(amount, this.partition);
+                this.buckets[duration] = { local, allocated, remote };
+                localIndex.push(local);
+                remoteIndex.push(remote);
+                allocatedIndex.push(allocated);
+            });
+        }
+        try {
+            // eslint-disable-next-line @typescript-eslint/unbound-method
+            await this.scheduleTask(2 /* SyncQuota */, this.syncQuota, this.updateConfig.interval, true);
+        }
+        catch (e) {
+            this.wakePromise = undefined;
+            throw e;
+        }
+        this.scheduleTask(3 /* CheckOutdated */, this.checkOutdated, this.updateConfig.interval * this.options.outdatedPeriod);
+        // eslint-disable-next-line @typescript-eslint/unbound-method
+        this.scheduleTask(4 /* ShouldSleep */, this.shouldSleep, this.maxDuration * this.options.idlePeriod);
+        this.wakePromise = undefined;
+        this.status = GlobalStatus.Normal;
+    }
+    sleep() {
+        if (this.status === GlobalStatus.Sleep || this.status === GlobalStatus.Dispose) {
+            return;
+        }
+        for (let i = 0 /* Begin */ + 1; i < 5 /* End */; i += 1) {
+            const task = this.taskInfo[i];
+            if (task.timer) {
+                clearInterval(task.timer);
+                task.timer = undefined;
+            }
+        }
+        this.idlePeriods = 0;
+        this.isAccessed = false;
+        this.status = GlobalStatus.Sleep;
+    }
+    async scheduleTask(type, timeout, interval, immediate = false) {
+        let task = this.taskInfo[type];
+        if (typeof task === "undefined") {
+            task = Object.create(null);
+            this.taskInfo[type] = task;
+        }
+        if (task.timer) {
+            clearInterval(task.timer);
+            task.timer = undefined;
+        }
+        task.timer = setInterval(() => {
+            if (task.isRunning) {
+                return;
+            }
+            task.isRunning = true;
+            const result = timeout.call(this, task);
+            if (typeof result === "object") {
+                void result.catch(err => this.logger.error(`[GlobalRatelimiter] [scheduleTask], call ${friendlyTaskType(type)} task`, err)).then(() => (task.isRunning = false));
+            }
+            else {
+                task.isRunning = false;
+            }
+        }, interval).unref();
+        if (immediate) {
+            task.isRunning = true;
+            try {
+                await timeout.call(this, task);
+            }
+            finally {
+                task.isRunning = false;
+            }
+        }
+    }
+    async updatePartition() {
+        if (this.registry.isDisposed) {
+            this.dispose();
+            return;
+        }
+        const instances = await this.registry.fetch(plugins_1.RegistryCategory.Instance, this.namespace, this.service);
+        let partition = 0;
+        instances.forEach((instance) => {
+            if (instance.status === 0 /* Normal */) { /** 仅计算健康状态的实例数 */
+                partition += 1;
+            }
+        });
+        if (partition !== this.partition) {
+            this.logger.info(`[GlobalRatelimiter] [updatePartition], partitions(${this.partition}) has changed to ${partition}`);
+            this.partition = partition;
+            Object.values(this.buckets).forEach(({ local, remote }) => {
+                local.setPartition(partition);
+                remote.setPartition(partition);
+            });
+        }
+    }
+    async syncQuota(task) {
+        if (this.backend.isDisposed) {
+            this.dispose();
+            return;
+        }
+        const allocated = Object.keys(this.buckets).map((duration) => {
+            const bucket = this.buckets[duration].allocated;
+            let amount;
+            switch (this.rule.resource) {
+                case rules_1.LimitResource.QPS: {
+                    amount = bucket.drain();
+                    break;
+                }
+                case rules_1.LimitResource.Concurrency: {
+                    amount = bucket.remainingTokens;
+                    break;
+                }
+                default: {
+                    (0, utils_1.UNREACHABLE)();
+                }
+            }
+            return {
+                amount,
+                duration: parseInt(duration, 10),
+                period: bucket.currentPeriod
+            };
+        });
+        let response;
+        try {
+            response = await this.backend.acquireQuota(this.namespace, this.service, this.rule, allocated);
+        }
+        catch (e) {
+            if (this.rule.resource === rules_1.LimitResource.QPS) {
+                /**
+                 * 如上报失败且 `allocated` 令牌桶还在当前周期时，
+                 * 将扣减数加回，下次继续上报
+                 */
+                allocated.forEach(({ amount, duration, period }) => {
+                    const bucket = this.buckets[duration].allocated;
+                    if (bucket.currentPeriod === period) {
+                        bucket.consume(amount * -1, false);
+                    }
+                });
+            }
+            throw e;
+        }
+        response.forEach(({ amount, duration }) => {
+            const bucket = this.buckets[duration];
+            if (typeof bucket === "undefined") {
+                (0, utils_1.UNREACHABLE)();
+            }
+            bucket.remote.update(amount);
+        });
+        /**
+         * 仅当状态为 `GlobalStatus.Outdated` 才进行切换，
+         * 避免状态竞争问题
+         */
+        if (this.status === GlobalStatus.Outdated) {
+            this.status = GlobalStatus.Normal;
+        }
+        task.timestamp = (0, process_1.uptime)();
+    }
+    checkOutdated() {
+        const updatedTime = this.taskInfo[2 /* SyncQuota */].timestamp;
+        if (this.status === GlobalStatus.Normal /** 仅当状态为 `GlobalStatus.Normal` 才进行切换，避免状态竞争问题 */
+            && updatedTime !== undefined
+            && (0, process_1.uptime)() - updatedTime >= (this.updateConfig.interval * this.options.outdatedPeriod) / utils_1.kSeconds) {
+            this.status = GlobalStatus.Outdated;
+            this.logger.debug("[GlobalRatelimiter] [checkOutdated], status has changed to Outdated");
+        }
+    }
+    shouldSleep() {
+        if (this.isAccessed) {
+            this.idlePeriods = 0;
+        }
+        else {
+            this.idlePeriods += 1;
+        }
+        this.isAccessed = false;
+        /*
+         * 为了降低消耗，当超过 `this.options.idlePeriod` 周期没有访问时，
+         * 自动进入休眠状态（停止一切定时器）
+         *
+         * note:
+         *  如外部不再强引用（WeakMap 为弱引用）当前对象，当进入休眠状态后，当前对象会被自动回收
+         */
+        if (this.idlePeriods > this.options.idlePeriod) {
+            this.sleep();
+        }
+    }
+}
+exports.GlobalRatelimiter = GlobalRatelimiter;
+//# sourceMappingURL=global.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"global.js","sourceRoot":"","sources":["../../../src/limiter/ratelimiter/global.ts"],"names":[],"mappings":";;;AAAA,qCAAiC;AAEjC,yCAA6C;AAG7C,2CAAyE;AAEzE,uCAAqF;AACrF,uCAA2E;AAC3E,sCAAwE;AACxE,iCAAgD;AAsBhD,MAAM,gBAAgB,GAAG,CAAC,IAAmB,EAAU,EAAE;IACvD,QAAQ,IAAI,EAAE;QACZ;YACE,OAAO,eAAe,CAAC;QACzB;YACE,OAAO,aAAa,CAAC;QACvB;YACE,OAAO,WAAW,CAAC;QACrB;YACE,OAAO,iBAAiB,CAAC;QAC3B;YACE,IAAA,mBAAW,GAAE,CAAC;KACjB;AACH,CAAC,CAAC;AAmBF,IAAK,YASJ;AATD,WAAK,YAAY;IACf,SAAS;IACT,qDAAO,CAAA;IACP,SAAS;IACT,iDAAK,CAAA;IACL,SAAS;IACT,mDAAM,CAAA;IACN,UAAU;IACV,uDAAQ,CAAA;AACV,CAAC,EATI,YAAY,KAAZ,YAAY,QAShB;AA0CD;;;GAGG;AACH,MAAa,iBAAiB;IAY5B,YACmB,SAAiB,EACjB,OAAe,EACf,IAAmB,EACnB,MAAc,EACd,QAAuB,EACvB,OAA+B,EAC/B,OAAiC;;QANjC,cAAS,GAAT,SAAS,CAAQ;QACjB,YAAO,GAAP,OAAO,CAAQ;QACf,SAAI,GAAJ,IAAI,CAAe;QACnB,WAAM,GAAN,MAAM,CAAQ;QACd,aAAQ,GAAR,QAAQ,CAAe;QACvB,YAAO,GAAP,OAAO,CAAwB;QAC/B,YAAO,GAAP,OAAO,CAA0B;QAlBnC,YAAO,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAoC,CAAC;QACjE,aAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAA+C,CAAC;QAGtF,cAAS,GAAG,CAAC,CAAC;QACd,gBAAW,GAAG,CAAC,CAAC;QAChB,eAAU,GAAG,KAAK,CAAC;QAEnB,WAAM,GAAG,YAAY,CAAC,KAAK,CAAC;QAYlC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE;YAC7B,MAAM,IAAI,sBAAa,CAAC,wBAAwB,CAAC,CAAC;SACnD;QAED,IAAI,SAAS,GAAa,EAAE,CAAC;QAC7B,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;YACpC,IAAI,CAAC,CAAC,QAAQ,IAAI,CAAC,GAAG,gBAAQ,CAAC,EAAE;gBAC/B,MAAM,IAAI,sBAAa,CAAC,YAAY,QAAQ,wCAAwC,CAAC,CAAC;aACvF;YACD,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,GAAG,gBAAQ,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC,GAAG,gBAAQ,CAAC;QAErD,IAAI,QAAQ,GAAG,MAAA,MAAA,IAAI,CAAC,IAAI,CAAC,MAAM,0CAAE,QAAQ,mCAAI,CAAC,CAAC;QAC/C,IAAI,QAAQ,GAAG,CAAC,EAAE;YAChB,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,GAAG,gBAAQ,CAAC,CAAC,CAAC;SACzC;QACD,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;QACvD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;YAC1B,IAAA,mBAAW,GAAE,CAAC;SACf;QACD,QAAQ,GAAG,CAAC,IAAA,6BAAqB,EAAC,GAAG,SAAS,CAAC,GAAG,gBAAQ,CAAC,GAAG,EAAE,CAAC;QACjE,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE;YAC1B,IAAA,mBAAW,GAAE,CAAC;SACf;QAED,IAAI,OAAO,GAAG,MAAA,MAAA,IAAI,CAAC,IAAI,CAAC,MAAM,0CAAE,OAAO,mCAAI,CAAC,CAAC;QAC7C,IAAI,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,EAAE;YAClB,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;SACtC;QAED,IAAI,CAAC,YAAY,GAAG;YAClB,QAAQ;YACR,OAAO;SACR,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAO,GAAG,KAAK;QAC3B,IAAI,IAAI,CAAC,MAAM,KAAK,YAAY,CAAC,OAAO,EAAE;YACxC,OAAO,KAAK,CAAC;SACd;QAED,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QAEvB,IAAI,IAAI,CAAC,MAAM,KAAK,YAAY,CAAC,KAAK,EAAE;YACtC,IAAI,OAAO,IAAI,CAAC,WAAW,KAAK,WAAW,EAAE;gBAC3C,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;aAChC;YACD,MAAM,IAAI,CAAC,WAAW,CAAC;YACvB,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;SAC9B;QAED,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,WAAW,EAAE;YACvC,IAAA,mBAAW,GAAE,CAAC;SACf;QAED,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QAElD,IAAI,MAAe,CAAC;QACpB,QAAQ,IAAI,CAAC,MAAM,EAAE;YACnB,KAAK,YAAY,CAAC,MAAM,CAAC,CAAC;gBACxB,MAAM,GAAG,IAAA,gBAAS,EAAC,MAAM,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;gBACpC,MAAM;aACP;YACD,KAAK,YAAY,CAAC,QAAQ,CAAC,CAAC;gBAC1B,MAAM,GAAG,IAAA,gBAAS,EAAC,KAAK,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;gBACnC,MAAM;aACP;YACD,OAAO,CAAC,CAAC;gBACP,IAAA,mBAAW,GAAE,CAAC;aACf;SACF;QAED,IAAI,MAAM,IAAI,CAAC,OAAO,EAAE;YACtB,IAAA,gBAAS,EAAC,KAAK,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;YAC3B,IAAA,gBAAS,EAAC,MAAM,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;YAC5B,IAAA,gBAAS,EAAC,SAAS,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YAEhC,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,KAAK,EAAE,EAAE;gBAChD,MAAM,EAAE,eAAe,EAAE,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;gBACzC,OAAO,CAAC,eAAe,GAAG,eAAe,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;YACzE,CAAC,CAAC,EAAE;gBACF;;;mBAGG;gBACH,6DAA6D;gBAC7D,IAAI,CAAC,YAAY,oBAAqB,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,IAAI,CAAC;qBACpF,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,oDAAoD,EAAE,GAAG,CAAC,CAAC,CAAC;aAC/F;SACF;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,MAAM;QACV,IAAI,IAAI,CAAC,MAAM,KAAK,YAAY,CAAC,OAAO,EAAE;YACxC,OAAO;SACR;QAED,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QAEvB,IAAI,IAAI,CAAC,MAAM,KAAK,YAAY,CAAC,KAAK,EAAE;YACtC,IAAI,OAAO,IAAI,CAAC,WAAW,KAAK,WAAW,EAAE;gBAC3C,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;aAChC;YACD,MAAM,IAAI,CAAC,WAAW,CAAC;YACvB,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC;SACtB;QAED,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,WAAW,EAAE;YACvC,IAAA,mBAAW,GAAE,CAAC;SACf;QAED,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QAElD,IAAI,CAAC,IAAA,gBAAS,EAAC,SAAS,EAAE,CAAC,EAAE,KAAK,CAAC,EAAE;YACnC,IAAA,mBAAW,GAAE,CAAC;SACf;QAED,IAAA,gBAAS,EAAC,KAAK,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QAC5B,IAAA,gBAAS,EAAC,MAAM,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QAEvB,IAAI,IAAI,CAAC,MAAM,KAAK,YAAY,CAAC,KAAK,EAAE;YACtC,IAAI,OAAO,IAAI,CAAC,WAAW,KAAK,WAAW,EAAE;gBAC3C,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;aAChC;YACD,MAAM,IAAI,CAAC,WAAW,CAAC;YACvB,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC;SAC5B;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAEO,OAAO;QACb,IAAI,CAAC,KAAK,EAAE,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC;IACrC,CAAC;IAED,qBAAqB;IACb,KAAK,CAAC,IAAI;QAChB,IAAI;YACF,6DAA6D;YAC7D,MAAM,IAAI,CAAC,YAAY,0BAA2B,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC;SAChH;QAAC,OAAO,CAAC,EAAE;YACV,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;YAC7B,MAAM,CAAC,CAAC;SACT;QAED;;;WAGG;QACH,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,WAAW,EAAE;YACvC,MAAM,UAAU,GAAkB,EAAE,CAAC;YACrC,MAAM,WAAW,GAAqB,EAAE,CAAC;YACzC,MAAM,cAAc,GAAkB,EAAE,CAAC;YACzC,IAAI,CAAC,OAAO,GAAG;gBACb,KAAK,EAAE,UAAU;gBACjB,MAAM,EAAE,WAAW;gBACnB,SAAS,EAAE,cAAc;aAC1B,CAAC;YAEF,gEAAgE;YAChE,IAAI,MAAkD,CAAC;YACvD,QAAQ,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;gBAC1B,KAAK,qBAAa,CAAC,WAAW,CAAC,CAAC;oBAC9B,MAAM,GAAG,oBAAW,CAAC;oBACrB,MAAM;iBACP;gBACD,KAAK,qBAAa,CAAC,GAAG,CAAC,CAAC;oBACtB,MAAM,GAAG,uBAAc,CAAC;oBACxB,MAAM;iBACP;gBACD,OAAO,CAAC,CAAC;oBACP,IAAA,mBAAW,GAAE,CAAC;iBACf;aACF;YAED,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE;gBACjD,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;gBAChF,MAAM,SAAS,GAAG,IAAI,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;gBACjE,MAAM,MAAM,GAAG,IAAI,uBAAc,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;gBAE1D,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;gBAEtD,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACvB,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACzB,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACjC,CAAC,CAAC,CAAC;SACJ;QAED,IAAI;YACF,6DAA6D;YAC7D,MAAM,IAAI,CAAC,YAAY,oBAAqB,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;SAC/F;QAAC,OAAO,CAAC,EAAE;YACV,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;YAC7B,MAAM,CAAC,CAAC;SACT;QAED,IAAI,CAAC,YAAY,wBAES,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CACrG,CAAC;QACF,6DAA6D;QAC7D,IAAI,CAAC,YAAY,sBAAuB,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAEtG,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;IACpC,CAAC;IAEO,KAAK;QACX,IAAI,IAAI,CAAC,MAAM,KAAK,YAAY,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,KAAK,YAAY,CAAC,OAAO,EAAE;YAC9E,OAAO;SACR;QAED,KAAK,IAAI,CAAC,GAAG,gBAAiB,CAAC,EAAE,CAAC,cAAe,EAAE,CAAC,IAAI,CAAC,EAAE;YACzD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAkB,CAAC,CAAC;YAC/C,IAAI,IAAI,CAAC,KAAK,EAAE;gBACd,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAC1B,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;aACxB;SACF;QAED,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC;IACnC,CAAC;IAOO,KAAK,CAAC,YAAY,CAAC,IAAmB,EAAE,OAA8D,EAAE,QAAgB,EAAE,SAAS,GAAG,KAAK;QACjJ,IAAI,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC/B,IAAI,OAAO,IAAI,KAAK,WAAW,EAAE;YAC/B,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAgB,CAAC;YAC1C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;SAC5B;QAED,IAAI,IAAI,CAAC,KAAK,EAAE;YACd,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;SACxB;QAED,IAAI,CAAC,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;YAC5B,IAAI,IAAI,CAAC,SAAS,EAAE;gBAClB,OAAO;aACR;YACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YAEtB,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACxC,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE;gBAC9B,KAAK,MAAM,CAAC,KAAK,CACf,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,4CAA4C,gBAAgB,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CACzG,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC;aACxC;iBAAM;gBACL,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;aACxB;QACH,CAAC,EAAE,QAAQ,CAAC,CAAC,KAAK,EAAE,CAAC;QAErB,IAAI,SAAS,EAAE;YACb,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,IAAI;gBACF,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;aAChC;oBAAS;gBACR,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;aACxB;SACF;IACH,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE;YAC5B,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,OAAO;SACR;QAED,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,0BAAgB,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAErG,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC7B,IAAI,QAAQ,CAAC,MAAM,mBAA0B,EAAE,EAAE,kBAAkB;gBACjE,SAAS,IAAI,CAAC,CAAC;aAChB;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,SAAS,KAAK,IAAI,CAAC,SAAS,EAAE;YAChC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,qDAAqD,IAAI,CAAC,SAAS,oBAAoB,SAAS,EAAE,CAAC,CAAC;YACrH,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;YAC3B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE;gBACxD,KAAK,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;gBAC9B,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;YACjC,CAAC,CAAC,CAAC;SACJ;IACH,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,IAAyB;QAC/C,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE;YAC3B,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,OAAO;SACR;QAED,MAAM,SAAS,GAAyC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;YACjG,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,SAA2B,CAAC;YAClE,IAAI,MAAc,CAAC;YACnB,QAAQ,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;gBAC1B,KAAK,qBAAa,CAAC,GAAG,CAAC,CAAC;oBACtB,MAAM,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC;oBACxB,MAAM;iBACP;gBACD,KAAK,qBAAa,CAAC,WAAW,CAAC,CAAC;oBAC9B,MAAM,GAAG,MAAM,CAAC,eAAe,CAAC;oBAChC,MAAM;iBACP;gBACD,OAAO,CAAC,CAAC;oBACP,IAAA,mBAAW,GAAE,CAAC;iBACf;aACF;YACD,OAAO;gBACL,MAAM;gBACN,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC;gBAChC,MAAM,EAAE,MAAM,CAAC,aAAa;aAC7B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;SAChG;QAAC,OAAO,CAAC,EAAE;YACV,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,KAAK,qBAAa,CAAC,GAAG,EAAE;gBAC5C;;;mBAGG;gBACH,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE;oBACjD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,SAA2B,CAAC;oBAClE,IAAI,MAAM,CAAC,aAAa,KAAK,MAAM,EAAE;wBACnC,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;qBACpC;gBACH,CAAC,CAAC,CAAC;aACJ;YACD,MAAM,CAAC,CAAC;SACT;QAED,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE;YACxC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACtC,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE;gBACjC,IAAA,mBAAW,GAAE,CAAC;aACf;YAED,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH;;;WAGG;QACH,IAAI,IAAI,CAAC,MAAM,KAAK,YAAY,CAAC,QAAQ,EAAE;YACzC,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;SACnC;QACD,IAAI,CAAC,SAAS,GAAG,IAAA,gBAAM,GAAE,CAAC;IAC5B,CAAC;IAEO,aAAa;QACnB,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,mBAAoB,CAAC,SAAS,CAAC;QAChE,IACE,IAAI,CAAC,MAAM,KAAK,YAAY,CAAC,MAAM,CAAC,iDAAiD;eAClF,WAAW,KAAK,SAAS;eACzB,IAAA,gBAAM,GAAE,GAAG,WAAW,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,GAAG,gBAAQ,EAClG;YACA,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,QAAQ,CAAC;YACpC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qEAAqE,CAAC,CAAC;SAC1F;IACH,CAAC;IAEO,WAAW;QACjB,IAAI,IAAI,CAAC,UAAU,EAAE;YACnB,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC;SACtB;aAAM;YACL,IAAI,CAAC,WAAW,IAAI,CAAC,CAAC;SACvB;QACD,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QAExB;;;;;;WAMG;QACH,IAAI,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE;YAC9C,IAAI,CAAC,KAAK,EAAE,CAAC;SACd;IACH,CAAC;CAEF;AAraD,8CAqaC"}

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

@@ -0,0 +1,3 @@
+export * from "./global";
+export * from "./local";
+export { Ratelimiter } from "./base";

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

@@ -0,0 +1,19 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./global"), exports);
+__exportStar(require("./local"), exports);
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/limiter/ratelimiter/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAAyB;AACzB,0CAAwB"}

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

@@ -0,0 +1,17 @@
+import { RatelimitRule } from "../../rules";
+import { InternalBucketOptions } from "../bucket";
+import { Ratelimiter } from "./base";
+export declare type LocalRatelimiterOptions = InternalBucketOptions;
+/**
+ * 本地限流
+ * 对应 `LimitType.Local`
+ */
+export declare class LocalRatelimiter implements Ratelimiter {
+    private readonly options;
+    private readonly buckets;
+    private readonly indexes;
+    constructor(rule: RatelimitRule, options: LocalRatelimiterOptions);
+    consume(preroll?: boolean): Promise<boolean>;
+    return(): Promise<void>;
+    getPartition(): Promise<number>;
+}

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

@@ -0,0 +1,52 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LocalRatelimiter = void 0;
+const errors_1 = require("../../errors");
+const rules_1 = require("../../rules");
+const utils_1 = require("../../utils");
+const bucket_1 = require("../bucket");
+const base_1 = require("./base");
+/**
+ * 本地限流
+ * 对应 `LimitType.Local`
+ */
+class LocalRatelimiter {
+    constructor(rule, options) {
+        this.options = options;
+        this.buckets = Object.create(null);
+        this.indexes = [];
+        let Bucket; // eslint-disable-line @typescript-eslint/naming-convention
+        switch (rule.resource) {
+            case rules_1.LimitResource.Concurrency: {
+                Bucket = bucket_1.BasicBucket;
+                break;
+            }
+            case rules_1.LimitResource.QPS: {
+                Bucket = bucket_1.InternalBucket;
+                break;
+            }
+            default: {
+                (0, utils_1.UNREACHABLE)();
+            }
+        }
+        if (rule.amounts.length === 0) {
+            throw new errors_1.ArgumentError("no valid |QuotaConfig|");
+        }
+        rule.amounts.forEach(({ amount, duration }) => {
+            const bucket = new Bucket(amount, 1, false, duration, this.options);
+            this.buckets[duration] = bucket;
+            this.indexes.push(bucket);
+        });
+    }
+    async consume(preroll = false) {
+        return (0, base_1.deduction)(this.indexes, 1, preroll);
+    }
+    async return() {
+        (0, base_1.deduction)(this.indexes, -1, false);
+    }
+    async getPartition() {
+        return 1;
+    }
+}
+exports.LocalRatelimiter = LocalRatelimiter;
+//# sourceMappingURL=local.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"local.js","sourceRoot":"","sources":["../../../src/limiter/ratelimiter/local.ts"],"names":[],"mappings":";;;AAAA,yCAA6C;AAC7C,uCAA2D;AAC3D,uCAA0C;AAC1C,sCAA+E;AAC/E,iCAAgD;AAIhD;;;GAGG;AACH,MAAa,gBAAgB;IAI3B,YAAY,IAAmB,EAAmB,OAAgC;QAAhC,YAAO,GAAP,OAAO,CAAyB;QAHjE,YAAO,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAgC,CAAC;QAC7D,YAAO,GAAkB,EAAE,CAAC;QAG3C,IAAI,MAAkD,CAAC,CAAC,2DAA2D;QACnH,QAAQ,IAAI,CAAC,QAAQ,EAAE;YACrB,KAAK,qBAAa,CAAC,WAAW,CAAC,CAAC;gBAC9B,MAAM,GAAG,oBAAW,CAAC;gBACrB,MAAM;aACP;YACD,KAAK,qBAAa,CAAC,GAAG,CAAC,CAAC;gBACtB,MAAM,GAAG,uBAAc,CAAC;gBACxB,MAAM;aACP;YACD,OAAO,CAAC,CAAC;gBACP,IAAA,mBAAW,GAAE,CAAC;aACf;SACF;QAED,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE;YAC7B,MAAM,IAAI,sBAAa,CAAC,wBAAwB,CAAC,CAAC;SACnD;QAED,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE;YAC5C,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YACpE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC;YAChC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAO,GAAG,KAAK;QAC3B,OAAO,IAAA,gBAAS,EAAC,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,MAAM;QACV,IAAA,gBAAS,EAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,OAAO,CAAC,CAAC;IACX,CAAC;CACF;AA1CD,4CA0CC"}

package/@tencent/polaris/dist/location.d.ts

@@ -0,0 +1,48 @@
+/**
+ * 位置信息
+ */
+export interface Location {
+    /**
+     * CMDB Region
+     */
+    region: string;
+    /**
+     * CMDB Zone
+     */
+    zone: string;
+    /**
+     * CMDB Campus(IDC)
+     */
+    campus: string;
+}
+/**
+ * 位置信息比较级别
+ */
+export declare const enum LocationMatchLevel {
+    Region = 0,
+    Zone = 1,
+    Campus = 2,
+    None = 3
+}
+/**
+ * 判断两个位置信息是否匹配
+ * @param a 位置信息 A
+ * @param b 位置信息 B
+ * @param level 比较级别
+ */
+export declare const isLocationMatch: (a: Location, b: Location, level: LocationMatchLevel) => boolean;
+/**
+ * 判断位置信息是否为空
+ * @param loc 位置信息
+ */
+export declare const isEmptyLocation: (loc?: Location | null | undefined) => boolean;
+/**
+ * 空白位置信息
+ */
+export declare const blankLocation: Location;
+/**
+ * 找到两个位置信息相交的部分
+ * @param a 位置信息 A
+ * @param b 位置信息 B
+ */
+export declare const intersectionLocation: (a: Partial<Location>, b: Partial<Location>) => Location;

package/@tencent/polaris/dist/location.js

@@ -0,0 +1,67 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.intersectionLocation = exports.blankLocation = exports.isEmptyLocation = exports.isLocationMatch = void 0;
+const utils_1 = require("./utils");
+/**
+ * 判断两个位置信息是否匹配
+ * @param a 位置信息 A
+ * @param b 位置信息 B
+ * @param level 比较级别
+ */
+const isLocationMatch = (a, b, level) => {
+    switch (level) {
+        case 2 /* Campus */: {
+            return a.campus === b.campus && a.zone === b.zone && a.region === b.region;
+        }
+        case 1 /* Zone */: {
+            return a.zone === b.zone && a.region === b.region;
+        }
+        case 0 /* Region */: {
+            return a.region === b.region;
+        }
+        case 3 /* None */: {
+            return false;
+        }
+        default: {
+            (0, utils_1.UNREACHABLE)();
+        }
+    }
+};
+exports.isLocationMatch = isLocationMatch;
+/**
+ * 判断位置信息是否为空
+ * @param loc 位置信息
+ */
+const isEmptyLocation = (loc) => {
+    if (!loc) {
+        return true;
+    }
+    return loc.campus === "" && loc.region === "" && loc.zone === "";
+};
+exports.isEmptyLocation = isEmptyLocation;
+/**
+ * 空白位置信息
+ */
+exports.blankLocation = Object.freeze({
+    campus: "",
+    region: "",
+    zone: ""
+});
+/**
+ * 找到两个位置信息相交的部分
+ * @param a 位置信息 A
+ * @param b 位置信息 B
+ */
+const intersectionLocation = (a, b) => {
+    const intersection = Object.assign({}, exports.blankLocation);
+    Object.keys(exports.blankLocation).forEach((key) => {
+        const valueB = b[key];
+        const valueA = a[key];
+        if (valueA === valueB && valueA !== undefined) {
+            intersection[key] = valueA;
+        }
+    });
+    return intersection;
+};
+exports.intersectionLocation = intersectionLocation;
+//# sourceMappingURL=location.js.map

package/@tencent/polaris/dist/location.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"location.js","sourceRoot":"","sources":["../src/location.ts"],"names":[],"mappings":";;;AAAA,mCAAsC;AAgCtC;;;;;GAKG;AACI,MAAM,eAAe,GAAG,CAAC,CAAW,EAAE,CAAW,EAAE,KAAyB,EAAW,EAAE;IAC9F,QAAQ,KAAK,EAAE;QACb,mBAA8B,CAAC,CAAC;YAC9B,OAAO,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,CAAC;SAC5E;QACD,iBAA4B,CAAC,CAAC;YAC5B,OAAO,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,CAAC;SACnD;QACD,mBAA8B,CAAC,CAAC;YAC9B,OAAO,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,CAAC;SAC9B;QACD,iBAA4B,CAAC,CAAC;YAC5B,OAAO,KAAK,CAAC;SACd;QACD,OAAO,CAAC,CAAC;YACP,IAAA,mBAAW,GAAE,CAAC;SACf;KACF;AACH,CAAC,CAAC;AAlBW,QAAA,eAAe,mBAkB1B;AAEF;;;GAGG;AACI,MAAM,eAAe,GAAG,CAAC,GAAqB,EAAW,EAAE;IAChE,IAAI,CAAC,GAAG,EAAE;QACR,OAAO,IAAI,CAAC;KACb;IACD,OAAO,GAAG,CAAC,MAAM,KAAK,EAAE,IAAI,GAAG,CAAC,MAAM,KAAK,EAAE,IAAI,GAAG,CAAC,IAAI,KAAK,EAAE,CAAC;AACnE,CAAC,CAAC;AALW,QAAA,eAAe,mBAK1B;AAEF;;GAEG;AACU,QAAA,aAAa,GAAa,MAAM,CAAC,MAAM,CAAC;IACnD,MAAM,EAAE,EAAE;IACV,MAAM,EAAE,EAAE;IACV,IAAI,EAAE,EAAE;CACT,CAAC,CAAC;AAEH;;;;GAIG;AACI,MAAM,oBAAoB,GAAG,CAAC,CAAoB,EAAE,CAAoB,EAAY,EAAE;IAC3F,MAAM,YAAY,qBAAkB,qBAAa,CAAE,CAAC;IAEpD,MAAM,CAAC,IAAI,CAAC,qBAAa,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;QACzC,MAAM,MAAM,GAAG,CAAC,CAAC,GAAqB,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,CAAC,CAAC,GAAqB,CAAC,CAAC;QACxC,IAAI,MAAM,KAAK,MAAM,IAAI,MAAM,KAAK,SAAS,EAAE;YAC7C,YAAY,CAAC,GAAqB,CAAC,GAAG,MAAM,CAAC;SAC9C;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,YAAY,CAAC;AACtB,CAAC,CAAC;AAZW,QAAA,oBAAoB,wBAY/B"}