@tencent-rtc/mcp 1.0.0

package/@tencent/polaris/dist/plugins/router/polaris/nearby.js

@@ -0,0 +1,145 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PolarisNearbyRouter = exports.MatchMode = void 0;
+const __1 = require("../../..");
+const plugins_1 = require("../../../plugins");
+const utils_1 = require("../../../utils");
+/** 就近路由匹配模式 */
+var MatchMode;
+(function (MatchMode) {
+    /** 不开启就近路由 */
+    MatchMode[MatchMode["Off"] = 0] = "Off";
+    /**
+     * 根据实例元数据信息自动切换，
+     * 如备选实例中存在任意一个未开启就近路由的实例，则不开启，
+     * 否则开启
+     */
+    MatchMode[MatchMode["Auto"] = 1] = "Auto";
+    /** 总是开启就近路由 */
+    MatchMode[MatchMode["Always"] = 2] = "Always";
+})(MatchMode = exports.MatchMode || (exports.MatchMode = {}));
+const kDefaultOptions = {
+    /**
+     * 匹配模式
+     */
+    mode: MatchMode.Auto,
+    /**
+     * 就近路由最大查询范围
+     */
+    maxRange: 0 /* Region */,
+    /**
+     * 就近路由最小查询范围
+     */
+    minRange: 1 /* Zone */,
+    /**
+     * 不健康实例比例达到多少进行降级
+     * 默认为 1，即全部不健康才降级
+     */
+    unhealthyDegrade: 1
+};
+/**
+ * 北极星定义的就近路由策略
+ * 优先匹配最小范围，失败则扩大一级范围重试，直到超过 `maxRange`
+ */
+class PolarisNearbyRouter {
+    constructor(options) {
+        this.type = plugins_1.PluginType.ServiceRouter;
+        this.name = "PolarisNearbyRouter";
+        this.requisite = plugins_1.RequisiteBitfield.None;
+        this.options = Object.assign(Object.assign({}, kDefaultOptions), options);
+    }
+    *query(callee, rules, caller, args) {
+        switch (this.options.mode) {
+            case MatchMode.Auto: {
+                /** 预查询，检查是否有节点不开启就近路由 */
+                yield {};
+                break;
+            }
+            case MatchMode.Off: {
+                yield {
+                    controller: {
+                        [plugins_1.RoutingCondition.Found]: {
+                            [plugins_1.RoutingAction.Bypass]: [this.name]
+                        },
+                        [plugins_1.RoutingCondition.NotFound]: {
+                            [plugins_1.RoutingAction.Bypass]: [this.name]
+                        }
+                    }
+                };
+                return;
+            }
+            case MatchMode.Always: {
+                break;
+            }
+            default: {
+                (0, utils_1.UNREACHABLE)();
+            }
+        }
+        // #region 查询特定范围实例
+        const { maxRange, minRange } = this.getRange(args);
+        for (let level = minRange; level >= maxRange; level -= 1) {
+            yield {
+                destination: {
+                    level
+                }
+            };
+        }
+        // #endregion
+    }
+    filter(instances, { destination }, args) {
+        /** MatchMode.Auto */
+        if (destination === undefined) {
+            /** 如备选实例中存在任意一个未开启就近路由的实例，则不开启就近路由 */
+            if (instances.some(instance => instance.metadata["internal-enable-nearby"] !== "true")) {
+                return {
+                    filtered: instances,
+                    action: {
+                        [plugins_1.RoutingAction.Bypass]: [this.name]
+                    }
+                };
+            }
+            /** 开启就近路由，进行实际查询 */
+            return {
+                filtered: []
+            };
+        }
+        if (destination.level !== this.getRange(args).maxRange) { /** 还存在其它可能的查询迭代 */
+            let unhealthyInstance = 0;
+            const totalInstances = instances.length;
+            for (let i = 0; i < totalInstances; i += 1) {
+                if (!(0, __1.isChoosableInstance)(instances[i])) {
+                    unhealthyInstance += 1;
+                }
+            }
+            if (unhealthyInstance / totalInstances >= this.options.unhealthyDegrade) {
+                return {
+                    filtered: []
+                };
+            }
+        }
+        return {
+            filtered: instances
+        };
+    }
+    getRange(args) {
+        const callArgs = args;
+        const { maxRange: presetMax, minRange: presetMin } = this.options;
+        let maxRange = presetMax;
+        let minRange = presetMin;
+        if (callArgs) {
+            const { maxRange: requiredMax, minRange: requiredMin } = callArgs;
+            if (requiredMax !== undefined) {
+                maxRange = requiredMax;
+            }
+            if (requiredMin !== undefined) {
+                minRange = requiredMin;
+            }
+        }
+        if (maxRange > minRange) {
+            throw new __1.ArgumentError(`|maxRange|(${maxRange}) must be larger than or equal to |minRange|(${minRange})`);
+        }
+        return { maxRange, minRange };
+    }
+}
+exports.PolarisNearbyRouter = PolarisNearbyRouter;
+//# sourceMappingURL=nearby.js.map

