@tencent-rtc/mcp 1.0.0

package/@tencent/polaris/dist/plugins/naming/polaris-server/adapter/monitor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"monitor.js","sourceRoot":"","sources":["../../../../../src/plugins/naming/polaris-server/adapter/monitor.ts"],"names":[],"mappings":";;;AAAA,+BAAoC;AAGpC,iDAA2E;AAC3E,6CAA4F;AAE5F,oCAAyD;AACzD,oCAA8D;AAC9D,iCAAgE;AAehE,MAAM,WAAW,GAAG,gBAAgB,CAAC;AAErC,MAAsB,qBAAsB,SAAQ,yBAAkB;IAMpE,KAAK,CAAC,iBAAiB,CAAC,SAAiB,EAAE,OAAe,EAAE,IAAkB;QAC5E,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE;YACtC,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;SACjC;QACD,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE;YAC5B,IAAA,mBAAW,GAAE,CAAC;SACf;QAED,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACxB,MAAM,EAAE,WAAW,EAAE,eAAe,EAAE,GAAG,MAAM,CAAC;QAChD,MAAM,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QACxC,MAAM,QAAQ,GAAuB,EAAE,CAAC;QACxC,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,GAAG,gBAAQ,CAAC;QAChD,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE;YAC9C,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE;gBACtD,IAAI,KAAK,GAAG,CAAC,EAAE;oBACb,yDAAyD;oBACzD,MAAM,sBAAsB,GAAG,CAAC,CAAC,CAAC,IAAI,GAAG,YAAY,CAAC,CAAC;oBACvD,MAAM,wBAAwB,GAAG,CAAC,CAAC,CAAC,KAAK,GAAG,YAAY,CAAC,CAAC;oBAC1D,wDAAwD;oBACxD,MAAM,EAAE,WAAW,EAAE,cAAc,EAAE,GAAG,MAAM,CAAC;oBAC/C,IAAI,MAAM,CAAC,cAAc,EAAE;wBACzB,MAAM,OAAO,GAAG;4BACd,GAAG,EAAE;gCACH,SAAS;gCACT,OAAO;gCACP,OAAO;gCACP,IAAI;gCACJ,aAAa,EAAE,GAAG,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI,EAAE;gCAClD,WAAW,EAAE,IAAI,CAAC,KAAM,CAAC,IAAI;6BAC9B;4BACD,KAAK,EAAE;gCACL,sBAAsB;gCACtB,wBAAwB;6BACzB;yBACF,CAAC;wBACF,MAAM,CAAC,KAAK,0BACa,IAAI,CAAC,IAAI,EAChC,mBAAmB,EAAE,cAAc,EAAE,mBAAmB,EAAE,OAAO,CAClE,CAAC;qBACH;oBACD,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,mBAAW,CAAC,OAAO,EAAE,cAAc,EAAE,0BAA0B,EAAE;wBACjG,EAAE,EAAE,IAAI,CAAC,GAAG,CAAC,IAAA,SAAM,GAAE,CAAC;wBACtB,GAAG,EAAE;4BACH,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC;4BAC9B,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;4BAC1B,aAAa,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;4BAC5D,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAM,CAAC,IAAI,CAAC;4BACvC,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;4BAC1B,eAAe,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;yBAChC;wBACD;;2BAEG;wBACH,KAAK,EAAE;4BACL,sBAAsB,EAAE,IAAI,CAAC,GAAG,CAAC,sBAAsB,CAAC;4BACxD,wBAAwB,EAAE,IAAI,CAAC,GAAG,CAAC,wBAAwB,CAAC;yBAC7D;qBACF,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;wBACnB,IAAI,CAAC,eAAe,CAAC,cAAc,EAAE,mBAAmB,EAAE,QAAQ,CAAC,CAAC;wBACpE,OAAO,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;oBAC1C,CAAC,CAAC,CAAC,CAAC;iBACL;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,CAAC,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,mBAAmB;QAC9F,MAAM,CAAC,KAAK,0BAAwB,IAAI,CAAC,IAAI,EAAE,mBAAmB,EAAE,eAAe,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;QAC7G,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,SAAiB,EAAE,OAAe,EAAE,IAAe;QACrE,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE;YACtC,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;SACjC;QAED,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACxB,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAC/B,IAAI,MAAM,CAAC,cAAc,EAAE;YACzB,MAAM,CAAC,KAAK,0BACa,IAAI,CAAC,IAAI,EAChC,eAAe,EAAE,WAAW,EAAE,UAAU,SAAS,IAAI,OAAO,aAAa,EAAE,IAAI,CAChF,CAAC;SACH;QACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,mBAAW,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,iBAAiB,EAAE;YAC9G,EAAE,EAAE,IAAA,SAAM,GAAE;YACZ,SAAS;YACT,OAAO;YACP,SAAS,EAAE,IAAI,CAAC,QAAQ;YACxB,mBAAmB,EAAE,IAAI,CAAC,0BAAgB,CAAC,QAAQ,CAAC,CAAC,UAAU;YAC/D,iBAAiB,EAAE,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,0BAAgB,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,mBAAmB,CAAC;YAC3G,kBAAkB,EAAE,IAAI,CAAC,0BAAgB,CAAC,IAAI,CAAC,CAAC,UAAU;YAC1D,eAAe,EAAE,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,0BAAgB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,iBAAiB,CAAC;SACpG,CAAC,CAAC;QACH,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,eAAe,EAAE,QAAQ,CAAC,CAAC;QAE7D,OAAO,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAAe,EAAE,KAAmB;QACtD,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE;YACtC,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;SACjC;QACD,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE;YAC5B,IAAA,mBAAW,GAAE,CAAC;SACf;QAED,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACxB,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAC/B,IAAI,MAAM,CAAC,cAAc,EAAE;YACzB,MAAM,CAAC,KAAK,0BACa,IAAI,CAAC,IAAI,EAChC,eAAe,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,EAAE,UAAU,KAAK,oBAAoB,CACrF,CAAC;SACH;QACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,mBAAW,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,yBAAyB,EAAE;YACtH,EAAE,EAAE,IAAI,CAAC,GAAG,CAAC,IAAA,SAAM,GAAE,CAAC;YACtB,GAAG,EAAE;gBACH,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;gBACtC,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC;gBAClC,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,sBAAc,CAAC;gBACxC,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC;gBAChC,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC;gBAC5B,MAAM,EAAE,GAAG,CAAC,MAAiD;gBAC7D,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC;gBAC1B,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC;aACpC;YACD,KAAK,EAAE;gBACL,wBAAwB,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC;aAC1C;SACF,CAAC,CAAC;QACH,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,eAAe,EAAE,QAAQ,CAAC,CAAC;QAE7D,OAAO,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAe,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;QAClD,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE;YACtC,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;SACjC;QAED,MAAM,EAAE,cAAc,EAAE,eAAe,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC;QACzD,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACxB,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAC/B,MAAM,CAAC,KAAK,0BAAwB,IAAI,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,qBAAqB,EAAE,MAAM,CAAC,CAAC;QAC3G,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,mBAAW,CAAC,OAAO,EAAE,cAAc,EAAE,yBAAyB,EAAE;YACzG,KAAK,EAAE,IAAI,CAAC,QAAQ;YACpB,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;YAC9B,gBAAgB,EAAE;gBAChB,OAAO,EAAE,GAAG,GAAG,gBAAQ;gBACvB,KAAK,EAAE,CAAC,GAAG,GAAG,gBAAQ,CAAC,GAAG,gBAAQ,GAAG,cAAM;aAC5C;YACD,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC;YACvC,OAAO,EAAE,sBAAc;YACvB,MAAM,EAAE,WAAW;SACpB,CAAC,CAAC;QACH,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,cAAc,EAAE,QAAQ,CAAC,CAAC;QAE5D;;;WAGG;QACH,IAAI,IAAI,CAAC,KAAK,EAAE;YACd,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;SAC1B;QACD,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC3B,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;YACvB,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,IAAI,kBAAkB,EAAE,GAAG,CAAC,CAAC,CAAC;QACnH,CAAC,EAAE,eAAe,CAAC,CAAC,KAAK,EAAE,CAAC;QAE5B,IAAI,CAAC,YAAY,CAAC,OAAO,GAAG,GAAG,EAAE;YAC/B,IAAI,IAAI,CAAC,KAAK,EAAE;gBACd,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACzB,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;aACxB;QACH,CAAC,CAAC;QAEF,OAAO,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,SAAiB,EAAE,OAAe,EAAE,IAAqB;QAC7E,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE;YACtC,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;SACjC;QAED,MAAM,YAAY,GAAiE,EAAE,CAAC;QACtF,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE;YACrD,YAAY,CAAC,IAAI,CAAC;gBAChB,IAAI;gBACJ,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;oBACzD,IAAI,EAAE,IAAA,mBAAW,EAAC,IAAI,CAAC;oBACvB,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC;oBAC7C,MAAM,EAAE,IAAA,6BAAqB,EAAC,MAAM,EAAE,KAAK,CAAC;oBAC5C,MAAM;iBACP,CAAC,CAAC;aACJ,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACxB,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAC/B,IAAI,MAAM,CAAC,cAAc,EAAE;YACzB,MAAM,CAAC,KAAK,0BAAwB,IAAI,CAAC,IAAI,EAAE,iBAAiB,EAAE,WAAW,EAAE,mBAAmB,EAAE;gBAClG,SAAS;gBACT,OAAO;gBACP,qBAAqB,EAAE,YAAY;gBACnC,WAAW,EAAE,IAAI,CAAC,OAAO;aAC1B,CAAC,CAAC;SACJ;QACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,mBAAW,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,qBAAqB,EAAE;YAClH,EAAE,EAAE,IAAA,SAAM,GAAE;YACZ,SAAS;YACT,OAAO;YACP,SAAS,EAAE,IAAI,CAAC,QAAQ;YACxB,qBAAqB,EAAE,YAAY;YACnC,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;gBACzD,IAAI,EAAE,IAAA,mBAAW,EAAC,IAAI,CAAC;gBACvB,MAAM,EAAE,wBAAgB,CAAC,GAAG;gBAC5B,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC;aAC5C,CAAC,CAAC;SACJ,CAAC,CAAC;QACH,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,iBAAiB,EAAE,QAAQ,CAAC,CAAC;QAE/D,OAAO,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;IAED,IAAY,QAAQ;QAClB,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE;YAC5B,OAAO,IAAI,CAAC,KAAK,CAAC;SACnB;QAED,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE;YAC5B,IAAA,mBAAW,GAAE,CAAC;SACf;QAED,MAAM,KAAK,GAAG;YACZ,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;YACnB,GAAG,EAAE,OAAO,CAAC,GAAG;YAChB,GAAG,EAAE,IAAA,SAAM,GAAE;SACd,CAAC;QACF,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,MAAM,CAAC,GAAW;QACxB,IAAI,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QACnB,IAAI,GAAG,KAAK,SAAS,EAAE;YACrB,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAoC,CAAC;YAC7D,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;SAChB;QACD,IAAI,GAAG,CAAC,GAAG,CAAC,KAAK,SAAS,EAAE;YAC1B,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,iBAAiB;SAChC;QACD,OAAO,GAAG,CAAC,GAAG,CAAE,EAAE,CAAC,CAAC,kCAAkC;IACxD,CAAC;IAEO,qBAAqB,CAC3B,SAAwB,EAAE,GAAW;QAErC,OAAO;YACL,QAAQ,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC/C,IAAI,EAAE,IAAA,mBAAW,EAAC,IAAI,CAAC;gBACvB,QAAQ;gBACR,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;aAC7B,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;CACF;AAlRD,sDAkRC"}

