Merge pull request #2058 from murgatroid99/grpc-js_outlier_detection

grpc-js: Add outlier detection LB policy

Author: murgatroid99

packages/grpc-js/gulpfile.ts

@@ -67,6 +67,7 @@ const compile = checkTask(() => execNpmCommand('compile'));
 const copyTestFixtures = checkTask(() => ncpP(`${jsCoreDir}/test/fixtures`, `${outDir}/test/fixtures`));
 
 const runTests = checkTask(() => {
+  process.env.GRPC_EXPERIMENTAL_ENABLE_OUTLIER_DETECTION = 'true';
   return gulp.src(`${outDir}/test/**/*.js`)
     .pipe(mocha({reporter: 'mocha-jenkins-reporter',
                  require: ['ts-node/register']}));

packages/grpc-js/src/channel.ts

@@ -49,6 +49,7 @@ import { Filter } from './filter';
 
 import { ConnectivityState } from './connectivity-state';
 import { ChannelInfo, ChannelRef, ChannelzCallTracker, ChannelzChildrenTracker, ChannelzTrace, registerChannelzChannel, SubchannelRef, unregisterChannelzRef } from './channelz';
+import { Subchannel } from './subchannel';
 
 /**
  * See https://nodejs.org/api/timers.html#timers_setinterval_callback_delay_args
@@ -451,7 +452,7 @@ export class ChannelImplementation implements Channel {
                 if (subchannelState === ConnectivityState.READY) {
                   try {
                     const pickExtraFilters = pickResult.extraFilterFactories.map(factory => factory.createFilter(callStream));
-                    pickResult.subchannel!.startCallStream(
+                    pickResult.subchannel?.getRealSubchannel().startCallStream(
                       finalMetadata,
                       callStream,
                       [...dynamicFilters, ...pickExtraFilters]

packages/grpc-js/src/duration.ts

@@ -0,0 +1,36 @@
+/*
+ * Copyright 2022 gRPC authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+export interface Duration {
+  seconds: number;
+  nanos: number;
+}
+
+export function msToDuration(millis: number): Duration {
+  return {
+    seconds: (millis / 1000) | 0,
+    nanos: (millis % 1000) * 1_000_000 | 0
+  };
+}
+
+export function durationToMs(duration: Duration): number {
+  return (duration.seconds * 1000 + duration.nanos / 1_000_000) | 0;
+}
+
+export function isDuration(value: any): value is Duration {
+  return (typeof value.seconds === 'number') && (typeof value.nanos === 'number');
+}

packages/grpc-js/src/experimental.ts

@@ -6,7 +6,8 @@ export {
   ConfigSelector,
 } from './resolver';
 export { GrpcUri, uriToString } from './uri-parser';
-export { ServiceConfig, Duration } from './service-config';
+export { Duration } from './duration';
+export { ServiceConfig } from './service-config';
 export { BackoffTimeout } from './backoff-timeout';
 export {
   LoadBalancer,
@@ -34,3 +35,4 @@ export { Call as CallStream } from './call-stream';
 export { Filter, BaseFilter, FilterFactory } from './filter';
 export { FilterStackFactory } from './filter-stack';
 export { registerAdminService } from './admin';
+export { SubchannelInterface, BaseSubchannelWrapper, ConnectivityStateListener } from './subchannel-interface'

packages/grpc-js/src/index.ts

@@ -273,6 +273,7 @@ import * as resolver_uds from './resolver-uds';
 import * as resolver_ip from './resolver-ip';
 import * as load_balancer_pick_first from './load-balancer-pick-first';
 import * as load_balancer_round_robin from './load-balancer-round-robin';
+import * as load_balancer_outlier_detection from './load-balancer-outlier-detection';
 import * as channelz from './channelz';
 
 const clientVersion = require('../../package.json').version;
@@ -284,5 +285,6 @@ const clientVersion = require('../../package.json').version;
   resolver_ip.setup();
   load_balancer_pick_first.setup();
   load_balancer_round_robin.setup();
+  load_balancer_outlier_detection.setup();
   channelz.setup();
 })();

packages/grpc-js/src/load-balancer-child-handler.ts

@@ -21,12 +21,12 @@ import {
   LoadBalancingConfig,
   createLoadBalancer,
 } from './load-balancer';
-import { Subchannel } from './subchannel';
 import { SubchannelAddress } from './subchannel-address';
 import { ChannelOptions } from './channel-options';
 import { ConnectivityState } from './connectivity-state';
 import { Picker } from './picker';
 import { ChannelRef, SubchannelRef } from './channelz';
+import { SubchannelInterface } from './subchannel-interface';
 
 const TYPE_NAME = 'child_load_balancer_helper';
 
@@ -40,7 +40,7 @@ export class ChildLoadBalancerHandler implements LoadBalancer {
     createSubchannel(
       subchannelAddress: SubchannelAddress,
       subchannelArgs: ChannelOptions
-    ): Subchannel {
+    ): SubchannelInterface {
       return this.parent.channelControlHelper.createSubchannel(
         subchannelAddress,
         subchannelArgs