package/@tencent/polaris/dist/plugins/router/polaris/nearby.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"nearby.js","sourceRoot":"","sources":["../../../../src/plugins/router/polaris/nearby.ts"],"names":[],"mappings":";;;AAAA,gCAAmH;AACnH,8CAAyJ;AACzJ,0CAA6C;AAE7C,eAAe;AACf,IAAY,SAWX;AAXD,WAAY,SAAS;IACnB,cAAc;IACd,uCAAG,CAAA;IACH;;;;OAIG;IACH,yCAAI,CAAA;IACJ,eAAe;IACf,6CAAM,CAAA;AACR,CAAC,EAXW,SAAS,GAAT,iBAAS,KAAT,iBAAS,QAWpB;AAED,MAAM,eAAe,GAAG;IACtB;;OAEG;IACH,IAAI,EAAE,SAAS,CAAC,IAAI;IAEpB;;OAEG;IACH,QAAQ,gBAA2B;IAEnC;;OAEG;IACH,QAAQ,cAAyB;IAEjC;;;OAGG;IACH,gBAAgB,EAAE,CAAC;CACpB,CAAC;AAKF;;;GAGG;AACH,MAAa,mBAAmB;IAM9B,YAAY,OAA6C;QALhD,SAAI,GAAG,oBAAU,CAAC,aAAa,CAAC;QAChC,SAAI,GAAW,qBAAqB,CAAC;QACrC,cAAS,GAAG,2BAAiB,CAAC,IAAI,CAAC;QAI1C,IAAI,CAAC,OAAO,mCACP,eAAe,GACf,OAAO,CACX,CAAC;IACJ,CAAC;IAED,CAAE,KAAK,CACL,MAAe,EAAE,KAAsB,EAAE,MAAgB,EAAE,IAAqB;QAEhF,QAAQ,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE;YACzB,KAAK,SAAS,CAAC,IAAI,CAAC,CAAC;gBACnB,yBAAyB;gBACzB,MAAM,EAAE,CAAC;gBACT,MAAM;aACP;YACD,KAAK,SAAS,CAAC,GAAG,CAAC,CAAC;gBAClB,MAAM;oBACJ,UAAU,EAAE;wBACV,CAAC,0BAAgB,CAAC,KAAK,CAAC,EAAE;4BACxB,CAAC,uBAAa,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;yBACpC;wBACD,CAAC,0BAAgB,CAAC,QAAQ,CAAC,EAAE;4BAC3B,CAAC,uBAAa,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;yBACpC;qBACF;iBACF,CAAC;gBACF,OAAO;aACR;YACD,KAAK,SAAS,CAAC,MAAM,CAAC,CAAC;gBACrB,MAAM;aACP;YACD,OAAO,CAAC,CAAC;gBACP,IAAA,mBAAW,GAAE,CAAC;aACf;SACF;QAED,mBAAmB;QACnB,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACnD,KAAK,IAAI,KAAK,GAAG,QAAQ,EAAE,KAAK,IAAI,QAAQ,EAAE,KAAK,IAAI,CAAC,EAAE;YACxD,MAAM;gBACJ,WAAW,EAAE;oBACX,KAAK;iBACN;aACF,CAAC;SACH;QACD,aAAa;IACf,CAAC;IAED,MAAM,CACJ,SAAqB,EAAE,EAAE,WAAW,EAA6B,EAAE,IAAqB;QAExF,qBAAqB;QACrB,IAAI,WAAW,KAAK,SAAS,EAAE;YAC7B,sCAAsC;YACtC,IAAI,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,wBAAwB,CAAC,KAAK,MAAM,CAAC,EAAE;gBACtF,OAAO;oBACL,QAAQ,EAAE,SAAS;oBACnB,MAAM,EAAE;wBACN,CAAC,uBAAa,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;qBACpC;iBACF,CAAC;aACH;YAED,oBAAoB;YACpB,OAAO;gBACL,QAAQ,EAAE,EAAE;aACb,CAAC;SACH;QAED,IAAI,WAAW,CAAC,KAAK,KAAK,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,mBAAmB;YAC3E,IAAI,iBAAiB,GAAG,CAAC,CAAC;YAC1B,MAAM,cAAc,GAAG,SAAS,CAAC,MAAM,CAAC;YACxC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,cAAc,EAAE,CAAC,IAAI,CAAC,EAAE;gBAC1C,IAAI,CAAC,IAAA,uBAAmB,EAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE;oBACtC,iBAAiB,IAAI,CAAC,CAAC;iBACxB;aACF;YAED,IAAI,iBAAiB,GAAG,cAAc,IAAI,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE;gBACvE,OAAO;oBACL,QAAQ,EAAE,EAAE;iBACb,CAAC;aACH;SACF;QAED,OAAO;YACL,QAAQ,EAAE,SAAS;SACpB,CAAC;IACJ,CAAC;IAEO,QAAQ,CAA8B,IAAqB;QACjE,MAAM,QAAQ,GAAG,IAA8B,CAAC;QAChD,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QAClE,IAAI,QAAQ,GAAG,SAAS,CAAC;QACzB,IAAI,QAAQ,GAAG,SAAS,CAAC;QACzB,IAAI,QAAQ,EAAE;YACZ,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,QAAQ,CAAC;YAClE,IAAI,WAAW,KAAK,SAAS,EAAE;gBAC7B,QAAQ,GAAG,WAAW,CAAC;aACxB;YACD,IAAI,WAAW,KAAK,SAAS,EAAE;gBAC7B,QAAQ,GAAG,WAAW,CAAC;aACxB;SACF;QAED,IAAI,QAAQ,GAAG,QAAQ,EAAE;YACvB,MAAM,IAAI,iBAAa,CAAC,cAAc,QAAQ,gDAAgD,QAAQ,GAAG,CAAC,CAAC;SAC5G;QAED,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;IAChC,CAAC;CACF;AAtHD,kDAsHC"}

