Add support for anonymous in-process servers. (#8589)

Support anonymous in-process servers, and InProcessChannelBuilder.forTarget.

Anonymous servers aren't registered statically, meaning they can't be looked up by name.
Only the AnonymousInProcessSocketAddress passed to InProcessServerBuilder.forAddress(),
(or subsequently fetched from Server.getListenSockets()) can be used to connect to the server.

Supporting InProcessChannelBuilder.forTarget is particularly useful for production
Android usage of in-process servers, where process startup latency is crucial.
A custom name resolver can be used to create the server instance on demand
without directly impacting the startup latency of in-process gRPC clients.

Together, these features support a more-standard approach to "OnDeviceServer" referenced in gRFC L73.
https://github.com/grpc/proposal/blob/master/L73-java-binderchannel.md#ondeviceserver
This commit is contained in:
markb74 2021-10-25 20:59:48 +02:00 committed by GitHub
parent 203515dd3d
commit 607362a7d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
8 changed files with 316 additions and 46 deletions

View File

@ -0,0 +1,58 @@
/*
* Copyright 2021 The 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.
*/
package io.grpc.inprocess;
import static com.google.common.base.Preconditions.checkState;
import io.grpc.ExperimentalApi;
import java.io.IOException;
import java.net.SocketAddress;
import javax.annotation.Nullable;
import javax.annotation.concurrent.GuardedBy;
/**
* Custom SocketAddress class for {@link InProcessTransport}, for
* a server which can only be referenced via this address instance.
*/
@ExperimentalApi("https://github.com/grpc/grpc-java/issues/8626")
public final class AnonymousInProcessSocketAddress extends SocketAddress {
private static final long serialVersionUID = -8567592561863414695L;
@Nullable
@GuardedBy("this")
private InProcessServer server;
/** Creates a new AnonymousInProcessSocketAddress. */
public AnonymousInProcessSocketAddress() { }
@Nullable
synchronized InProcessServer getServer() {
return server;
}
synchronized void setServer(InProcessServer server) throws IOException {
if (this.server != null) {
throw new IOException("Server instance already registered");
}
this.server = server;
}
synchronized void clearServer(InProcessServer server) {
checkState(this.server == server);
this.server = null;
}
}

View File

@ -55,15 +55,28 @@ public final class InProcessChannelBuilder extends
* @return a new builder * @return a new builder
*/ */
public static InProcessChannelBuilder forName(String name) { public static InProcessChannelBuilder forName(String name) {
return new InProcessChannelBuilder(name); return forAddress(new InProcessSocketAddress(checkNotNull(name, "name")));
} }
/** /**
* Always fails. Call {@link #forName} instead. * Create a channel builder that will connect to the server referenced by the given target URI.
* Only intended for use with a custom name resolver.
*
* @param target the identity of the server to connect to
* @return a new builder
*/ */
@DoNotCall("Unsupported. Use forName() instead")
public static InProcessChannelBuilder forTarget(String target) { public static InProcessChannelBuilder forTarget(String target) {
throw new UnsupportedOperationException("call forName() instead"); return new InProcessChannelBuilder(null, checkNotNull(target, "target"));
}
/**
* Create a channel builder that will connect to the server referenced by the given address.
*
* @param address the address of the server to connect to
* @return a new builder
*/
public static InProcessChannelBuilder forAddress(SocketAddress address) {
return new InProcessChannelBuilder(checkNotNull(address, "address"), null);
} }
/** /**
@ -75,13 +88,11 @@ public final class InProcessChannelBuilder extends
} }
private final ManagedChannelImplBuilder managedChannelImplBuilder; private final ManagedChannelImplBuilder managedChannelImplBuilder;
private final String name;
private ScheduledExecutorService scheduledExecutorService; private ScheduledExecutorService scheduledExecutorService;
private int maxInboundMetadataSize = Integer.MAX_VALUE; private int maxInboundMetadataSize = Integer.MAX_VALUE;
private boolean transportIncludeStatusCause = false; private boolean transportIncludeStatusCause = false;
private InProcessChannelBuilder(String name) { private InProcessChannelBuilder(@Nullable SocketAddress directAddress, @Nullable String target) {
this.name = checkNotNull(name, "name");
final class InProcessChannelTransportFactoryBuilder implements ClientTransportFactoryBuilder { final class InProcessChannelTransportFactoryBuilder implements ClientTransportFactoryBuilder {
@Override @Override
@ -90,8 +101,13 @@ public final class InProcessChannelBuilder extends
} }
} }
managedChannelImplBuilder = new ManagedChannelImplBuilder(new InProcessSocketAddress(name), if (directAddress != null) {
"localhost", new InProcessChannelTransportFactoryBuilder(), null); managedChannelImplBuilder = new ManagedChannelImplBuilder(directAddress, "localhost",
new InProcessChannelTransportFactoryBuilder(), null);
} else {
managedChannelImplBuilder = new ManagedChannelImplBuilder(target,
new InProcessChannelTransportFactoryBuilder(), null);
}
// In-process transport should not record its traffic to the stats module. // In-process transport should not record its traffic to the stats module.
// https://github.com/grpc/grpc-java/issues/2284 // https://github.com/grpc/grpc-java/issues/2284
@ -204,7 +220,7 @@ public final class InProcessChannelBuilder extends
ClientTransportFactory buildTransportFactory() { ClientTransportFactory buildTransportFactory() {
return new InProcessClientTransportFactory( return new InProcessClientTransportFactory(
name, scheduledExecutorService, maxInboundMetadataSize, transportIncludeStatusCause); scheduledExecutorService, maxInboundMetadataSize, transportIncludeStatusCause);
} }
void setStatsEnabled(boolean value) { void setStatsEnabled(boolean value) {
@ -215,7 +231,6 @@ public final class InProcessChannelBuilder extends
* Creates InProcess transports. Exposed for internal use, as it should be private. * Creates InProcess transports. Exposed for internal use, as it should be private.
*/ */
static final class InProcessClientTransportFactory implements ClientTransportFactory { static final class InProcessClientTransportFactory implements ClientTransportFactory {
private final String name;
private final ScheduledExecutorService timerService; private final ScheduledExecutorService timerService;
private final boolean useSharedTimer; private final boolean useSharedTimer;
private final int maxInboundMetadataSize; private final int maxInboundMetadataSize;
@ -223,10 +238,8 @@ public final class InProcessChannelBuilder extends
private final boolean includeCauseWithStatus; private final boolean includeCauseWithStatus;
private InProcessClientTransportFactory( private InProcessClientTransportFactory(
String name,
@Nullable ScheduledExecutorService scheduledExecutorService, @Nullable ScheduledExecutorService scheduledExecutorService,
int maxInboundMetadataSize, boolean includeCauseWithStatus) { int maxInboundMetadataSize, boolean includeCauseWithStatus) {
this.name = name;
useSharedTimer = scheduledExecutorService == null; useSharedTimer = scheduledExecutorService == null;
timerService = useSharedTimer timerService = useSharedTimer
? SharedResourceHolder.get(GrpcUtil.TIMER_SERVICE) : scheduledExecutorService; ? SharedResourceHolder.get(GrpcUtil.TIMER_SERVICE) : scheduledExecutorService;
@ -242,7 +255,7 @@ public final class InProcessChannelBuilder extends
} }
// TODO(carl-mastrangelo): Pass channelLogger in. // TODO(carl-mastrangelo): Pass channelLogger in.
return new InProcessTransport( return new InProcessTransport(
name, maxInboundMetadataSize, options.getAuthority(), options.getUserAgent(), addr, maxInboundMetadataSize, options.getAuthority(), options.getUserAgent(),
options.getEagAttributes(), includeCauseWithStatus); options.getEagAttributes(), includeCauseWithStatus);
} }

View File

@ -40,11 +40,16 @@ final class InProcessServer implements InternalServer {
private static final ConcurrentMap<String, InProcessServer> registry private static final ConcurrentMap<String, InProcessServer> registry
= new ConcurrentHashMap<>(); = new ConcurrentHashMap<>();
static InProcessServer findServer(String name) { static InProcessServer findServer(SocketAddress addr) {
return registry.get(name); if (addr instanceof AnonymousInProcessSocketAddress) {
return ((AnonymousInProcessSocketAddress) addr).getServer();
} else if (addr instanceof InProcessSocketAddress) {
return registry.get(((InProcessSocketAddress) addr).getName());
}
return null;
} }
private final String name; private final SocketAddress listenAddress;
private final int maxInboundMetadataSize; private final int maxInboundMetadataSize;
private final List<ServerStreamTracer.Factory> streamTracerFactories; private final List<ServerStreamTracer.Factory> streamTracerFactories;
private ServerListener listener; private ServerListener listener;
@ -60,7 +65,7 @@ final class InProcessServer implements InternalServer {
InProcessServer( InProcessServer(
InProcessServerBuilder builder, InProcessServerBuilder builder,
List<? extends ServerStreamTracer.Factory> streamTracerFactories) { List<? extends ServerStreamTracer.Factory> streamTracerFactories) {
this.name = builder.name; this.listenAddress = builder.listenAddress;
this.schedulerPool = builder.schedulerPool; this.schedulerPool = builder.schedulerPool;
this.maxInboundMetadataSize = builder.maxInboundMetadataSize; this.maxInboundMetadataSize = builder.maxInboundMetadataSize;
this.streamTracerFactories = this.streamTracerFactories =
@ -72,14 +77,25 @@ final class InProcessServer implements InternalServer {
this.listener = serverListener; this.listener = serverListener;
this.scheduler = schedulerPool.getObject(); this.scheduler = schedulerPool.getObject();
// Must be last, as channels can start connecting after this point. // Must be last, as channels can start connecting after this point.
registerInstance();
}
private void registerInstance() throws IOException {
if (listenAddress instanceof AnonymousInProcessSocketAddress) {
((AnonymousInProcessSocketAddress) listenAddress).setServer(this);
} else if (listenAddress instanceof InProcessSocketAddress) {
String name = ((InProcessSocketAddress) listenAddress).getName();
if (registry.putIfAbsent(name, this) != null) { if (registry.putIfAbsent(name, this) != null) {
throw new IOException("name already registered: " + name); throw new IOException("name already registered: " + name);
} }
} else {
throw new AssertionError();
}
} }
@Override @Override
public SocketAddress getListenSocketAddress() { public SocketAddress getListenSocketAddress() {
return new InProcessSocketAddress(name); return listenAddress;
} }
@Override @Override
@ -99,9 +115,7 @@ final class InProcessServer implements InternalServer {
@Override @Override
public void shutdown() { public void shutdown() {
if (!registry.remove(name, this)) { unregisterInstance();
throw new AssertionError();
}
scheduler = schedulerPool.returnObject(scheduler); scheduler = schedulerPool.returnObject(scheduler);
synchronized (this) { synchronized (this) {
shutdown = true; shutdown = true;
@ -109,9 +123,22 @@ final class InProcessServer implements InternalServer {
} }
} }
private void unregisterInstance() {
if (listenAddress instanceof AnonymousInProcessSocketAddress) {
((AnonymousInProcessSocketAddress) listenAddress).clearServer(this);
} else if (listenAddress instanceof InProcessSocketAddress) {
String name = ((InProcessSocketAddress) listenAddress).getName();
if (!registry.remove(name, this)) {
throw new AssertionError();
}
} else {
throw new AssertionError();
}
}
@Override @Override
public String toString() { public String toString() {
return MoreObjects.toStringHelper(this).add("name", name).toString(); return MoreObjects.toStringHelper(this).add("listenAddress", listenAddress).toString();
} }
synchronized ServerTransportListener register(InProcessTransport transport) { synchronized ServerTransportListener register(InProcessTransport transport) {

View File

@ -34,6 +34,7 @@ import io.grpc.internal.ServerImplBuilder;
import io.grpc.internal.ServerImplBuilder.ClientTransportServersBuilder; import io.grpc.internal.ServerImplBuilder.ClientTransportServersBuilder;
import io.grpc.internal.SharedResourcePool; import io.grpc.internal.SharedResourcePool;
import java.io.File; import java.io.File;
import java.net.SocketAddress;
import java.util.List; import java.util.List;
import java.util.UUID; import java.util.UUID;
import java.util.concurrent.ScheduledExecutorService; import java.util.concurrent.ScheduledExecutorService;
@ -81,7 +82,16 @@ public final class InProcessServerBuilder extends
* @return a new builder * @return a new builder
*/ */
public static InProcessServerBuilder forName(String name) { public static InProcessServerBuilder forName(String name) {
return new InProcessServerBuilder(name); return forAddress(new InProcessSocketAddress(checkNotNull(name, "name")));
}
/**
* Create a server builder which listens on the given address.
* @param listenAddress The SocketAddress this server will listen on.
* @return a new builder
*/
public static InProcessServerBuilder forAddress(SocketAddress listenAddress) {
return new InProcessServerBuilder(listenAddress);
} }
/** /**
@ -100,13 +110,13 @@ public final class InProcessServerBuilder extends
} }
private final ServerImplBuilder serverImplBuilder; private final ServerImplBuilder serverImplBuilder;
final String name; final SocketAddress listenAddress;
int maxInboundMetadataSize = Integer.MAX_VALUE; int maxInboundMetadataSize = Integer.MAX_VALUE;
ObjectPool<ScheduledExecutorService> schedulerPool = ObjectPool<ScheduledExecutorService> schedulerPool =
SharedResourcePool.forResource(GrpcUtil.TIMER_SERVICE); SharedResourcePool.forResource(GrpcUtil.TIMER_SERVICE);
private InProcessServerBuilder(String name) { private InProcessServerBuilder(SocketAddress listenAddress) {
this.name = Preconditions.checkNotNull(name, "name"); this.listenAddress = checkNotNull(listenAddress, "listenAddress");
final class InProcessClientTransportServersBuilder implements ClientTransportServersBuilder { final class InProcessClientTransportServersBuilder implements ClientTransportServersBuilder {
@Override @Override

View File

@ -59,6 +59,7 @@ import io.grpc.internal.ServerTransportListener;
import io.grpc.internal.StatsTraceContext; import io.grpc.internal.StatsTraceContext;
import io.grpc.internal.StreamListener; import io.grpc.internal.StreamListener;
import java.io.InputStream; import java.io.InputStream;
import java.net.SocketAddress;
import java.util.ArrayDeque; import java.util.ArrayDeque;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Collections; import java.util.Collections;
@ -80,7 +81,7 @@ final class InProcessTransport implements ServerTransport, ConnectionClientTrans
private static final Logger log = Logger.getLogger(InProcessTransport.class.getName()); private static final Logger log = Logger.getLogger(InProcessTransport.class.getName());
private final InternalLogId logId; private final InternalLogId logId;
private final String name; private final SocketAddress address;
private final int clientMaxInboundMetadataSize; private final int clientMaxInboundMetadataSize;
private final String authority; private final String authority;
private final String userAgent; private final String userAgent;
@ -119,10 +120,10 @@ final class InProcessTransport implements ServerTransport, ConnectionClientTrans
} }
}; };
private InProcessTransport(String name, int maxInboundMetadataSize, String authority, private InProcessTransport(SocketAddress address, int maxInboundMetadataSize, String authority,
String userAgent, Attributes eagAttrs, String userAgent, Attributes eagAttrs,
Optional<ServerListener> optionalServerListener, boolean includeCauseWithStatus) { Optional<ServerListener> optionalServerListener, boolean includeCauseWithStatus) {
this.name = name; this.address = address;
this.clientMaxInboundMetadataSize = maxInboundMetadataSize; this.clientMaxInboundMetadataSize = maxInboundMetadataSize;
this.authority = authority; this.authority = authority;
this.userAgent = GrpcUtil.getGrpcUserAgent("inprocess", userAgent); this.userAgent = GrpcUtil.getGrpcUserAgent("inprocess", userAgent);
@ -130,18 +131,18 @@ final class InProcessTransport implements ServerTransport, ConnectionClientTrans
this.attributes = Attributes.newBuilder() this.attributes = Attributes.newBuilder()
.set(GrpcAttributes.ATTR_SECURITY_LEVEL, SecurityLevel.PRIVACY_AND_INTEGRITY) .set(GrpcAttributes.ATTR_SECURITY_LEVEL, SecurityLevel.PRIVACY_AND_INTEGRITY)
.set(GrpcAttributes.ATTR_CLIENT_EAG_ATTRS, eagAttrs) .set(GrpcAttributes.ATTR_CLIENT_EAG_ATTRS, eagAttrs)
.set(Grpc.TRANSPORT_ATTR_REMOTE_ADDR, new InProcessSocketAddress(name)) .set(Grpc.TRANSPORT_ATTR_REMOTE_ADDR, address)
.set(Grpc.TRANSPORT_ATTR_LOCAL_ADDR, new InProcessSocketAddress(name)) .set(Grpc.TRANSPORT_ATTR_LOCAL_ADDR, address)
.build(); .build();
this.optionalServerListener = optionalServerListener; this.optionalServerListener = optionalServerListener;
logId = InternalLogId.allocate(getClass(), name); logId = InternalLogId.allocate(getClass(), address.toString());
this.includeCauseWithStatus = includeCauseWithStatus; this.includeCauseWithStatus = includeCauseWithStatus;
} }
public InProcessTransport( public InProcessTransport(
String name, int maxInboundMetadataSize, String authority, String userAgent, SocketAddress address, int maxInboundMetadataSize, String authority, String userAgent,
Attributes eagAttrs, boolean includeCauseWithStatus) { Attributes eagAttrs, boolean includeCauseWithStatus) {
this(name, maxInboundMetadataSize, authority, userAgent, eagAttrs, this(address, maxInboundMetadataSize, authority, userAgent, eagAttrs,
Optional.<ServerListener>absent(), includeCauseWithStatus); Optional.<ServerListener>absent(), includeCauseWithStatus);
} }
@ -150,7 +151,7 @@ final class InProcessTransport implements ServerTransport, ConnectionClientTrans
Attributes eagAttrs, ObjectPool<ScheduledExecutorService> serverSchedulerPool, Attributes eagAttrs, ObjectPool<ScheduledExecutorService> serverSchedulerPool,
List<ServerStreamTracer.Factory> serverStreamTracerFactories, List<ServerStreamTracer.Factory> serverStreamTracerFactories,
ServerListener serverListener) { ServerListener serverListener) {
this(name, maxInboundMetadataSize, authority, userAgent, eagAttrs, this(new InProcessSocketAddress(name), maxInboundMetadataSize, authority, userAgent, eagAttrs,
Optional.of(serverListener), false); Optional.of(serverListener), false);
this.serverMaxInboundMetadataSize = maxInboundMetadataSize; this.serverMaxInboundMetadataSize = maxInboundMetadataSize;
this.serverSchedulerPool = serverSchedulerPool; this.serverSchedulerPool = serverSchedulerPool;
@ -165,7 +166,7 @@ final class InProcessTransport implements ServerTransport, ConnectionClientTrans
serverScheduler = serverSchedulerPool.getObject(); serverScheduler = serverSchedulerPool.getObject();
serverTransportListener = optionalServerListener.get().transportCreated(this); serverTransportListener = optionalServerListener.get().transportCreated(this);
} else { } else {
InProcessServer server = InProcessServer.findServer(name); InProcessServer server = InProcessServer.findServer(address);
if (server != null) { if (server != null) {
serverMaxInboundMetadataSize = server.getMaxInboundMetadataSize(); serverMaxInboundMetadataSize = server.getMaxInboundMetadataSize();
serverSchedulerPool = server.getScheduledExecutorServicePool(); serverSchedulerPool = server.getScheduledExecutorServicePool();
@ -176,7 +177,7 @@ final class InProcessTransport implements ServerTransport, ConnectionClientTrans
} }
} }
if (serverTransportListener == null) { if (serverTransportListener == null) {
shutdownStatus = Status.UNAVAILABLE.withDescription("Could not find server: " + name); shutdownStatus = Status.UNAVAILABLE.withDescription("Could not find server: " + address);
final Status localShutdownStatus = shutdownStatus; final Status localShutdownStatus = shutdownStatus;
return new Runnable() { return new Runnable() {
@Override @Override
@ -194,8 +195,8 @@ final class InProcessTransport implements ServerTransport, ConnectionClientTrans
public void run() { public void run() {
synchronized (InProcessTransport.this) { synchronized (InProcessTransport.this) {
Attributes serverTransportAttrs = Attributes.newBuilder() Attributes serverTransportAttrs = Attributes.newBuilder()
.set(Grpc.TRANSPORT_ATTR_REMOTE_ADDR, new InProcessSocketAddress(name)) .set(Grpc.TRANSPORT_ATTR_REMOTE_ADDR, address)
.set(Grpc.TRANSPORT_ATTR_LOCAL_ADDR, new InProcessSocketAddress(name)) .set(Grpc.TRANSPORT_ATTR_LOCAL_ADDR, address)
.build(); .build();
serverStreamAttributes = serverTransportListener.transportReady(serverTransportAttrs); serverStreamAttributes = serverTransportListener.transportReady(serverTransportAttrs);
clientTransportListener.transportReady(); clientTransportListener.transportReady();
@ -307,7 +308,7 @@ final class InProcessTransport implements ServerTransport, ConnectionClientTrans
public String toString() { public String toString() {
return MoreObjects.toStringHelper(this) return MoreObjects.toStringHelper(this)
.add("logId", logId.getId()) .add("logId", logId.getId())
.add("name", name) .add("address", address)
.toString(); .toString();
} }

View File

@ -0,0 +1,104 @@
/*
* Copyright 2021 The 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.
*/
package io.grpc.inprocess;
import static com.google.common.truth.Truth.assertThat;
import static org.junit.Assert.fail;
import com.google.common.testing.EqualsTester;
import io.grpc.ServerStreamTracer;
import java.io.IOException;
import java.util.Collections;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
/** Unit tests for {@link AnonymousInProcessSocketAddress}. */
@RunWith(JUnit4.class)
public class AnonymousInProcessSocketAddressTest {
@Test
public void defaultState() {
AnonymousInProcessSocketAddress addr = new AnonymousInProcessSocketAddress();
assertThat(addr.getServer()).isNull();
}
@Test
public void setServer() throws Exception {
AnonymousInProcessSocketAddress addr = new AnonymousInProcessSocketAddress();
InProcessServer server = createAnonymousServer();
addr.setServer(server);
assertThat(addr.getServer()).isSameInstanceAs(server);
}
@Test
public void setServerTwice() throws Exception {
AnonymousInProcessSocketAddress addr = new AnonymousInProcessSocketAddress();
InProcessServer server = createAnonymousServer();
addr.setServer(server);
try {
addr.setServer(server);
fail("Expected IOException on attempt to set server twice");
} catch (IOException ioe) {
// Expected.
}
}
@Test
public void clearServer() throws Exception {
AnonymousInProcessSocketAddress addr = new AnonymousInProcessSocketAddress();
InProcessServer server = createAnonymousServer();
addr.setServer(server);
addr.clearServer(server);
assertThat(addr.getServer()).isNull();
}
@Test
public void clearServerWrongInstance() throws Exception {
AnonymousInProcessSocketAddress addr = new AnonymousInProcessSocketAddress();
addr.setServer(createAnonymousServer());
try {
addr.clearServer(createAnonymousServer());
fail("Expected IllegalStateException on attempt to clear the wrong server");
} catch (IllegalStateException ise) {
// Expected.
}
}
@Test
public void equality() throws IOException {
AnonymousInProcessSocketAddress addrA = new AnonymousInProcessSocketAddress();
AnonymousInProcessSocketAddress addrB = new AnonymousInProcessSocketAddress();
AnonymousInProcessSocketAddress addrC = new AnonymousInProcessSocketAddress();
InProcessServer server = createAnonymousServer();
// Ensure two addresses with the same server are still distinct from each other.
addrA.setServer(server);
addrB.setServer(server);
new EqualsTester()
.addEqualityGroup(addrA)
.addEqualityGroup(addrB)
.addEqualityGroup(addrC)
.testEquals();
}
private InProcessServer createAnonymousServer() {
AnonymousInProcessSocketAddress unused = new AnonymousInProcessSocketAddress();
InProcessServerBuilder builder = InProcessServerBuilder.forAddress(unused);
return new InProcessServer(builder, Collections.<ServerStreamTracer.Factory>emptyList());
}
}

View File

@ -0,0 +1,57 @@
/*
* Copyright 2021 The 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.
*/
package io.grpc.inprocess;
import static com.google.common.truth.Truth.assertThat;
import io.grpc.ServerStreamTracer;
import io.grpc.internal.GrpcUtil;
import io.grpc.internal.InternalServer;
import io.grpc.internal.ManagedClientTransport;
import java.util.List;
import org.junit.After;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
/** Unit tests for {@link InProcessTransport} with an anonymous server. */
@RunWith(JUnit4.class)
public final class AnonymousInProcessTransportTest extends InProcessTransportTest {
private AnonymousInProcessSocketAddress address = new AnonymousInProcessSocketAddress();
@After
@Override
public void tearDown() throws InterruptedException {
super.tearDown();
assertThat(address.getServer()).isNull();
}
@Override
protected InternalServer newServer(
List<ServerStreamTracer.Factory> streamTracerFactories) {
InProcessServerBuilder builder = InProcessServerBuilder.forAddress(address)
.maxInboundMetadataSize(GrpcUtil.DEFAULT_MAX_HEADER_LIST_SIZE);
return new InProcessServer(builder, streamTracerFactories);
}
@Override
protected ManagedClientTransport newClientTransport(InternalServer server) {
return new InProcessTransport(
address, GrpcUtil.DEFAULT_MAX_HEADER_LIST_SIZE,
testAuthority(server), USER_AGENT, eagAttrs(), false);
}
}

View File

@ -54,7 +54,7 @@ import org.junit.runners.JUnit4;
public class InProcessTransportTest extends AbstractTransportTest { public class InProcessTransportTest extends AbstractTransportTest {
private static final String TRANSPORT_NAME = "perfect-for-testing"; private static final String TRANSPORT_NAME = "perfect-for-testing";
private static final String AUTHORITY = "a-testing-authority"; private static final String AUTHORITY = "a-testing-authority";
private static final String USER_AGENT = "a-testing-user-agent"; protected static final String USER_AGENT = "a-testing-user-agent";
@Rule @Rule
public final GrpcCleanupRule grpcCleanupRule = new GrpcCleanupRule(); public final GrpcCleanupRule grpcCleanupRule = new GrpcCleanupRule();
@ -82,8 +82,8 @@ public class InProcessTransportTest extends AbstractTransportTest {
@Override @Override
protected ManagedClientTransport newClientTransport(InternalServer server) { protected ManagedClientTransport newClientTransport(InternalServer server) {
return new InProcessTransport( return new InProcessTransport(
TRANSPORT_NAME, GrpcUtil.DEFAULT_MAX_HEADER_LIST_SIZE, testAuthority(server), USER_AGENT, new InProcessSocketAddress(TRANSPORT_NAME), GrpcUtil.DEFAULT_MAX_HEADER_LIST_SIZE,
eagAttrs(), false); testAuthority(server), USER_AGENT, eagAttrs(), false);
} }
@Override @Override