grpc
/
grpc-java
mirror of https://github.com/grpc/grpc-java.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

example-orca: fix lost streamTracerFactor (#9411)

This commit is contained in:
yifeizhuang 2022-07-25 16:28:58 -07:00 committed by GitHub
parent 5f1a1d4f35
commit cc9505f6bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 19 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
examples/example-orca/src/main/java/io/grpc/examples/orca/CustomBackendMetricsLoadBalancerProvider.java
View File

@ -124,16 +124,24 @@ final class CustomBackendMetricsLoadBalancerProvider extends LoadBalancerProvide
return result; return result;
} }
// Installs ORCA per-query metrics reporting listener. // Installs ORCA per-query metrics reporting listener.
return LoadBalancer.PickResult.withSubchannel( final OrcaPerRequestUtil.OrcaPerRequestReportListener orcaListener =
result.getSubchannel(), new OrcaPerRequestUtil.OrcaPerRequestReportListener() {
OrcaPerRequestUtil.getInstance().newOrcaClientStreamTracerFactory( @Override
new OrcaPerRequestUtil.OrcaPerRequestReportListener() { public void onLoadReport(MetricReport orcaLoadReport) {
@Override System.out.println("Example load balancer received per-rpc metrics report:\n"
public void onLoadReport(MetricReport orcaLoadReport) { + orcaLoadReport);
System.out.println("Example load balancer received per-rpc metrics report:\n" }
+ orcaLoadReport); };
} if (result.getStreamTracerFactory() == null) {
})); return LoadBalancer.PickResult.withSubchannel(
result.getSubchannel(),
OrcaPerRequestUtil.getInstance().newOrcaClientStreamTracerFactory(orcaListener));
} else {
return LoadBalancer.PickResult.withSubchannel(
result.getSubchannel(),
OrcaPerRequestUtil.getInstance().newOrcaClientStreamTracerFactory(
result.getStreamTracerFactory(), orcaListener));
}
} }
} }
} }

1
services/BUILD.bazel
View File

@ -43,6 +43,7 @@ java_library(
"//api", "//api",
"//context", "//context",
"@com_google_code_findbugs_jsr305//jar", "@com_google_code_findbugs_jsr305//jar",
"@com_google_errorprone_error_prone_annotations//jar",
"@com_google_guava_guava//jar", "@com_google_guava_guava//jar",
], ],
) )