package/@tencent/polaris/dist/plugins/router/polaris/rule.d.ts

@@ -0,0 +1,11 @@
+import { DestinationQuery, PluginType, RequisiteBitfield, ServiceRouterPlugin } from "../../../plugins";
+import { RoutingRules } from "../../../rules";
+import { Service } from "../../../service";
+export declare class PolarisRuleRouter implements ServiceRouterPlugin {
+    readonly type = PluginType.ServiceRouter;
+    readonly name: string;
+    readonly requisite: RequisiteBitfield;
+    private variables?;
+    setVariables(variables: Record<string, string>): void;
+    query(callee: Service, rules?: RoutingRules[], caller?: Service): Generator<Partial<DestinationQuery>, void, void>;
+}

package/@tencent/polaris/dist/plugins/router/polaris/rule.js

@@ -0,0 +1,174 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PolarisRuleRouter = void 0;
+const metadata_1 = require("../../../metadata");
+const plugins_1 = require("../../../plugins");
+const utils_1 = require("../../../utils");
+const kMaxPriority = 9;
+const kEmptyObject = Object.freeze(Object.create(null));
+class PolarisRuleRouter {
+    constructor() {
+        this.type = plugins_1.PluginType.ServiceRouter;
+        this.name = "PolarisRuleRouter";
+        this.requisite = plugins_1.RequisiteBitfield.Rule;
+    }
+    setVariables(variables) {
+        this.variables = variables;
+    }
+    // eslint-disable-next-line max-lines-per-function
+    *query(callee, rules, caller) {
+        let service = "";
+        let metadata;
+        if (caller) {
+            ({ service, metadata } = caller);
+        }
+        if (!rules) {
+            (0, utils_1.UNREACHABLE)();
+        }
+        const len = rules.length;
+        /**
+         * 规则不存在，不走任何规则匹配，
+         * 直接匹配被调服务名，查询被调全量实例
+         */
+        if (len === 0) {
+            yield {
+                destination: {
+                    service: callee.service
+                }
+            };
+            return;
+        }
+        /*
+         * Step 1:
+         *  规则列表由前往后匹配，越靠前优先级越高
+         */
+        for (let i = 0; i < len; i += 1) {
+            let matches;
+            const rule = rules[i];
+            /*
+             * Step 2:
+             *  匹配 `sources` 源规则
+             *
+             * ```ts
+             * [].every(() => {}) // true
+             * [].some(() => {}) // false
+             * ```
+             */
+            if (!rule.sources.every((source) => {
+                /**
+                 * `source.service` 为 "*" 时，匹配所有源 `service`
+                 */
+                if (source.service !== "*" && source.service !== service) {
+                    return true;
+                }
+                if (source.metadata && metadata) {
+                    const result = (0, metadata_1.isMetadataMatch)(metadata, source.metadata, this.variables);
+                    if (result) {
+                        matches = result;
+                        return false;
+                    }
+                    return true;
+                }
+                if (source.metadata !== undefined || metadata !== undefined) {
+                    return true;
+                }
+                return false;
+            })) {
+                /*
+                 * Step 3:
+                 *  依据目标规则优先级 `priority`，由前往后匹配目标规则
+                 */
+                for (let priority = 0; priority <= kMaxPriority;) {
+                    const destinations = [];
+                    let steps = 0;
+                    rule.destinations.forEach((destination) => {
+                        const delta = destination.priority - priority;
+                        if (delta === 0) {
+                            destinations.push(destination);
+                        }
+                        else if (delta > 0) {
+                            /*
+                             * note:
+                             *  由于各目标规则优先级可能非连续，
+                             *  故计算最接近的下一个优先级的步长 `steps`，
+                             *  以减小循环次数，优化性能
+                             */
+                            if (steps === 0) {
+                                steps = delta;
+                            }
+                            else {
+                                steps = Math.min(delta, steps);
+                            }
+                        }
+                    });
+                    priority += steps;
+                    /*
+                     * Step 4:
+                     *  在同优先级的目标规则中，根据规则权重 `weight` 进行分配
+                     */
+                    if (destinations.length === 1) { /** fast case */
+                        yield {
+                            controller: {
+                                [plugins_1.RoutingCondition.Found]: {
+                                    [plugins_1.RoutingAction.Break]: true
+                                }
+                            },
+                            destination: destinations[0],
+                            source: {
+                                parameters: matches !== null && matches !== void 0 ? matches : kEmptyObject
+                            }
+                        };
+                    }
+                    else { /** slow case */
+                        let totalWeight = destinations.reduce((acc, cur) => acc + cur.weight, 0);
+                        /**
+                         * 如果目标规则权重和为 0，则不采用
+                         */
+                        while (totalWeight > 0) {
+                            /*
+                             * 采用随机轮训的方式，选取目标规则
+                             */
+                            const selectedValue = Math.random() * totalWeight;
+                            let beginValue = 0;
+                            let endValue = 0;
+                            for (let idx = 0; idx < destinations.length; idx += 1) {
+                                const destination = destinations[idx];
+                                endValue = beginValue + destination.weight;
+                                if (selectedValue >= beginValue && selectedValue < endValue) {
+                                    totalWeight -= destination.weight;
+                                    destinations.splice(idx, 1);
+                                    yield {
+                                        controller: {
+                                            [plugins_1.RoutingCondition.Found]: {
+                                                [plugins_1.RoutingAction.Break]: true
+                                            }
+                                        },
+                                        source: {
+                                            parameters: matches !== null && matches !== void 0 ? matches : kEmptyObject
+                                        },
+                                        destination
+                                    };
+                                    break;
+                                }
+                                beginValue = endValue;
+                            }
+                        }
+                    }
+                    /**
+                     * 所有优先级均已遍历完成
+                     */
+                    if (steps === 0) {
+                        break;
+                    }
+                }
+                /**
+                 * 一旦命中了一个特定的源规则，
+                 * 则不再匹配剩余规则
+                 */
+                break;
+            }
+        }
+    }
+}
+exports.PolarisRuleRouter = PolarisRuleRouter;
+//# sourceMappingURL=rule.js.map