package/@tencent/polaris/dist/plugins/naming/polaris-server/adapter/ratelimit.d.ts

@@ -0,0 +1,19 @@
+import { QuotaConfig, RatelimitRule } from "../../../..";
+import { DataWithRevision, RatelimitServicePlugin } from "../../../../plugins";
+import { BaseAdapterOptions, PolarisBaseAdapter } from "./base";
+export interface RatelimitAdapterOptions extends BaseAdapterOptions {
+    /**
+     * 流量控制后端服务名
+     * __请勿随意修改此配置__
+     */
+    ratelimitService: string;
+}
+export declare abstract class PolarisRatelimitAdapter extends PolarisBaseAdapter implements RatelimitServicePlugin {
+    protected readonly options: PolarisBaseAdapter["options"] & RatelimitAdapterOptions;
+    private ratelimitRuleProcessor?;
+    private ratelimitInfo?;
+    ratelimitRules(namespace: string, service: string, revision?: string): Promise<DataWithRevision<RatelimitRule[]>>;
+    acquireQuota(namespace: string, service: string, rule: RatelimitRule, used: QuotaConfig[]): Promise<QuotaConfig[]>;
+    private procRatelimitRule;
+    private amountToLimiter;
+}

package/@tencent/polaris/dist/plugins/naming/polaris-server/adapter/ratelimit.js

