Java Code Examples for io.grpc.LoadBalancer#Helper

@VisibleForTesting
static OrcaReportingHelperWrapper newOrcaReportingHelperWrapper(
    LoadBalancer.Helper delegate,
    OrcaOobReportListener listener,
    BackoffPolicy.Provider backoffPolicyProvider,
    Supplier<Stopwatch> stopwatchSupplier) {
  final OrcaReportingHelper orcaHelper =
      new OrcaReportingHelper(delegate, listener, backoffPolicyProvider, stopwatchSupplier);

  return new OrcaReportingHelperWrapper() {
    @Override
    public void setReportingConfig(OrcaReportingConfig config) {
      orcaHelper.setReportingConfig(config);
    }

    @Override
    public Helper asHelper() {
      return orcaHelper;
    }
  };
}
 

OrcaReportingHelper(
    LoadBalancer.Helper delegate,
    OrcaOobReportListener listener,
    BackoffPolicy.Provider backoffPolicyProvider,
    Supplier<Stopwatch> stopwatchSupplier) {
  this.delegate = checkNotNull(delegate, "delegate");
  this.listener = checkNotNull(listener, "listener");
  this.backoffPolicyProvider = checkNotNull(backoffPolicyProvider, "backoffPolicyProvider");
  this.stopwatchSupplier = checkNotNull(stopwatchSupplier, "stopwatchSupplier");
  syncContext = checkNotNull(delegate.getSynchronizationContext(), "syncContext");
}
 

@Override
public OrcaReportingHelperWrapper newOrcaReportingHelperWrapper(
    LoadBalancer.Helper delegate,
    OrcaOobReportListener listener) {
  return newOrcaReportingHelperWrapper(
      delegate,
      listener,
      new ExponentialBackoffPolicy.Provider(),
      GrpcUtil.STOPWATCH_SUPPLIER);
}
 

@Override
public LoadBalancer newLoadBalancer(LoadBalancer.Helper helper) {
  return
      new GrpclbLoadBalancer(
          helper,
          new CachedSubchannelPool(helper),
          TimeProvider.SYSTEM_TIME_PROVIDER,
          Stopwatch.createUnstarted(),
          new ExponentialBackoffPolicy.Provider());
}
 

@Override
public void handleResolvedAddresses(ResolvedAddresses resolvedAddresses) {
  LrsConfig config = (LrsConfig) resolvedAddresses.getLoadBalancingPolicyConfig();
  LoadStatsStore store =
      resolvedAddresses.getAttributes().get(XdsAttributes.ATTR_CLUSTER_SERVICE_LOAD_STATS_STORE);
  checkNotNull(config, "missing LRS lb config");
  checkNotNull(store, "missing cluster service stats object");
  checkAndSetUp(config, store);

  if (switchingLoadBalancer == null) {
    loadStatsStore.addLocality(config.locality);
    final ClientLoadCounter counter = loadStatsStore.getLocalityCounter(config.locality);
    LoadBalancer.Helper loadRecordingHelper = new ForwardingLoadBalancerHelper() {
      @Override
      protected Helper delegate() {
        return helper;
      }

      @Override
      public void updateBalancingState(ConnectivityState newState, SubchannelPicker newPicker) {
        SubchannelPicker loadRecordingPicker =
            new LoadRecordingSubchannelPicker(counter, newPicker);
        super.updateBalancingState(newState, loadRecordingPicker);
      }
    };
    switchingLoadBalancer = new GracefulSwitchLoadBalancer(loadRecordingHelper);
  }
  String updatedChildPolicyName = config.childPolicy.getProvider().getPolicyName();
  if (!Objects.equals(childPolicyName, updatedChildPolicyName)) {
    switchingLoadBalancer.switchTo(config.childPolicy.getProvider());
    childPolicyName = updatedChildPolicyName;
  }
  ResolvedAddresses downStreamResult =
      resolvedAddresses.toBuilder()
          .setLoadBalancingPolicyConfig(config.childPolicy.getConfig())
          .build();
  switchingLoadBalancer.handleResolvedAddresses(downStreamResult);
}
 

@Override
public LoadBalancer newLoadBalancer(LoadBalancer.Helper helper) {
  return new RoundRobinLoadBalancer(helper);
}
 

@Override
public LoadBalancer newLoadBalancer(LoadBalancer.Helper helper) {
  return provider.newLoadBalancer(helper);
}
 

@Override
public LoadBalancer newLoadBalancer(LoadBalancer.Helper helper) {
  return new PickFirstLoadBalancer(helper);
}
 

@Override
public LoadBalancer newLoadBalancer(LoadBalancer.Helper helper) {
  return new RlsLoadBalancer(helper);
}
 

@Override
public LoadBalancer newLoadBalancer(LoadBalancer.Helper helper) {
  return new LrsLoadBalancer(helper);
}
 

LrsLoadBalancer(LoadBalancer.Helper helper) {
  this.helper = checkNotNull(helper, "helper");
}
 

@Override
public LoadBalancer newLoadBalancer(LoadBalancer.Helper helper) {
    return new CustomLoadBalancer(helper);
}
 

@Override
protected LoadBalancer.Helper delegate() {
  return mockDelegate;
}
 

@Override
protected LoadBalancer.Helper delegate() {
  return mockDelegate;
}
 

@Override
public LoadBalancer newLoadBalancer(LoadBalancer.Helper helper) {
  return new RoundRobinLoadBalancer(helper);
}
 

@Override
public LoadBalancer newLoadBalancer(LoadBalancer.Helper helper) {
  return new PickFirstLoadBalancer(helper);
}
 

@Override
public LoadBalancer newLoadBalancer(LoadBalancer.Helper helper) {
  return new GrpclbLoadBalancer(
      helper, new CachedSubchannelPool(), TimeProvider.SYSTEM_TIME_PROVIDER,
      new ExponentialBackoffPolicy.Provider());
}
 

/**
 * Returns the underlying helper.
 */
protected abstract LoadBalancer.Helper delegate();
 

/**
 * Returns a wrapped {@link LoadBalancer.Helper}. Subchannels created through it will retrieve
 * ORCA load reports if the server supports it.
 */
public abstract LoadBalancer.Helper asHelper();
 

/**
 * Returns the underlying helper.
 */
protected abstract LoadBalancer.Helper delegate();