package/@tencent/polaris/dist/plugins/router/polaris/rule.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rule.js","sourceRoot":"","sources":["../../../../src/plugins/router/polaris/rule.ts"],"names":[],"mappings":";;;AAAA,gDAA8D;AAC9D,8CAAyI;AAGzI,0CAA6C;AAE7C,MAAM,YAAY,GAAG,CAAC,CAAC;AACvB,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAA0B,CAAC;AAEjF,MAAa,iBAAiB;IAA9B;QACW,SAAI,GAAG,oBAAU,CAAC,aAAa,CAAC;QAChC,SAAI,GAAW,mBAAmB,CAAC;QACnC,cAAS,GAAG,2BAAiB,CAAC,IAAI,CAAC;IA2K9C,CAAC;IAvKC,YAAY,CAAC,SAAiC;QAC5C,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,kDAAkD;IAClD,CAAE,KAAK,CAAC,MAAe,EAAE,KAAsB,EAAE,MAAgB;QAC/D,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,QAA8B,CAAC;QACnC,IAAI,MAAM,EAAE;YACV,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC,CAAC;SAClC;QAED,IAAI,CAAC,KAAK,EAAE;YACV,IAAA,mBAAW,GAAE,CAAC;SACf;QAED,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC;QAEzB;;;WAGG;QACH,IAAI,GAAG,KAAK,CAAC,EAAE;YACb,MAAM;gBACJ,WAAW,EAAE;oBACX,OAAO,EAAE,MAAM,CAAC,OAAO;iBACxB;aACF,CAAC;YACF,OAAO;SACR;QAED;;;WAGG;QACH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE;YAC/B,IAAI,OAA2C,CAAC;YAChD,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACtB;;;;;;;;eAQG;YACH,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,MAAM,EAAE,EAAE;gBACjC;;mBAEG;gBACH,IAAI,MAAM,CAAC,OAAO,KAAK,GAAG,IAAI,MAAM,CAAC,OAAO,KAAK,OAAO,EAAE;oBACxD,OAAO,IAAI,CAAC;iBACb;gBAED,IAAI,MAAM,CAAC,QAAQ,IAAI,QAAQ,EAAE;oBAC/B,MAAM,MAAM,GAAG,IAAA,0BAAe,EAAC,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAU,CAAC,CAAC;oBAC3E,IAAI,MAAM,EAAE;wBACV,OAAO,GAAG,MAAM,CAAC;wBACjB,OAAO,KAAK,CAAC;qBACd;oBACD,OAAO,IAAI,CAAC;iBACb;gBAED,IAAI,MAAM,CAAC,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,SAAS,EAAE;oBAC3D,OAAO,IAAI,CAAC;iBACb;gBAED,OAAO,KAAK,CAAC;YACf,CAAC,CAAC,EAAE;gBACF;;;mBAGG;gBACH,KAAK,IAAI,QAAQ,GAAG,CAAC,EAAE,QAAQ,IAAI,YAAY,GAAG;oBAChD,MAAM,YAAY,GAAsB,EAAE,CAAC;oBAC3C,IAAI,KAAK,GAAG,CAAC,CAAC;oBAEd,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;wBACxC,MAAM,KAAK,GAAG,WAAW,CAAC,QAAQ,GAAG,QAAQ,CAAC;wBAC9C,IAAI,KAAK,KAAK,CAAC,EAAE;4BACf,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;yBAChC;6BAAM,IAAI,KAAK,GAAG,CAAC,EAAE;4BACpB;;;;;+BAKG;4BACH,IAAI,KAAK,KAAK,CAAC,EAAE;gCACf,KAAK,GAAG,KAAK,CAAC;6BACf;iCAAM;gCACL,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;6BAChC;yBACF;oBACH,CAAC,CAAC,CAAC;oBAEH,QAAQ,IAAI,KAAK,CAAC;oBAElB;;;uBAGG;oBACH,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,EAAE,gBAAgB;wBAC/C,MAAM;4BACJ,UAAU,EAAE;gCACV,CAAC,0BAAgB,CAAC,KAAK,CAAC,EAAE;oCACxB,CAAC,uBAAa,CAAC,KAAK,CAAC,EAAE,IAAI;iCAC5B;6BACF;4BACD,WAAW,EAAE,YAAY,CAAC,CAAC,CAAC;4BAC5B,MAAM,EAAE;gCACN,UAAU,EAAE,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,YAAY;6BACpC;yBACF,CAAC;qBACH;yBAAM,EAAE,gBAAgB;wBACvB,IAAI,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;wBACzE;;2BAEG;wBACH,OAAO,WAAW,GAAG,CAAC,EAAE;4BACtB;;+BAEG;4BACH,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,WAAW,CAAC;4BAClD,IAAI,UAAU,GAAG,CAAC,CAAC;4BACnB,IAAI,QAAQ,GAAG,CAAC,CAAC;4BACjB,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,YAAY,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE;gCACrD,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;gCACtC,QAAQ,GAAG,UAAU,GAAG,WAAW,CAAC,MAAM,CAAC;gCAC3C,IAAI,aAAa,IAAI,UAAU,IAAI,aAAa,GAAG,QAAQ,EAAE;oCAC3D,WAAW,IAAI,WAAW,CAAC,MAAM,CAAC;oCAClC,YAAY,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;oCAC5B,MAAM;wCACJ,UAAU,EAAE;4CACV,CAAC,0BAAgB,CAAC,KAAK,CAAC,EAAE;gDACxB,CAAC,uBAAa,CAAC,KAAK,CAAC,EAAE,IAAI;6CAC5B;yCACF;wCACD,MAAM,EAAE;4CACN,UAAU,EAAE,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,YAAY;yCACpC;wCACD,WAAW;qCACZ,CAAC;oCACF,MAAM;iCACP;gCACD,UAAU,GAAG,QAAQ,CAAC;6BACvB;yBACF;qBACF;oBAED;;uBAEG;oBACH,IAAI,KAAK,KAAK,CAAC,EAAE;wBACf,MAAM;qBACP;iBACF;gBAED;;;mBAGG;gBACH,MAAM;aACP;SACF;IACH,CAAC;CACF;AA9KD,8CA8KC"}