@@ -0,0 +1,222 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PolarisRatelimitAdapter = void 0;
+const process_1 = require("process");
+const __1 = require("../../../..");
+const utils_1 = require("../../../../utils");
+const rules_1 = require("../rules");
+const types_1 = require("../types");
+const utils_2 = require("../utils");
+const base_1 = require("./base");
+class PolarisRatelimitAdapter extends base_1.PolarisBaseAdapter {
+    async ratelimitRules(namespace, service, revision) {
+        if (!this.initializeStatus.initialized) {
+            await this.waitForInitialized();
+        }
+        const { logger } = this;
+        const { transaction } = logger;
+        if (logger.tracingEnabled) {
+            logger.trace("Plugins" /* Plugins */, this.name, "ratelimitRules", transaction, `request ${namespace}.${service}${(revision !== null && revision !== void 0 ? revision : "") && `@${revision}`}`);
+        }
+        const response = await this.requestBackend(types_1.ServiceType.Discover, this.options.discoverService, "discover", {
+            type: types_1.DiscoverRequestType.RATE_LIMIT,
+            service: {
+                namespace: this.box(namespace),
+                name: this.box(service)
+            }
+        });
+        this.tracingResponse(transaction, "ratelimitRules", response);
+        this.maybeErrorResponse(response, namespace, service);
+        const { rateLimit } = response;
+        if (!(rateLimit === null || rateLimit === void 0 ? void 0 : rateLimit.rules)
+            || (revision !== undefined && this.unbox(rateLimit.revision, "") === revision)) {
+            logger.trace("Plugins" /* Plugins */, this.name, "ratelimitRules", transaction, "empty rules or revision is equal");
+            return {
+                data: [],
+                revision: (rateLimit === null || rateLimit === void 0 ? void 0 : rateLimit.rules) ? (revision !== null && revision !== void 0 ? revision : "") : ""
+            };
+        }
+        let { ratelimitRuleProcessor } = this;
+        if (ratelimitRuleProcessor === undefined) {
+            ratelimitRuleProcessor = new rules_1.RuleProcessor(this.unbox.bind(this));
+            this.ratelimitRuleProcessor = ratelimitRuleProcessor;
+        }
+        const { rules } = rateLimit;
+        const data = [];
+        for (let i = 0; i < rules.length; i += 1) { // eslint-disable-line @typescript-eslint/prefer-for-of
+            const rule = this.procRatelimitRule(ratelimitRuleProcessor, rules[i]);
+            if (rule) {
+                data.push(rule);
+            }
+        }
+        logger.trace("Plugins" /* Plugins */, this.name, "ratelimitRules", transaction, "returned rules is", data);
+        return {
+            data,
+            revision: this.unbox(rateLimit.revision, "")
+        };
+    }
+    // eslint-disable-next-line max-lines-per-function
+    async acquireQuota(namespace, service, rule, used) {
+        if (!this.initializeStatus.initialized) {
+            await this.waitForInitialized();
+        }
+        const { logger } = this;
+        const { transaction } = logger;
+        const key = `${namespace}#${service}`;
+        logger.trace("Plugins" /* Plugins */, this.name, "acquireQuota", transaction, "select backend with key:", key);
+        const instanceResponse = await this.selectBackend(this.buildStickyConsumer(), this.options.ratelimitService, key);
+        if (logger.tracingEnabled) {
+            const { host, port } = instanceResponse.instance;
+            logger.trace("Plugins" /* Plugins */, this.name, "acquireQuota", transaction, `selected backend is ${host}:${port}`);
+        }
+        let { ratelimitInfo } = this;
+        if (ratelimitInfo === undefined) {
+            ratelimitInfo = Object.create(null);
+            this.ratelimitInfo = ratelimitInfo;
+        }
+        let info = ratelimitInfo[key];
+        /**
+         * 如当前选出的远端实例与上次访问实例不相同（如前一实例配熔断等），则重置全部状态
+         */
+        if (!info || info.backendInstance !== instanceResponse.instance) {
+            logger.trace("Plugins" /* Plugins */, this.name, "acquireQuota", transaction, "current selected instance not equal to the last result, reset all status");
+            info = {
+                backendInstance: instanceResponse.instance,
+                initializedStatus: new WeakMap(),
+                diffTime: 0
+            };
+            ratelimitInfo[key] = info;
+        }
+        const request = {
+            key: this.box(`${rule.id}#${rule.revision}`),
+            namespace: this.box(namespace),
+            service: this.box(service),
+            /**
+             * |server| = |local| + |diff|
+             */
+            timestamp: this.box(Math.floor(Date.now() + info.diffTime))
+        };
+        const initialized = info.initializedStatus.get(rule);
+        let promise;
+        let method;
+        const prx = this.pool.getOrCreateClient(instanceResponse.instance, types_1.ServiceType.Ratelimit);
+        const startTime = (0, process_1.uptime)();
+        if (initialized === true) { /** 如当前限流规则已初始化完成 */
+            request.useds = this.amountToLimiter(used);
+            logger.trace("Plugins" /* Plugins */, this.name, "acquireQuota", transaction, "call backend acquireQuota with args:", request);
+            promise = prx.acquireQuota(request);
+            method = "Acquire";
+        }
+        else if (initialized === undefined) { /** 如当前限流规则在远端未初始化，则先进行初始化 */
+            request.totals = this.amountToLimiter(rule.amounts);
+            logger.trace("Plugins" /* Plugins */, this.name, "acquireQuota", transaction, "call backend initializeQuota with args:", request);
+            promise = prx.initializeQuota(request);
+            method = "Initialize";
+            info.initializedStatus.set(rule, promise);
+        }
+        else { /** 当前限流规则正在初始化 */
+            logger.trace("Plugins" /* Plugins */, this.name, "acquireQuota", transaction, "waiting for initialized");
+            await initialized;
+            logger.trace("Plugins" /* Plugins */, this.name, "acquireQuota", transaction, "initialize completed, Re-call");
+            return this.acquireQuota(namespace, service, rule, used);
+        }
+        let response;
+        try {
+            response = await promise;
+        }
+        catch (e) {
+            logger.trace("Plugins" /* Plugins */, this.name, "acquireQuota", transaction, "call backend exception with", e);
+            instanceResponse.update(false, ((0, process_1.uptime)() - startTime) * utils_1.kSeconds);
+            if (method === "Initialize") {
+                info.initializedStatus.delete(rule);
+            }
+            throw e;
+        }
+        const code = this.unbox(response.code, NaN);
+        this.tracingResponse(transaction, "acquireQuota", response);
+        instanceResponse.update(true, ((0, process_1.uptime)() - startTime) * utils_1.kSeconds, `${code || 0}`);
+        try {
+            this.maybeErrorResponse(response, namespace, service);
+        }
+        catch (e) {
+            if (method === "Initialize") {
+                info.initializedStatus.delete(rule);
+            }
+            throw e;
+        }
+        const serverTime = this.unbox(response.timestamp, NaN);
+        if (!Number.isNaN(serverTime)) {
+            /**
+             * |diff| = |server| + |TTL| - |local|
+             */
+            info.diffTime = serverTime + ((((0, process_1.uptime)() - startTime) / 2) * utils_1.kSeconds) - Date.now();
+            logger.trace("Plugins" /* Plugins */, this.name, "acquireQuota", transaction, "recalculate diff time is", info.diffTime);
+        }
+        if (method === "Initialize") {
+            info.initializedStatus.set(rule, true);
+            logger.trace("Plugins" /* Plugins */, this.name, "acquireQuota", transaction, "initialize completed, Re-call");
+            return this.acquireQuota(namespace, service, rule, used);
+        }
+        const result = [];
+        response.sum_useds.forEach(({ amount, duration }) => {
+            const globalAmount = this.unbox(amount, NaN);
+            if (!Number.isNaN(globalAmount) && duration) {
+                result.push({
+                    amount: globalAmount,
+                    duration: (0, utils_2.protobuf2ms)(duration)
+                });
+            }
+            else {
+                (0, utils_1.UNREACHABLE)();
+            }
+        });
+        logger.trace("Plugins" /* Plugins */, this.name, "acquireQuota", transaction, "quota config is", result);
+        return result;
+    }
+    procRatelimitRule(processor, rule) {
+        const { id, amounts, action, cluster, labels, resource, priority, report: ruleReport, revision: ruleRevision, type, disable } = rule;
+        const ruleId = this.unbox(id, "");
+        const rulePriority = this.unbox(priority, NaN);
+        if (ruleId !== "" && !Number.isNaN(rulePriority) && amounts) {
+            const result = {
+                resource: resource !== null && resource !== void 0 ? resource : __1.LimitResource.QPS,
+                type: type !== null && type !== void 0 ? type : __1.LimitType.Global,
+                id: ruleId,
+                priority: rulePriority,
+                revision: this.unbox(ruleRevision, ""),
+                action: this.unbox(action, ""),
+                cluster: this.unbox(cluster, ""),
+                enable: !this.unbox(disable, false),
+                labels: processor.produceMetadata(labels),
+                amounts: amounts.map(({ maxAmount, validDuration }) => ({
+                    amount: this.unbox(maxAmount, NaN),
+                    duration: validDuration ? (0, utils_2.protobuf2ms)(validDuration) : NaN
+                })).filter(({ amount, duration }) => amount >= 0 && duration >= 1 * utils_1.kSeconds /** 最小周期为 1s */)
+            };
+            if (result.type === __1.LimitType.Global) {
+                const percent = this.unbox(ruleReport === null || ruleReport === void 0 ? void 0 : ruleReport.amountPercent, NaN) / 100;
+                const interval = (ruleReport === null || ruleReport === void 0 ? void 0 : ruleReport.interval) ? (0, utils_2.protobuf2ms)(ruleReport.interval) : NaN;
+                if (percent > 0 && percent <= 100 && Number.isNaN(interval)) {
+                    result.report = {
+                        percent,
+                        interval
+                    };
+                }
+            }
+            if (result.amounts.length > 0) {
+                return result;
+            }
+        }
+    }
+    amountToLimiter(amounts) {
+        return amounts.map(({ amount, duration }) => ({
+            amount: this.box(amount),
+            /**
+             * 在 `protobuf` 的 JSON format 中，`google.protobuf.Duration` 类型为 string，所以这里需要强制转换
+             */
+            duration: (0, utils_2.ms2protobuf)(duration, this.format)
+        }));
+    }
+}
+exports.PolarisRatelimitAdapter = PolarisRatelimitAdapter;
+//# sourceMappingURL=ratelimit.js.map

