Make AltsServerBuilder extend ForwardingServerBuilder.

This commit is contained in:
Chengyuan Zhang 2021-03-04 18:23:09 -08:00
parent 18f5fc2d7f
commit 9e2c817cd8
1 changed files with 7 additions and 9 deletions

View File

@ -20,6 +20,7 @@ import io.grpc.BindableService;
import io.grpc.CompressorRegistry; import io.grpc.CompressorRegistry;
import io.grpc.DecompressorRegistry; import io.grpc.DecompressorRegistry;
import io.grpc.ExperimentalApi; import io.grpc.ExperimentalApi;
import io.grpc.ForwardingServerBuilder;
import io.grpc.HandlerRegistry; import io.grpc.HandlerRegistry;
import io.grpc.Server; import io.grpc.Server;
import io.grpc.ServerBuilder; import io.grpc.ServerBuilder;
@ -30,7 +31,6 @@ import io.grpc.ServerTransportFilter;
import io.grpc.netty.NettyServerBuilder; import io.grpc.netty.NettyServerBuilder;
import java.io.File; import java.io.File;
import java.net.InetSocketAddress; import java.net.InetSocketAddress;
import java.util.List;
import java.util.concurrent.Executor; import java.util.concurrent.Executor;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
@ -39,7 +39,7 @@ import java.util.concurrent.TimeUnit;
* a production server on Google Cloud Platform. * a production server on Google Cloud Platform.
*/ */
@ExperimentalApi("https://github.com/grpc/grpc-java/issues/4151") @ExperimentalApi("https://github.com/grpc/grpc-java/issues/4151")
public final class AltsServerBuilder extends ServerBuilder<AltsServerBuilder> { public final class AltsServerBuilder extends ForwardingServerBuilder<AltsServerBuilder> {
private final NettyServerBuilder delegate; private final NettyServerBuilder delegate;
private final AltsServerCredentials.Builder credentialsBuilder = private final AltsServerCredentials.Builder credentialsBuilder =
new AltsServerCredentials.Builder(); new AltsServerCredentials.Builder();
@ -69,6 +69,11 @@ public final class AltsServerBuilder extends ServerBuilder<AltsServerBuilder> {
return this; return this;
} }
@Override
protected ServerBuilder<?> delegate() {
return delegate;
}
/** {@inheritDoc} */ /** {@inheritDoc} */
@Override @Override
public AltsServerBuilder handshakeTimeout(long timeout, TimeUnit unit) { public AltsServerBuilder handshakeTimeout(long timeout, TimeUnit unit) {
@ -118,13 +123,6 @@ public final class AltsServerBuilder extends ServerBuilder<AltsServerBuilder> {
return this; return this;
} }
/** {@inheritDoc} */
@Override
public AltsServerBuilder addServices(List<ServerServiceDefinition> services) {
delegate.addServices(services);
return this;
}
/** {@inheritDoc} */ /** {@inheritDoc} */
@Override @Override
public AltsServerBuilder fallbackHandlerRegistry(HandlerRegistry fallbackRegistry) { public AltsServerBuilder fallbackHandlerRegistry(HandlerRegistry fallbackRegistry) {