package/@tencent/polaris/dist/plugins/router/trpc/canary.d.ts

@@ -0,0 +1,25 @@
+import { PluginType, RequisiteBitfield, ServiceRouterPlugin } from "../../../plugins";
+import { RoutingRules } from "../../../rules";
+import { Service } from "../../../service";
+export declare type CanaryArgs = string;
+export declare class TRPCCanaryRouter implements ServiceRouterPlugin {
+    readonly type = PluginType.ServiceRouter;
+    readonly name: string;
+    readonly requisite: RequisiteBitfield;
+    query<AdditionalArgs = CanaryArgs>(callee: Service, rules?: RoutingRules[], caller?: Service, args?: AdditionalArgs): Generator<{
+        controller?: undefined;
+        destination?: undefined;
+    } | {
+        controller: {
+            0: {
+                0: boolean;
+            };
+        };
+        destination: {
+            service: string;
+            metadata: {
+                canary: string;
+            };
+        };
+    }, void, unknown>;
+}

package/@tencent/polaris/dist/plugins/router/trpc/canary.js

@@ -0,0 +1,36 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TRPCCanaryRouter = void 0;
+const plugins_1 = require("../../../plugins");
+class TRPCCanaryRouter {
+    constructor() {
+        this.type = plugins_1.PluginType.ServiceRouter;
+        this.name = "tRPCCanaryRouter";
+        this.requisite = plugins_1.RequisiteBitfield.None;
+    }
+    *query(callee, rules, caller, args) {
+        // 没有传金丝雀的标签时，不作过滤
+        if (typeof args === "undefined") {
+            yield {};
+            return;
+        }
+        // 查找金丝雀标签匹配的实例
+        yield {
+            controller: {
+                [plugins_1.RoutingCondition.Found]: {
+                    [plugins_1.RoutingAction.Break]: true
+                }
+            },
+            destination: {
+                service: "*",
+                metadata: {
+                    canary: String(args)
+                }
+            }
+        };
+        // 金丝雀标签未匹配，则不作过滤
+        yield {};
+    }
+}
+exports.TRPCCanaryRouter = TRPCCanaryRouter;
+//# sourceMappingURL=canary.js.map

package/@tencent/polaris/dist/plugins/router/trpc/canary.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"canary.js","sourceRoot":"","sources":["../../../../src/plugins/router/trpc/canary.ts"],"names":[],"mappings":";;;AAAA,8CAAuH;AAMvH,MAAa,gBAAgB;IAA7B;QACW,SAAI,GAAG,oBAAU,CAAC,aAAa,CAAC;QAChC,SAAI,GAAW,kBAAkB,CAAC;QAClC,cAAS,GAAG,2BAAiB,CAAC,IAAI,CAAC;IA6B9C,CAAC;IA3BC,CAAE,KAAK,CACL,MAAe,EAAE,KAAsB,EAAE,MAAgB,EAAE,IAAqB;QAEhF,kBAAkB;QAClB,IAAI,OAAO,IAAI,KAAK,WAAW,EAAE;YAC/B,MAAM,EAAE,CAAC;YACT,OAAO;SACR;QAED,eAAe;QACf,MAAM;YACJ,UAAU,EAAE;gBACV,CAAC,0BAAgB,CAAC,KAAK,CAAC,EAAE;oBACxB,CAAC,uBAAa,CAAC,KAAK,CAAC,EAAE,IAAI;iBAC5B;aACF;YACD,WAAW,EAAE;gBACX,OAAO,EAAE,GAAG;gBACZ,QAAQ,EAAE;oBACR,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC;iBACrB;aACF;SACF,CAAC;QAEF,iBAAiB;QACjB,MAAM,EAAE,CAAC;IACX,CAAC;CACF;AAhCD,4CAgCC"}