package/@tencent/polaris/dist/plugins/naming/polaris-server/adapter/ratelimit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ratelimit.js","sourceRoot":"","sources":["../../../../../src/plugins/naming/polaris-server/adapter/ratelimit.ts"],"names":[],"mappings":";;;AAAA,qCAAiC;AAEjC,mCAAkG;AAGlG,6CAA0D;AAE1D,oCAAyC;AACzC,oCAA4D;AAC5D,oCAAoD;AACpD,iCAAgE;AAgBhE,MAAsB,uBAAwB,SAAQ,yBAAkB;IAKtE,KAAK,CAAC,cAAc,CAClB,SAAiB,EAAE,OAAe,EAAE,QAAiB;QAErD,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE;YACtC,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;SACjC;QAED,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACxB,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAC/B,IAAI,MAAM,CAAC,cAAc,EAAE;YACzB,MAAM,CAAC,KAAK,0BACa,IAAI,CAAC,IAAI,EAChC,gBAAgB,EAAE,WAAW,EAAE,WAAW,SAAS,IAAI,OAAO,GAAG,CAAC,QAAQ,aAAR,QAAQ,cAAR,QAAQ,GAAI,EAAE,CAAC,IAAI,IAAI,QAAS,EAAE,EAAE,CACvG,CAAC;SACH;QACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,mBAAW,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,UAAU,EAAE;YACzG,IAAI,EAAE,2BAAmB,CAAC,UAAU;YACpC,OAAO,EAAE;gBACP,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC;gBAC9B,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;aACxB;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,gBAAgB,EAAE,QAAQ,CAAC,CAAC;QAC9D,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QAEtD,MAAM,EAAE,SAAS,EAAE,GAAG,QAAQ,CAAC;QAE/B,IACE,CAAC,CAAC,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,KAAK,CAAC;eAChB,CAAC,QAAQ,KAAK,SAAS,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,CAAC,KAAK,QAAQ,CAAC,EAC9E;YACA,MAAM,CAAC,KAAK,0BACa,IAAI,CAAC,IAAI,EAChC,gBAAgB,EAAE,WAAW,EAAE,kCAAkC,CAClE,CAAC;YACF,OAAO;gBACL,IAAI,EAAE,EAAE;gBACR,QAAQ,EAAE,CAAA,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,KAAK,EAAC,CAAC,CAAC,CAAC,QAAQ,aAAR,QAAQ,cAAR,QAAQ,GAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE;aACnD,CAAC;SACH;QAED,IAAI,EAAE,sBAAsB,EAAE,GAAG,IAAI,CAAC;QACtC,IAAI,sBAAsB,KAAK,SAAS,EAAE;YACxC,sBAAsB,GAAG,IAAI,qBAAa,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAClE,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;SACtD;QAED,MAAM,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC;QAC5B,MAAM,IAAI,GAAoB,EAAE,CAAC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,EAAE,uDAAuD;YACjG,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YACtE,IAAI,IAAI,EAAE;gBACR,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;aACjB;SACF;QAED,MAAM,CAAC,KAAK,0BAAwB,IAAI,CAAC,IAAI,EAAE,gBAAgB,EAAE,WAAW,EAAE,mBAAmB,EAAE,IAAI,CAAC,CAAC;QACzG,OAAO;YACL,IAAI;YACJ,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,CAAC;SAC7C,CAAC;IACJ,CAAC;IAED,kDAAkD;IAClD,KAAK,CAAC,YAAY,CAChB,SAAiB,EAAE,OAAe,EAAE,IAAmB,EAAE,IAAmB;QAE5E,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,WAAW,EAAE;YACtC,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;SACjC;QAED,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACxB,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;QAC/B,MAAM,GAAG,GAAG,GAAG,SAAS,IAAI,OAAO,EAAE,CAAC;QACtC,MAAM,CAAC,KAAK,0BAAwB,IAAI,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,0BAA0B,EAAE,GAAG,CAAC,CAAC;QAC7G,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,mBAAmB,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC;QAClH,IAAI,MAAM,CAAC,cAAc,EAAE;YACzB,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,gBAAgB,CAAC,QAAQ,CAAC;YACjD,MAAM,CAAC,KAAK,0BACa,IAAI,CAAC,IAAI,EAChC,cAAc,EAAE,WAAW,EAAE,uBAAuB,IAAI,IAAI,IAAI,EAAE,CACnE,CAAC;SACH;QAED,IAAI,EAAE,aAAa,EAAE,GAAG,IAAI,CAAC;QAC7B,IAAI,aAAa,KAAK,SAAS,EAAE;YAC/B,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAkB,CAAC;YACrD,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;SACpC;QAED,IAAI,IAAI,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;QAC9B;;WAEG;QACH,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,eAAe,KAAK,gBAAgB,CAAC,QAAQ,EAAE;YAC/D,MAAM,CAAC,KAAK,0BACa,IAAI,CAAC,IAAI,EAChC,cAAc,EAAE,WAAW,EAAE,0EAA0E,CACxG,CAAC;YACF,IAAI,GAAG;gBACL,eAAe,EAAE,gBAAgB,CAAC,QAAQ;gBAC1C,iBAAiB,EAAE,IAAI,OAAO,EAAE;gBAChC,QAAQ,EAAE,CAAC;aACZ,CAAC;YACF,aAAa,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;SAC3B;QAED,MAAM,OAAO,GAA8D;YACzE,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC5C,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC;YAC9B,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;YAC1B;;eAEG;YACH,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;SAC5D,CAAC;QAEF,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,OAA2E,CAAC;QAChF,IAAI,MAAgC,CAAC;QACrC,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,QAAQ,EAAE,mBAAW,CAAC,SAAS,CAAC,CAAC;QAC1F,MAAM,SAAS,GAAG,IAAA,gBAAM,GAAE,CAAC;QAE3B,IAAI,WAAW,KAAK,IAAI,EAAE,EAAE,oBAAoB;YAC9C,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,CAAC,KAAK,0BACa,IAAI,CAAC,IAAI,EAChC,cAAc,EAAE,WAAW,EAAE,sCAAsC,EAAE,OAAO,CAC7E,CAAC;YACF,OAAO,GAAG,GAAG,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACpC,MAAM,GAAG,SAAS,CAAC;SACpB;aAAM,IAAI,WAAW,KAAK,SAAS,EAAE,EAAE,6BAA6B;YACnE,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACpD,MAAM,CAAC,KAAK,0BACa,IAAI,CAAC,IAAI,EAChC,cAAc,EAAE,WAAW,EAAE,yCAAyC,EAAE,OAAO,CAChF,CAAC;YACF,OAAO,GAAG,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YACvC,MAAM,GAAG,YAAY,CAAC;YACtB,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;SAC3C;aAAM,EAAE,kBAAkB;YACzB,MAAM,CAAC,KAAK,0BAAwB,IAAI,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,yBAAyB,CAAC,CAAC;YACvG,MAAM,WAAW,CAAC;YAClB,MAAM,CAAC,KAAK,0BAAwB,IAAI,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,+BAA+B,CAAC,CAAC;YAC7G,OAAO,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;SAC1D;QAED,IAAI,QAAQ,CAAC;QACb,IAAI;YACF,QAAQ,GAAG,MAAM,OAAO,CAAC;SAC1B;QAAC,OAAO,CAAC,EAAE;YACV,MAAM,CAAC,KAAK,0BAAwB,IAAI,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,6BAA6B,EAAE,CAAC,CAAC,CAAC;YAC9G,gBAAgB,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,IAAA,gBAAM,GAAE,GAAG,SAAS,CAAC,GAAG,gBAAQ,CAAC,CAAC;YAClE,IAAI,MAAM,KAAK,YAAY,EAAE;gBAC3B,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;aACrC;YACD,MAAM,CAAC,CAAC;SACT;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QAC5C,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,cAAc,EAAE,QAAQ,CAAC,CAAC;QAC5D,gBAAgB,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,IAAA,gBAAM,GAAE,GAAG,SAAS,CAAC,GAAG,gBAAQ,EAAE,GAAG,IAAI,IAAI,CAAC,EAAE,CAAC,CAAC;QAEjF,IAAI;YACF,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;SACvD;QAAC,OAAO,CAAC,EAAE;YACV,IAAI,MAAM,KAAK,YAAY,EAAE;gBAC3B,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;aACrC;YACD,MAAM,CAAC,CAAC;SACT;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;QACvD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;YAC7B;;eAEG;YACH,IAAI,CAAC,QAAQ,GAAG,UAAU,GAAG,CAAC,CAAC,CAAC,IAAA,gBAAM,GAAE,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,gBAAQ,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACpF,MAAM,CAAC,KAAK,0BACa,IAAI,CAAC,IAAI,EAChC,cAAc,EAAE,WAAW,EAAE,0BAA0B,EAAE,IAAI,CAAC,QAAQ,CACvE,CAAC;SACH;QAED,IAAI,MAAM,KAAK,YAAY,EAAE;YAC3B,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACvC,MAAM,CAAC,KAAK,0BAAwB,IAAI,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,+BAA+B,CAAC,CAAC;YAC7G,OAAO,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;SAC1D;QAED,MAAM,MAAM,GAAkB,EAAE,CAAC;QAEjC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE;YAClD,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAC7C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,QAAQ,EAAE;gBAC3C,MAAM,CAAC,IAAI,CAAC;oBACV,MAAM,EAAE,YAAY;oBACpB,QAAQ,EAAE,IAAA,mBAAW,EAAC,QAAQ,CAAC;iBAChC,CAAC,CAAC;aACJ;iBAAM;gBACL,IAAA,mBAAW,GAAE,CAAC;aACf;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,0BAAwB,IAAI,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,CAAC,CAAC;QACvG,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,iBAAiB,CACvB,SAAwB,EAAE,IAAiD;QAE3E,MAAM,EACJ,EAAE,EAAE,OAAO,EAAE,MAAM,EACnB,OAAO,EAAE,MAAM,EAAE,QAAQ,EACzB,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EACpE,GAAG,IAAI,CAAC;QACT,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAClC,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;QAE/C,IAAI,MAAM,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,OAAO,EAAE;YAC3D,MAAM,MAAM,GAAkB;gBAC5B,QAAQ,EAAE,QAAQ,aAAR,QAAQ,cAAR,QAAQ,GAAI,iBAAa,CAAC,GAAa;gBACjD,IAAI,EAAE,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,aAAS,CAAC,MAAgB;gBACxC,EAAE,EAAE,MAAM;gBACV,QAAQ,EAAE,YAAY;gBACtB,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,EAAE,CAAC;gBACtC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC9B,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;gBAChC,MAAM,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC;gBACnC,MAAM,EAAE,SAAS,CAAC,eAAe,CAAC,MAAM,CAAC;gBACzC,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;oBACtD,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,GAAG,CAAC;oBAClC,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,IAAA,mBAAW,EAAC,aAAa,CAAC,CAAC,CAAC,CAAC,GAAG;iBAC3D,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,MAAM,IAAI,CAAC,IAAI,QAAQ,IAAI,CAAC,GAAG,gBAAQ,CAAC,eAAe,CAAC;aAC9F,CAAC;YACF,IAAI,MAAM,CAAC,IAAI,KAAK,aAAS,CAAC,MAAM,EAAE;gBACpC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,aAAa,EAAE,GAAG,CAAC,GAAG,GAAG,CAAC;gBACjE,MAAM,QAAQ,GAAG,CAAA,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,QAAQ,EAAC,CAAC,CAAC,IAAA,mBAAW,EAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;gBAC/E,IAAI,OAAO,GAAG,CAAC,IAAI,OAAO,IAAI,GAAG,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE;oBAC3D,MAAM,CAAC,MAAM,GAAG;wBACd,OAAO;wBACP,QAAQ;qBACT,CAAC;iBACH;aACF;YACD,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC7B,OAAO,MAAM,CAAC;aACf;SACF;IACH,CAAC;IAEO,eAAe,CAAC,OAAsB;QAC5C,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YAC5C,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC;YACxB;;eAEG;YACH,QAAQ,EAAE,IAAA,mBAAW,EAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAiE;SAC7G,CAAC,CAAC,CAAC;IACN,CAAC;CACF;AA1QD,0DA0QC"}