package/@tencent/polaris/dist/plugins/router/trpc/env.d.ts

@@ -0,0 +1,10 @@
+import { DestinationQuery, PluginType, RequisiteBitfield, ServiceRouterPlugin } from "../../../plugins";
+import { RoutingRules } from "../../../rules";
+import { Service } from "../../../service";
+export declare type EnvArgs = string[];
+export declare class TRPCEnvRouter implements ServiceRouterPlugin {
+    readonly type = PluginType.ServiceRouter;
+    readonly name: string;
+    readonly requisite: RequisiteBitfield;
+    query<AdditionalArgs = EnvArgs>(callee: Service, rules?: RoutingRules[], caller?: Service, args?: AdditionalArgs): Generator<Partial<DestinationQuery>, void, void>;
+}

package/@tencent/polaris/dist/plugins/router/trpc/env.js

@@ -0,0 +1,34 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TRPCEnvRouter = void 0;
+const __1 = require("../../..");
+const plugins_1 = require("../../../plugins");
+class TRPCEnvRouter {
+    constructor() {
+        this.type = plugins_1.PluginType.ServiceRouter;
+        this.name = "tRPCEnvRouter";
+        this.requisite = plugins_1.RequisiteBitfield.None;
+    }
+    *query(callee, rules, caller, args) {
+        if (!Array.isArray(args)) {
+            throw new __1.ArgumentError("|args| must be an array");
+        }
+        for (let i = 0; i < args.length; i += 1) {
+            yield {
+                controller: {
+                    [plugins_1.RoutingCondition.Found]: {
+                        [plugins_1.RoutingAction.Break]: true
+                    }
+                },
+                destination: {
+                    service: "*",
+                    metadata: {
+                        env: args[i].toString()
+                    }
+                }
+            };
+        }
+    }
+}
+exports.TRPCEnvRouter = TRPCEnvRouter;
+//# sourceMappingURL=env.js.map

package/@tencent/polaris/dist/plugins/router/trpc/env.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"env.js","sourceRoot":"","sources":["../../../../src/plugins/router/trpc/env.ts"],"names":[],"mappings":";;;AAAA,gCAAyC;AACzC,8CAAyI;AAMzI,MAAa,aAAa;IAA1B;QACW,SAAI,GAAG,oBAAU,CAAC,aAAa,CAAC;QAChC,SAAI,GAAW,eAAe,CAAC;QAC/B,cAAS,GAAG,2BAAiB,CAAC,IAAI,CAAC;IAyB9C,CAAC;IAvBC,CAAE,KAAK,CACL,MAAe,EAAE,KAAsB,EAAE,MAAgB,EAAE,IAAqB;QAEhF,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;YACxB,MAAM,IAAI,iBAAa,CAAC,yBAAyB,CAAC,CAAC;SACpD;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YACvC,MAAM;gBACJ,UAAU,EAAE;oBACV,CAAC,0BAAgB,CAAC,KAAK,CAAC,EAAE;wBACxB,CAAC,uBAAa,CAAC,KAAK,CAAC,EAAE,IAAI;qBAC5B;iBACF;gBACD,WAAW,EAAE;oBACX,OAAO,EAAE,GAAG;oBACZ,QAAQ,EAAE;wBACR,GAAG,EAAG,IAAgB,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE;qBACrC;iBACF;aACF,CAAC;SACH;IACH,CAAC;CACF;AA5BD,sCA4BC"}

package/@tencent/polaris/dist/plugins/router/trpc/set.d.ts

@@ -0,0 +1,56 @@
+import { Instance } from "../../../instance";
+import { DestinationQuery, FilteredResult, PluginType, RequisiteBitfield, ServiceRouterPlugin } from "../../../plugins";
+import { RoutingRules } from "../../../rules";
+import { Service } from "../../../service";
+/** 匹配类型 */
+export declare enum MatchType {
+    /**
+     * @description
+     * 旁路此模块，不做 `Set` 路由
+     */
+    NoSet = 0,
+    /**
+     * @description
+     * 匹配流程如下：
+     *  1. 精确匹配，如存在实例（不考虑实例状态）则返回，并旁路就近路由
+     *  2. 匹配其通配组，Set 组名为 "*"，如存在实例（不考虑实例状态）则返回，并旁路就近路由
+     *  3. 如被调未启用 Set（被调不存在可以匹配上 `Set 名` 的实例）则旁路此模块
+     */
+    SourceSet = 1,
+    /**
+     * @description
+     * 匹配流程如下：
+     *  1. 精确匹配，如存在实例（不考虑实例状态）则返回，并旁路就近路由
+     *  2. 匹配其通配组，Set 组名为 "*"，如存在实例（不考虑实例状态）则返回，并旁路就近路由
+     */
+    DestinationSet = 2
+}
+export interface SetInfo {
+    /** Set 名 */
+    name: string;
+    /** Set 地区 */
+    area: string;
+    /** Set 组名 */
+    group: string;
+}
+export interface SetArgs {
+    /**
+     * 匹配类型
+     */
+    match: MatchType;
+    /**
+     * 需匹配的 Set 信息
+     */
+    set: SetInfo;
+    /**
+     * 开启按 Set 就近访问
+     */
+    enableNearby?: boolean;
+}
+export declare class TRPCSetRouter implements ServiceRouterPlugin {
+    readonly type = PluginType.ServiceRouter;
+    readonly name: string;
+    readonly requisite: RequisiteBitfield;
+    query<AdditionalArgs = SetArgs>(callee: Service, rules?: RoutingRules[], caller?: Service, args?: AdditionalArgs): Generator<Partial<DestinationQuery>, void, void>;
+    filter?<AdditionalArgs>(instances: Instance[], query: Partial<DestinationQuery>, args?: AdditionalArgs): FilteredResult;
+}