package/@tencent/polaris/dist/plugins/naming/polaris-server/connector/grpc/bidi.d.ts

@@ -0,0 +1,26 @@
+/// <reference types="node" />
+import { ClientDuplexStream } from "@grpc/grpc-js";
+import { EventEmitter } from "events";
+export declare const enum StreamCallerEvent {
+    PushReceived = "PushReceived"
+}
+export declare enum PacketDirection {
+    Request = 0,
+    Response = 1
+}
+export interface KeyGenerator<RequestType, ResponseType> {
+    (direction: PacketDirection.Request, request: RequestType): string;
+    (direction: PacketDirection.Response, response: ResponseType): string;
+}
+export declare class StreamCaller<RequestType = unknown, ResponseType = unknown> extends EventEmitter {
+    private readonly streamCreator;
+    private readonly keyGenerator;
+    private readonly timeout;
+    private callStream;
+    private activeRequests;
+    constructor(streamCreator: () => ClientDuplexStream<RequestType, ResponseType>, keyGenerator: KeyGenerator<RequestType, ResponseType>, timeout: number);
+    init(): void;
+    request(request: RequestType): Promise<ResponseType>;
+    close(e?: Error): void;
+    private handleResponse;
+}

package/@tencent/polaris/dist/plugins/naming/polaris-server/connector/grpc/bidi.js