package/@tencent/polaris/dist/plugins/router/trpc/set.js

@@ -0,0 +1,106 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TRPCSetRouter = exports.MatchType = void 0;
+const plugins_1 = require("../../../plugins");
+/** 匹配类型 */
+var MatchType;
+(function (MatchType) {
+    /**
+     * @description
+     * 旁路此模块，不做 `Set` 路由
+     */
+    MatchType[MatchType["NoSet"] = 0] = "NoSet";
+    /**
+     * @description
+     * 匹配流程如下：
+     *  1. 精确匹配，如存在实例（不考虑实例状态）则返回，并旁路就近路由
+     *  2. 匹配其通配组，Set 组名为 "*"，如存在实例（不考虑实例状态）则返回，并旁路就近路由
+     *  3. 如被调未启用 Set（被调不存在可以匹配上 `Set 名` 的实例）则旁路此模块
+     */
+    MatchType[MatchType["SourceSet"] = 1] = "SourceSet";
+    /**
+     * @description
+     * 匹配流程如下：
+     *  1. 精确匹配，如存在实例（不考虑实例状态）则返回，并旁路就近路由
+     *  2. 匹配其通配组，Set 组名为 "*"，如存在实例（不考虑实例状态）则返回，并旁路就近路由
+     */
+    MatchType[MatchType["DestinationSet"] = 2] = "DestinationSet";
+})(MatchType = exports.MatchType || (exports.MatchType = {}));
+/** 预置的旁路行为 */
+const presetBypass = {
+    Set: Object.freeze({
+        [plugins_1.RoutingAction.Bypass]: ["tRPCSetRouter"]
+    }),
+    Nearby: Object.freeze({
+        [plugins_1.RoutingAction.Bypass]: ["PolarisNearbyRouter"]
+    })
+};
+class TRPCSetRouter {
+    constructor() {
+        this.type = plugins_1.PluginType.ServiceRouter;
+        this.name = "tRPCSetRouter";
+        this.requisite = plugins_1.RequisiteBitfield.None;
+    }
+    *query(callee, rules, caller, args) {
+        const callArgs = args;
+        if (callArgs === null || callArgs === void 0 ? void 0 : callArgs.match) { /** MatchType.NoSet === 0 */
+            yield {
+                destination: {
+                    metadata: {
+                        "internal-enable-set": "Y"
+                    }
+                }
+            };
+        }
+        else {
+            yield {};
+        }
+    }
+    filter(instances, query, args) {
+        const callArgs = args;
+        if (!callArgs || callArgs.match === MatchType.NoSet) {
+            return {
+                filtered: instances,
+                action: presetBypass.Set
+            };
+        }
+        let enabledSet = false;
+        const matchedInstances = [];
+        const genericInstances = [];
+        const { name, area, group } = callArgs.set;
+        const exactArea = `${name}.${area}`;
+        const exactGroup = `${name}.${area}.${group}`;
+        const genericGroup = `${name}.${area}.*`;
+        instances.forEach((instance) => {
+            const instanceSet = instance.metadata["internal-set-name"];
+            if (instanceSet.startsWith(name)) {
+                /** 仅当 Set 名匹配时，才认为开启 Set 调用 */
+                enabledSet = true;
+                if ((group === "*" && instanceSet.startsWith(exactArea)) /** group 为 "*" 时匹配所有组 */
+                    || instanceSet === exactGroup /** 精确匹配 */) {
+                    matchedInstances.push(instance);
+                }
+                else if (instanceSet === genericGroup) { /** 通配组 */
+                    genericInstances.push(instance);
+                }
+            }
+        });
+        const result = {
+            filtered: matchedInstances.length > 0 ? matchedInstances : genericInstances
+        };
+        /**
+         * 默认不按 set 开启就近访问，找到可用节点，就会忽略 NearBy 插件
+         * enableNearby 参数可以开启按 Set 就近访问
+         */
+        if (result.filtered.length > 0 && !callArgs.enableNearby) {
+            result.action = presetBypass.Nearby;
+        }
+        else if (!enabledSet && callArgs.match === MatchType.SourceSet) {
+            result.action = presetBypass.Set;
+            result.filtered = instances;
+        }
+        return result;
+    }
+}
+exports.TRPCSetRouter = TRPCSetRouter;
+//# sourceMappingURL=set.js.map

package/@tencent/polaris/dist/plugins/router/trpc/set.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"set.js","sourceRoot":"","sources":["../../../../src/plugins/router/trpc/set.ts"],"names":[],"mappings":";;;AACA,8CAAuI;AAIvI,WAAW;AACX,IAAY,SAuBX;AAvBD,WAAY,SAAS;IACnB;;;OAGG;IACH,2CAAS,CAAA;IAET;;;;;;OAMG;IACH,mDAAS,CAAA;IAET;;;;;OAKG;IACH,6DAAc,CAAA;AAChB,CAAC,EAvBW,SAAS,GAAT,iBAAS,KAAT,iBAAS,QAuBpB;AA8BD,cAAc;AACd,MAAM,YAAY,GAAG;IACnB,GAAG,EAAE,MAAM,CAAC,MAAM,CAAC;QACjB,CAAC,uBAAa,CAAC,MAAM,CAAC,EAAE,CAAC,eAAe,CAAC;KAC1C,CAAC;IACF,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC;QACpB,CAAC,uBAAa,CAAC,MAAM,CAAC,EAAE,CAAC,qBAAqB,CAAC;KAChD,CAAC;CACH,CAAC;AAEF,MAAa,aAAa;IAA1B;QACW,SAAI,GAAG,oBAAU,CAAC,aAAa,CAAC;QAChC,SAAI,GAAW,eAAe,CAAC;QAC/B,cAAS,GAAG,2BAAiB,CAAC,IAAI,CAAC;IAqE9C,CAAC;IAnEC,CAAE,KAAK,CACL,MAAe,EAAE,KAAsB,EAAE,MAAgB,EAAE,IAAqB;QAEhF,MAAM,QAAQ,GAAG,IAA2B,CAAC;QAC7C,IAAI,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,KAAK,EAAE,EAAE,4BAA4B;YACjD,MAAM;gBACJ,WAAW,EAAE;oBACX,QAAQ,EAAE;wBACR,qBAAqB,EAAE,GAAG;qBAC3B;iBACF;aACF,CAAC;SACH;aAAM;YACL,MAAM,EAAE,CAAC;SACV;IACH,CAAC;IAED,MAAM,CACJ,SAAqB,EAAE,KAAgC,EAAE,IAAqB;QAE9E,MAAM,QAAQ,GAAG,IAA2B,CAAC;QAE7C,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS,CAAC,KAAK,EAAE;YACnD,OAAO;gBACL,QAAQ,EAAE,SAAS;gBACnB,MAAM,EAAE,YAAY,CAAC,GAAG;aACzB,CAAC;SACH;QAED,IAAI,UAAU,GAAG,KAAgB,CAAC;QAClC,MAAM,gBAAgB,GAAe,EAAE,CAAC;QACxC,MAAM,gBAAgB,GAAe,EAAE,CAAC;QACxC,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,QAAQ,CAAC,GAAG,CAAC;QAC3C,MAAM,SAAS,GAAG,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC;QACpC,MAAM,UAAU,GAAG,GAAG,IAAI,IAAI,IAAI,IAAI,KAAK,EAAE,CAAC;QAC9C,MAAM,YAAY,GAAG,GAAG,IAAI,IAAI,IAAI,IAAI,CAAC;QAEzC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC7B,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC;YAC3D,IAAI,WAAW,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE;gBAChC,+BAA+B;gBAC/B,UAAU,GAAG,IAAI,CAAC;gBAClB,IACE,CAAC,KAAK,KAAK,GAAG,IAAI,WAAW,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,yBAAyB;uBAC3E,WAAW,KAAK,UAAU,CAAC,WAAW,EACzC;oBACA,gBAAgB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;iBACjC;qBAAM,IAAI,WAAW,KAAK,YAAY,EAAE,EAAE,UAAU;oBACnD,gBAAgB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;iBACjC;aACF;QACH,CAAC,CAAC,CAAC;QACH,MAAM,MAAM,GAAmB;YAC7B,QAAQ,EAAE,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,gBAAgB;SAC5E,CAAC;QACF;;;WAGG;QACH,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE;YACxD,MAAM,CAAC,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;SACrC;aAAM,IAAI,CAAC,UAAU,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS,CAAC,SAAS,EAAE;YAChE,MAAM,CAAC,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC;YACjC,MAAM,CAAC,QAAQ,GAAG,SAAS,CAAC;SAC7B;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAxED,sCAwEC"}

package/@tencent/polaris/dist/plugins/shaping/unirate.d.ts

@@ -0,0 +1,31 @@
+import { PluginType, TrafficShapingPlugin } from "../../plugins";
+import { RatelimitRule } from "../../rules";
+declare const kDefaultOptions: {
+    /**
+     * 最长等待时间 (ms)
+     */
+    maxWaitingTime: number;
+    /**
+     * 空闲回收周期
+     */
+    idlePeriod: number;
+};
+export declare type UnirateTrafficShapingOptions = typeof kDefaultOptions;
+/**
+ * Leaky bucket
+ * (https://en.wikipedia.org/wiki/Leaky_bucket)
+ */
+export declare class UnirateTrafficShaping implements TrafficShapingPlugin {
+    readonly type = PluginType.TrafficShaping;
+    readonly name: string;
+    private readonly options;
+    private readonly buckets;
+    private disposed;
+    constructor(options?: Partial<UnirateTrafficShapingOptions>);
+    dispose(): void;
+    get isDisposed(): boolean;
+    inFlow(rule: RatelimitRule, partition: number): Promise<void>;
+    private allocate;
+    private buildBucket;
+}
+export {};