@@ -0,0 +1,99 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StreamCaller = exports.PacketDirection = void 0;
+const events_1 = require("events");
+const __1 = require("../../../../..");
+const utils_1 = require("../../../../../utils");
+var PacketDirection;
+(function (PacketDirection) {
+    PacketDirection[PacketDirection["Request"] = 0] = "Request";
+    PacketDirection[PacketDirection["Response"] = 1] = "Response";
+})(PacketDirection = exports.PacketDirection || (exports.PacketDirection = {}));
+class Deferred {
+    constructor() {
+        this.promise = new Promise((resolve, reject) => {
+            this.resolve = resolve;
+            this.reject = reject;
+        });
+        Object.freeze(this);
+    }
+}
+class StreamCaller extends events_1.EventEmitter {
+    constructor(streamCreator, keyGenerator, timeout) {
+        super();
+        this.streamCreator = streamCreator;
+        this.keyGenerator = keyGenerator;
+        this.timeout = timeout;
+        this.callStream = null;
+        this.activeRequests = Object.create(null);
+    }
+    init() {
+        if (this.callStream === null) {
+            this.callStream = this.streamCreator();
+            this.callStream.on("data", this.handleResponse.bind(this));
+            this.callStream.on("end", this.close.bind(this));
+            this.callStream.on("error", this.close.bind(this));
+        }
+    }
+    async request(request) {
+        const priKey = this.keyGenerator(PacketDirection.Request, request);
+        let call = this.activeRequests[priKey];
+        if (!call) {
+            const deferred = new Deferred();
+            call = {
+                deferred,
+                timer: setTimeout(() => {
+                    deferred.reject(new __1.TimeoutError("Exceed deadline"));
+                    delete this.activeRequests[priKey];
+                }, this.timeout).unref()
+            };
+            this.activeRequests[priKey] = call;
+            this.init();
+            this.callStream.write(request);
+        }
+        return call.deferred.promise;
+    }
+    close(e) {
+        const requests = Object.values(this.activeRequests);
+        if (requests.length > 0) {
+            requests.forEach(({ timer, deferred: { reject } }) => {
+                clearTimeout(timer);
+                if (e) {
+                    reject(new __1.NetworkError(e));
+                }
+                else {
+                    reject(new __1.NetworkError("Stream end"));
+                }
+            });
+            this.activeRequests = Object.create(null);
+        }
+        const { callStream } = this;
+        if (callStream !== null) {
+            this.callStream = null;
+            callStream.cancel();
+            callStream.end();
+            callStream.removeAllListeners();
+            callStream.on("error", () => { }); /** no longer receive error event on this stream */
+        }
+    }
+    handleResponse(response) {
+        const priKey = this.keyGenerator(PacketDirection.Response, response);
+        if (priKey === "") {
+            (0, utils_1.UNREACHABLE)();
+            return;
+        }
+        const keys = Object.keys(this.activeRequests);
+        for (let i = 0; i < keys.length; i += 1) { // eslint-disable-line @typescript-eslint/prefer-for-of
+            if (keys[i] === priKey) {
+                const { deferred, timer } = this.activeRequests[priKey];
+                delete this.activeRequests[priKey];
+                clearTimeout(timer);
+                deferred.resolve(response);
+                return;
+            }
+        }
+        this.emit("PushReceived" /* PushReceived */, response);
+    }
+}
+exports.StreamCaller = StreamCaller;
+//# sourceMappingURL=bidi.js.map

package/@tencent/polaris/dist/plugins/naming/polaris-server/connector/grpc/bidi.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bidi.js","sourceRoot":"","sources":["../../../../../../src/plugins/naming/polaris-server/connector/grpc/bidi.ts"],"names":[],"mappings":";;;AACA,mCAAsC;AAEtC,sCAA4D;AAC5D,gDAAmD;AAMnD,IAAY,eAGX;AAHD,WAAY,eAAe;IACzB,2DAAO,CAAA;IACP,6DAAQ,CAAA;AACV,CAAC,EAHW,eAAe,GAAf,uBAAe,KAAf,uBAAe,QAG1B;AAYD,MAAM,QAAQ;IAKZ;QACE,IAAI,CAAC,OAAO,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC7C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;YACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACvB,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACtB,CAAC;CACF;AAED,MAAa,YAA4D,SAAQ,qBAAY;IAI3F,YACmB,aAAkE,EAClE,YAAqD,EACrD,OAAe;QAEhC,KAAK,EAAE,CAAC;QAJS,kBAAa,GAAb,aAAa,CAAqD;QAClE,iBAAY,GAAZ,YAAY,CAAyC;QACrD,YAAO,GAAP,OAAO,CAAQ;QAN1B,eAAU,GAAyD,IAAI,CAAC;QACxE,mBAAc,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAiC,CAAC;IAQ7E,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,UAAU,KAAK,IAAI,EAAE;YAC5B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;YACvC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAC3D,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YACjD,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;SACpD;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAoB;QAChC,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACnE,IAAI,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QACvC,IAAI,CAAC,IAAI,EAAE;YACT,MAAM,QAAQ,GAAG,IAAI,QAAQ,EAAgB,CAAC;YAC9C,IAAI,GAAG;gBACL,QAAQ;gBACR,KAAK,EAAE,UAAU,CAAC,GAAG,EAAE;oBACrB,QAAQ,CAAC,MAAM,CAAC,IAAI,gBAAY,CAAC,iBAAiB,CAAC,CAAC,CAAC;oBACrD,OAAO,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;gBACrC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE;aACzB,CAAC;YACF,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC;YAEnC,IAAI,CAAC,IAAI,EAAE,CAAC;YACZ,IAAI,CAAC,UAAW,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;SACjC;QAED,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,CAAS;QACb,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAoD,CAAC;QACvG,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;YACvB,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE;gBACnD,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpB,IAAI,CAAC,EAAE;oBACL,MAAM,CAAC,IAAI,gBAAY,CAAC,CAAC,CAAC,CAAC,CAAC;iBAC7B;qBAAM;oBACL,MAAM,CAAC,IAAI,gBAAY,CAAC,YAAY,CAAC,CAAC,CAAC;iBACxC;YACH,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAiC,CAAC;SAC3E;QAED,MAAM,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC;QAC5B,IAAI,UAAU,KAAK,IAAI,EAAE;YACvB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACvB,UAAU,CAAC,MAAM,EAAE,CAAC;YACpB,UAAU,CAAC,GAAG,EAAE,CAAC;YACjB,UAAU,CAAC,kBAAkB,EAAE,CAAC;YAChC,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC,mDAAmD;SACtF;IACH,CAAC;IAEO,cAAc,CAAC,QAAsB;QAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACrE,IAAI,MAAM,KAAK,EAAE,EAAE;YACjB,IAAA,mBAAW,GAAE,CAAC;YACd,OAAO;SACR;QACD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,EAAE,uDAAuD;YAChG,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,MAAM,EAAE;gBACtB,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAE,CAAC;gBACzD,OAAO,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;gBACnC,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpB,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBAC3B,OAAO;aACR;SACF;QAED,IAAI,CAAC,IAAI,oCAAiC,QAAQ,CAAC,CAAC;IACtD,CAAC;CACF;AArFD,oCAqFC"}

package/@tencent/polaris/dist/plugins/naming/polaris-server/connector/grpc/index.d.ts

@@ -0,0 +1,60 @@
+import type * as grpc from "@grpc/grpc-js"; /** only import type */
+import { PartialSubset } from "../../../../../utils";
+import type { BoxType, PolarisClient } from "../.."; /** only import type */
+import { PolarisServerAdapter, ServerAdapterOptions } from "../../adapter";
+import { ServiceType } from "../../types";
+import { ProtobufFormat } from "../../utils";
+declare type ServiceClientCtor = ReturnType<typeof grpc.makeClientConstructor>;
+declare const kDefaultOptions: {
+    /**
+     * 调用超时时间
+     */
+    timeout: number;
+    /**
+     * gRPC 配置项
+     */
+    grpc: {
+        /**
+         * gRPC CallOptions 配置项
+         */
+        call: Partial<Omit<grpc.CallOptions, "deadline">>;
+        /**
+         * gRPC ChannelOptions 配置项
+         */
+        channel: Partial<Partial<grpc.ChannelOptions>>;
+    };
+};
+export declare type PolarisGRPCClientOptions = typeof kDefaultOptions & ServerAdapterOptions;
+export declare class PolarisGRPCClient extends PolarisServerAdapter {
+    static readonly protocol = "grpc";
+    static readonly port = 8081;
+    static isSupported(): boolean;
+    readonly name = "PolarisGRPCClient";
+    readonly protocol = "grpc";
+    readonly format = ProtobufFormat.Original;
+    protected readonly options: PolarisGRPCClientOptions;
+    private readonly clientConstructor;
+    constructor(remotes: string[], options?: Partial<PartialSubset<PolarisGRPCClientOptions, "grpc">>);
+    box<T>(value: T): BoxType<T> | T | null | undefined;
+    unbox<T>(value: BoxType<T> | T | null | undefined, defaultValue: T | (() => T)): T;
+    /**
+     * 实例化 Service Client，可重载此方法实现差异化配置
+     * @param address 远端地址
+     * @param type 远端类型
+     * @param ServiceClient Service Client Class，函数必须返回它的实例
+     * @param credentials grpc.credentials
+     */
+    protected instantiateServiceClient(address: string, type: ServiceType, ServiceClient: ServiceClientCtor, credentials: typeof grpc.credentials): InstanceType<ServiceClientCtor>;
+    /**
+     * 获取 grpc.CallOptions 配置，可重载此方法实现差异化配置
+     * @param address 远端地址
+     * @param method 方法名
+     * @param streaming 是否为 Bidi Stream
+     */
+    protected getCallOptions(address: string, method: string, streaming: boolean): grpc.CallOptions;
+    protected buildClient<T extends PolarisClient>(address: string, type: ServiceType): T;
+    private requestKeyGenerator;
+    private responseKeyGenerator;
+    private keyGenerator;
+}
+export {};