lryan | 56e307f | 2014-12-05 13:25:08 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2014, Google Inc. All rights reserved. |
| 3 | * |
| 4 | * Redistribution and use in source and binary forms, with or without |
| 5 | * modification, are permitted provided that the following conditions are |
| 6 | * met: |
| 7 | * |
| 8 | * * Redistributions of source code must retain the above copyright |
| 9 | * notice, this list of conditions and the following disclaimer. |
| 10 | * * Redistributions in binary form must reproduce the above |
| 11 | * copyright notice, this list of conditions and the following disclaimer |
| 12 | * in the documentation and/or other materials provided with the |
| 13 | * distribution. |
| 14 | * |
| 15 | * * Neither the name of Google Inc. nor the names of its |
| 16 | * contributors may be used to endorse or promote products derived from |
| 17 | * this software without specific prior written permission. |
| 18 | * |
| 19 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| 20 | * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| 21 | * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| 22 | * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| 23 | * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 24 | * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| 25 | * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| 26 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| 27 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 28 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 29 | * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 30 | */ |
| 31 | |
nmittler | b687bdc | 2015-08-31 16:13:39 -0700 | [diff] [blame] | 32 | package io.grpc.internal; |
zhangkun | 8d6d12e | 2014-10-15 13:04:19 -0700 | [diff] [blame] | 33 | |
Carl Mastrangelo | 82a79d8 | 2015-12-07 14:40:11 -0800 | [diff] [blame] | 34 | import static com.google.common.base.MoreObjects.firstNonNull; |
Kun Zhang | 527fd67 | 2016-06-17 09:47:41 -0700 | [diff] [blame] | 35 | import static com.google.common.base.Preconditions.checkArgument; |
Carl Mastrangelo | 82a79d8 | 2015-12-07 14:40:11 -0800 | [diff] [blame] | 36 | |
Kun Zhang | 631a9d5 | 2016-06-02 16:47:36 -0700 | [diff] [blame] | 37 | import com.google.common.annotations.VisibleForTesting; |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 38 | import com.google.common.base.Preconditions; |
buchgr | 602473d | 2015-11-11 12:53:08 +0100 | [diff] [blame] | 39 | import com.google.common.util.concurrent.MoreExecutors; |
Kun Zhang | cce8eac | 2017-01-05 10:48:13 -0800 | [diff] [blame] | 40 | import com.google.instrumentation.stats.Stats; |
| 41 | import com.google.instrumentation.stats.StatsContextFactory; |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 42 | import io.grpc.Attributes; |
| 43 | import io.grpc.ClientInterceptor; |
Carl Mastrangelo | 82a79d8 | 2015-12-07 14:40:11 -0800 | [diff] [blame] | 44 | import io.grpc.CompressorRegistry; |
| 45 | import io.grpc.DecompressorRegistry; |
Kun Zhang | a9bd947 | 2017-02-21 17:11:03 -0800 | [diff] [blame^] | 46 | import io.grpc.LoadBalancer; |
Kun Zhang | d17a7b5 | 2017-01-10 15:30:12 -0800 | [diff] [blame] | 47 | import io.grpc.ManagedChannel; |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 48 | import io.grpc.ManagedChannelBuilder; |
| 49 | import io.grpc.NameResolver; |
Eric Anderson | 56a2938 | 2016-06-11 11:55:49 -0700 | [diff] [blame] | 50 | import io.grpc.NameResolverProvider; |
Kun Zhang | a9bd947 | 2017-02-21 17:11:03 -0800 | [diff] [blame^] | 51 | import io.grpc.PickFirstBalancerFactory; |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 52 | import io.grpc.ResolvedServerInfo; |
Lukasz Strzalkowski | 23652c5 | 2016-08-19 16:56:56 +0200 | [diff] [blame] | 53 | import io.grpc.ResolvedServerInfoGroup; |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 54 | import java.net.SocketAddress; |
| 55 | import java.net.URI; |
Kun Zhang | 631a9d5 | 2016-06-02 16:47:36 -0700 | [diff] [blame] | 56 | import java.net.URISyntaxException; |
Eric Anderson | 0df3d5e | 2015-06-25 18:00:00 -0700 | [diff] [blame] | 57 | import java.util.ArrayList; |
| 58 | import java.util.Arrays; |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 59 | import java.util.Collections; |
Eric Anderson | 0df3d5e | 2015-06-25 18:00:00 -0700 | [diff] [blame] | 60 | import java.util.List; |
Louis Ryan | 6a782a0 | 2015-09-03 14:27:48 -0700 | [diff] [blame] | 61 | import java.util.concurrent.Executor; |
Kun Zhang | 527fd67 | 2016-06-17 09:47:41 -0700 | [diff] [blame] | 62 | import java.util.concurrent.TimeUnit; |
zhangkun | 8d6d12e | 2014-10-15 13:04:19 -0700 | [diff] [blame] | 63 | import javax.annotation.Nullable; |
| 64 | |
| 65 | /** |
| 66 | * The base class for channel builders. |
nathanmittler | 0304b3d | 2014-10-24 13:39:13 -0700 | [diff] [blame] | 67 | * |
nmittler | b687bdc | 2015-08-31 16:13:39 -0700 | [diff] [blame] | 68 | * @param <T> The concrete type of this builder. |
zhangkun | 8d6d12e | 2014-10-15 13:04:19 -0700 | [diff] [blame] | 69 | */ |
nmittler | b687bdc | 2015-08-31 16:13:39 -0700 | [diff] [blame] | 70 | public abstract class AbstractManagedChannelImplBuilder |
| 71 | <T extends AbstractManagedChannelImplBuilder<T>> extends ManagedChannelBuilder<T> { |
Kun Zhang | 1624715 | 2015-12-08 17:52:05 -0800 | [diff] [blame] | 72 | private static final String DIRECT_ADDRESS_SCHEME = "directaddress"; |
Eric Anderson | aeeebb7 | 2014-12-19 16:41:03 -0800 | [diff] [blame] | 73 | |
Kun Zhang | 527fd67 | 2016-06-17 09:47:41 -0700 | [diff] [blame] | 74 | /** |
| 75 | * An idle timeout larger than this would disable idle mode. |
| 76 | */ |
| 77 | @VisibleForTesting |
| 78 | static final long IDLE_MODE_MAX_TIMEOUT_DAYS = 30; |
| 79 | |
| 80 | /** |
Kun Zhang | d74091f | 2016-09-16 22:34:27 -0700 | [diff] [blame] | 81 | * The default idle timeout. |
| 82 | */ |
| 83 | @VisibleForTesting |
| 84 | static final long IDLE_MODE_DEFAULT_TIMEOUT_MILLIS = TimeUnit.MINUTES.toMillis(30); |
| 85 | |
| 86 | /** |
Kun Zhang | 527fd67 | 2016-06-17 09:47:41 -0700 | [diff] [blame] | 87 | * An idle timeout smaller than this would be capped to it. |
| 88 | */ |
| 89 | @VisibleForTesting |
| 90 | static final long IDLE_MODE_MIN_TIMEOUT_MILLIS = TimeUnit.SECONDS.toMillis(1); |
| 91 | |
ejona | 7235a39 | 2015-01-13 13:38:54 -0800 | [diff] [blame] | 92 | @Nullable |
Louis Ryan | 6a782a0 | 2015-09-03 14:27:48 -0700 | [diff] [blame] | 93 | private Executor executor; |
Kun Zhang | d17a7b5 | 2017-01-10 15:30:12 -0800 | [diff] [blame] | 94 | |
Eric Anderson | 0df3d5e | 2015-06-25 18:00:00 -0700 | [diff] [blame] | 95 | private final List<ClientInterceptor> interceptors = new ArrayList<ClientInterceptor>(); |
zhangkun | 8d6d12e | 2014-10-15 13:04:19 -0700 | [diff] [blame] | 96 | |
Kun Zhang | efac679 | 2015-10-22 14:59:44 -0700 | [diff] [blame] | 97 | private final String target; |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 98 | |
| 99 | @Nullable |
| 100 | private final SocketAddress directServerAddress; |
| 101 | |
nmittler | 8c1d38a | 2015-06-01 08:31:00 -0700 | [diff] [blame] | 102 | @Nullable |
| 103 | private String userAgent; |
| 104 | |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 105 | @Nullable |
| 106 | private String authorityOverride; |
| 107 | |
| 108 | @Nullable |
| 109 | private NameResolver.Factory nameResolverFactory; |
| 110 | |
Kun Zhang | a9bd947 | 2017-02-21 17:11:03 -0800 | [diff] [blame^] | 111 | private LoadBalancer.Factory loadBalancerFactory; |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 112 | |
Carl Mastrangelo | 82a79d8 | 2015-12-07 14:40:11 -0800 | [diff] [blame] | 113 | @Nullable |
| 114 | private DecompressorRegistry decompressorRegistry; |
| 115 | |
| 116 | @Nullable |
| 117 | private CompressorRegistry compressorRegistry; |
| 118 | |
Kun Zhang | d74091f | 2016-09-16 22:34:27 -0700 | [diff] [blame] | 119 | private long idleTimeoutMillis = IDLE_MODE_DEFAULT_TIMEOUT_MILLIS; |
Kun Zhang | 527fd67 | 2016-06-17 09:47:41 -0700 | [diff] [blame] | 120 | |
Carl Mastrangelo | fdd062c | 2016-10-26 16:55:41 -0700 | [diff] [blame] | 121 | private int maxInboundMessageSize = GrpcUtil.DEFAULT_MAX_MESSAGE_SIZE; |
| 122 | |
| 123 | // Can be overriden by subclasses. |
| 124 | @Override |
| 125 | public T maxInboundMessageSize(int max) { |
| 126 | checkArgument(max >= 0, "negative max"); |
| 127 | maxInboundMessageSize = max; |
| 128 | return thisT(); |
| 129 | } |
| 130 | |
| 131 | protected final int maxInboundMessageSize() { |
| 132 | return maxInboundMessageSize; |
| 133 | } |
| 134 | |
Kun Zhang | 132f7a9 | 2016-10-06 17:15:24 -0700 | [diff] [blame] | 135 | @Nullable |
Kun Zhang | cce8eac | 2017-01-05 10:48:13 -0800 | [diff] [blame] | 136 | private StatsContextFactory statsFactory; |
Kun Zhang | 132f7a9 | 2016-10-06 17:15:24 -0700 | [diff] [blame] | 137 | |
Kun Zhang | efac679 | 2015-10-22 14:59:44 -0700 | [diff] [blame] | 138 | protected AbstractManagedChannelImplBuilder(String target) { |
Carl Mastrangelo | 1285477 | 2016-08-12 14:52:00 -0700 | [diff] [blame] | 139 | this.target = Preconditions.checkNotNull(target, "target"); |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 140 | this.directServerAddress = null; |
| 141 | } |
| 142 | |
Kun Zhang | 631a9d5 | 2016-06-02 16:47:36 -0700 | [diff] [blame] | 143 | /** |
| 144 | * Returns a target string for the SocketAddress. It is only used as a placeholder, because |
| 145 | * DirectAddressNameResolverFactory will not actually try to use it. However, it must be a valid |
| 146 | * URI. |
| 147 | */ |
| 148 | @VisibleForTesting |
| 149 | static String makeTargetStringForDirectAddress(SocketAddress address) { |
| 150 | try { |
| 151 | return new URI(DIRECT_ADDRESS_SCHEME, "", "/" + address, null).toString(); |
| 152 | } catch (URISyntaxException e) { |
| 153 | // It should not happen. |
| 154 | throw new RuntimeException(e); |
| 155 | } |
| 156 | } |
| 157 | |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 158 | protected AbstractManagedChannelImplBuilder(SocketAddress directServerAddress, String authority) { |
Kun Zhang | 631a9d5 | 2016-06-02 16:47:36 -0700 | [diff] [blame] | 159 | this.target = makeTargetStringForDirectAddress(directServerAddress); |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 160 | this.directServerAddress = directServerAddress; |
| 161 | this.nameResolverFactory = new DirectAddressNameResolverFactory(directServerAddress, authority); |
| 162 | } |
| 163 | |
Eric Anderson | 6122daf | 2015-09-03 12:14:30 -0700 | [diff] [blame] | 164 | @Override |
buchgr | 602473d | 2015-11-11 12:53:08 +0100 | [diff] [blame] | 165 | public final T directExecutor() { |
| 166 | return executor(MoreExecutors.directExecutor()); |
| 167 | } |
| 168 | |
| 169 | @Override |
Louis Ryan | 6a782a0 | 2015-09-03 14:27:48 -0700 | [diff] [blame] | 170 | public final T executor(Executor executor) { |
nmittler | 777e928 | 2015-08-19 10:01:52 -0700 | [diff] [blame] | 171 | this.executor = executor; |
Eric Anderson | 0df3d5e | 2015-06-25 18:00:00 -0700 | [diff] [blame] | 172 | return thisT(); |
| 173 | } |
| 174 | |
Eric Anderson | 6122daf | 2015-09-03 12:14:30 -0700 | [diff] [blame] | 175 | @Override |
nmittler | b687bdc | 2015-08-31 16:13:39 -0700 | [diff] [blame] | 176 | public final T intercept(List<ClientInterceptor> interceptors) { |
Eric Anderson | 0df3d5e | 2015-06-25 18:00:00 -0700 | [diff] [blame] | 177 | this.interceptors.addAll(interceptors); |
| 178 | return thisT(); |
| 179 | } |
| 180 | |
Eric Anderson | 6122daf | 2015-09-03 12:14:30 -0700 | [diff] [blame] | 181 | @Override |
nmittler | b687bdc | 2015-08-31 16:13:39 -0700 | [diff] [blame] | 182 | public final T intercept(ClientInterceptor... interceptors) { |
Eric Anderson | 0df3d5e | 2015-06-25 18:00:00 -0700 | [diff] [blame] | 183 | return intercept(Arrays.asList(interceptors)); |
| 184 | } |
| 185 | |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 186 | @Override |
| 187 | public final T nameResolverFactory(NameResolver.Factory resolverFactory) { |
| 188 | Preconditions.checkState(directServerAddress == null, |
| 189 | "directServerAddress is set (%s), which forbids the use of NameResolverFactory", |
| 190 | directServerAddress); |
| 191 | this.nameResolverFactory = resolverFactory; |
| 192 | return thisT(); |
| 193 | } |
| 194 | |
| 195 | @Override |
Kun Zhang | a9bd947 | 2017-02-21 17:11:03 -0800 | [diff] [blame^] | 196 | public final T loadBalancerFactory(LoadBalancer.Factory loadBalancerFactory) { |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 197 | Preconditions.checkState(directServerAddress == null, |
Kun Zhang | a9bd947 | 2017-02-21 17:11:03 -0800 | [diff] [blame^] | 198 | "directServerAddress is set (%s), which forbids the use of LoadBalancer.Factory", |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 199 | directServerAddress); |
| 200 | this.loadBalancerFactory = loadBalancerFactory; |
Kun Zhang | d17a7b5 | 2017-01-10 15:30:12 -0800 | [diff] [blame] | 201 | return thisT(); |
| 202 | } |
| 203 | |
Carl Mastrangelo | 82a79d8 | 2015-12-07 14:40:11 -0800 | [diff] [blame] | 204 | @Override |
Carl Mastrangelo | 82a79d8 | 2015-12-07 14:40:11 -0800 | [diff] [blame] | 205 | public final T decompressorRegistry(DecompressorRegistry registry) { |
| 206 | this.decompressorRegistry = registry; |
| 207 | return thisT(); |
| 208 | } |
| 209 | |
| 210 | @Override |
Carl Mastrangelo | 82a79d8 | 2015-12-07 14:40:11 -0800 | [diff] [blame] | 211 | public final T compressorRegistry(CompressorRegistry registry) { |
| 212 | this.compressorRegistry = registry; |
| 213 | return thisT(); |
| 214 | } |
| 215 | |
Eric Anderson | 6122daf | 2015-09-03 12:14:30 -0700 | [diff] [blame] | 216 | @Override |
Carl Mastrangelo | 1cc76d8 | 2016-05-24 16:29:26 -0700 | [diff] [blame] | 217 | public final T userAgent(@Nullable String userAgent) { |
nmittler | 8c1d38a | 2015-06-01 08:31:00 -0700 | [diff] [blame] | 218 | this.userAgent = userAgent; |
nmittler | b687bdc | 2015-08-31 16:13:39 -0700 | [diff] [blame] | 219 | return thisT(); |
nmittler | 8c1d38a | 2015-06-01 08:31:00 -0700 | [diff] [blame] | 220 | } |
| 221 | |
Eric Anderson | 6122daf | 2015-09-03 12:14:30 -0700 | [diff] [blame] | 222 | @Override |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 223 | public final T overrideAuthority(String authority) { |
| 224 | this.authorityOverride = checkAuthority(authority); |
| 225 | return thisT(); |
| 226 | } |
| 227 | |
Kun Zhang | 527fd67 | 2016-06-17 09:47:41 -0700 | [diff] [blame] | 228 | @Override |
| 229 | public final T idleTimeout(long value, TimeUnit unit) { |
| 230 | checkArgument(value > 0, "idle timeout is %s, but must be positive", value); |
| 231 | // We convert to the largest unit to avoid overflow |
| 232 | if (unit.toDays(value) >= IDLE_MODE_MAX_TIMEOUT_DAYS) { |
| 233 | // This disables idle mode |
Kun Zhang | a9bd947 | 2017-02-21 17:11:03 -0800 | [diff] [blame^] | 234 | this.idleTimeoutMillis = ManagedChannelImpl.IDLE_TIMEOUT_MILLIS_DISABLE; |
Kun Zhang | 527fd67 | 2016-06-17 09:47:41 -0700 | [diff] [blame] | 235 | } else { |
| 236 | this.idleTimeoutMillis = Math.max(unit.toMillis(value), IDLE_MODE_MIN_TIMEOUT_MILLIS); |
| 237 | } |
| 238 | return thisT(); |
| 239 | } |
| 240 | |
Kun Zhang | 132f7a9 | 2016-10-06 17:15:24 -0700 | [diff] [blame] | 241 | /** |
Kun Zhang | 737cd16 | 2017-01-20 17:20:44 -0800 | [diff] [blame] | 242 | * Override the default stats implementation. |
Kun Zhang | 132f7a9 | 2016-10-06 17:15:24 -0700 | [diff] [blame] | 243 | */ |
| 244 | @VisibleForTesting |
Kun Zhang | 737cd16 | 2017-01-20 17:20:44 -0800 | [diff] [blame] | 245 | protected T statsContextFactory(StatsContextFactory statsFactory) { |
Kun Zhang | cce8eac | 2017-01-05 10:48:13 -0800 | [diff] [blame] | 246 | this.statsFactory = statsFactory; |
Kun Zhang | 132f7a9 | 2016-10-06 17:15:24 -0700 | [diff] [blame] | 247 | return thisT(); |
| 248 | } |
| 249 | |
Kun Zhang | 527fd67 | 2016-06-17 09:47:41 -0700 | [diff] [blame] | 250 | @VisibleForTesting |
| 251 | final long getIdleTimeoutMillis() { |
| 252 | return idleTimeoutMillis; |
| 253 | } |
| 254 | |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 255 | /** |
| 256 | * Verifies the authority is valid. This method exists as an escape hatch for putting in an |
| 257 | * authority that is valid, but would fail the default validation provided by this |
| 258 | * implementation. |
| 259 | */ |
| 260 | protected String checkAuthority(String authority) { |
| 261 | return GrpcUtil.checkAuthority(authority); |
| 262 | } |
| 263 | |
| 264 | @Override |
Kun Zhang | d17a7b5 | 2017-01-10 15:30:12 -0800 | [diff] [blame] | 265 | public ManagedChannel build() { |
Sky Ao | 1d8aefa | 2016-04-18 23:43:30 +0800 | [diff] [blame] | 266 | ClientTransportFactory transportFactory = buildTransportFactory(); |
| 267 | if (authorityOverride != null) { |
| 268 | transportFactory = new AuthorityOverridingTransportFactory( |
| 269 | transportFactory, authorityOverride); |
| 270 | } |
Eric Anderson | bfcb04c | 2016-07-11 10:02:23 -0700 | [diff] [blame] | 271 | NameResolver.Factory nameResolverFactory = this.nameResolverFactory; |
| 272 | if (nameResolverFactory == null) { |
| 273 | // Avoid loading the provider unless necessary, as a way to workaround a possibly-costly |
| 274 | // and poorly optimized getResource() call on Android. If any other piece of code calls |
| 275 | // getResource(), then this shouldn't be a problem unless called on the UI thread. |
| 276 | nameResolverFactory = NameResolverProvider.asFactory(); |
| 277 | } |
Kun Zhang | a9bd947 | 2017-02-21 17:11:03 -0800 | [diff] [blame^] | 278 | return new ManagedChannelImpl( |
Kun Zhang | a6b1d89 | 2017-02-21 14:26:23 -0800 | [diff] [blame] | 279 | target, |
| 280 | // TODO(carl-mastrangelo): Allow clients to pass this in |
| 281 | new ExponentialBackoffPolicy.Provider(), |
| 282 | nameResolverFactory, |
| 283 | getNameResolverParams(), |
Kun Zhang | a9bd947 | 2017-02-21 17:11:03 -0800 | [diff] [blame^] | 284 | firstNonNull(loadBalancerFactory, PickFirstBalancerFactory.getInstance()), |
Kun Zhang | a6b1d89 | 2017-02-21 14:26:23 -0800 | [diff] [blame] | 285 | transportFactory, |
| 286 | firstNonNull(decompressorRegistry, DecompressorRegistry.getDefaultInstance()), |
| 287 | firstNonNull(compressorRegistry, CompressorRegistry.getDefaultInstance()), |
| 288 | SharedResourcePool.forResource(GrpcUtil.TIMER_SERVICE), |
| 289 | getExecutorPool(executor), |
| 290 | SharedResourcePool.forResource(GrpcUtil.SHARED_CHANNEL_EXECUTOR), |
Kun Zhang | a9bd947 | 2017-02-21 17:11:03 -0800 | [diff] [blame^] | 291 | GrpcUtil.STOPWATCH_SUPPLIER, |
| 292 | idleTimeoutMillis, |
| 293 | userAgent, |
| 294 | interceptors, |
| 295 | firstNonNull( |
| 296 | statsFactory, |
Kun Zhang | a6b1d89 | 2017-02-21 14:26:23 -0800 | [diff] [blame] | 297 | firstNonNull(Stats.getStatsContextFactory(), NoopStatsContextFactory.INSTANCE))); |
zhangkun | 8d6d12e | 2014-10-15 13:04:19 -0700 | [diff] [blame] | 298 | } |
| 299 | |
| 300 | /** |
Kun Zhang | edd57c9 | 2015-10-27 12:47:29 -0700 | [diff] [blame] | 301 | * Subclasses should override this method to provide the {@link ClientTransportFactory} |
| 302 | * appropriate for this channel. This method is meant for Transport implementors and should not |
| 303 | * be used by normal users. |
zhangkun | 8d6d12e | 2014-10-15 13:04:19 -0700 | [diff] [blame] | 304 | */ |
nmittler | 777e928 | 2015-08-19 10:01:52 -0700 | [diff] [blame] | 305 | protected abstract ClientTransportFactory buildTransportFactory(); |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 306 | |
Kun Zhang | edd57c9 | 2015-10-27 12:47:29 -0700 | [diff] [blame] | 307 | /** |
| 308 | * Subclasses can override this method to provide additional parameters to {@link |
| 309 | * NameResolver.Factory#newNameResolver}. The default implementation returns {@link |
Carl Mastrangelo | 82a79d8 | 2015-12-07 14:40:11 -0800 | [diff] [blame] | 310 | * Attributes#EMPTY}. |
Kun Zhang | edd57c9 | 2015-10-27 12:47:29 -0700 | [diff] [blame] | 311 | */ |
| 312 | protected Attributes getNameResolverParams() { |
| 313 | return Attributes.EMPTY; |
| 314 | } |
| 315 | |
Kun Zhang | d17a7b5 | 2017-01-10 15:30:12 -0800 | [diff] [blame] | 316 | private static ObjectPool<? extends Executor> getExecutorPool(final @Nullable Executor executor) { |
| 317 | if (executor != null) { |
| 318 | return new ObjectPool<Executor>() { |
| 319 | @Override |
| 320 | public Executor getObject() { |
| 321 | return executor; |
| 322 | } |
| 323 | |
| 324 | @Override |
| 325 | public Executor returnObject(Object returned) { |
| 326 | return null; |
| 327 | } |
| 328 | }; |
| 329 | } else { |
| 330 | return SharedResourcePool.forResource(GrpcUtil.SHARED_CHANNEL_EXECUTOR); |
| 331 | } |
| 332 | } |
| 333 | |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 334 | private static class AuthorityOverridingTransportFactory implements ClientTransportFactory { |
| 335 | final ClientTransportFactory factory; |
Sky Ao | 1d8aefa | 2016-04-18 23:43:30 +0800 | [diff] [blame] | 336 | final String authorityOverride; |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 337 | |
| 338 | AuthorityOverridingTransportFactory( |
Sky Ao | 1d8aefa | 2016-04-18 23:43:30 +0800 | [diff] [blame] | 339 | ClientTransportFactory factory, String authorityOverride) { |
| 340 | this.factory = Preconditions.checkNotNull(factory, "factory should not be null"); |
| 341 | this.authorityOverride = Preconditions.checkNotNull( |
| 342 | authorityOverride, "authorityOverride should not be null"); |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 343 | } |
| 344 | |
| 345 | @Override |
Kun Zhang | 432cec7 | 2016-05-29 14:43:10 -0700 | [diff] [blame] | 346 | public ConnectionClientTransport newClientTransport(SocketAddress serverAddress, |
Carl Mastrangelo | 1cc76d8 | 2016-05-24 16:29:26 -0700 | [diff] [blame] | 347 | String authority, @Nullable String userAgent) { |
| 348 | return factory.newClientTransport(serverAddress, authorityOverride, userAgent); |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 349 | } |
| 350 | |
| 351 | @Override |
buchgr | 3c68c05 | 2016-03-17 00:14:43 +0100 | [diff] [blame] | 352 | public void close() { |
| 353 | factory.close(); |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 354 | } |
| 355 | } |
| 356 | |
| 357 | private static class DirectAddressNameResolverFactory extends NameResolver.Factory { |
| 358 | final SocketAddress address; |
| 359 | final String authority; |
| 360 | |
| 361 | DirectAddressNameResolverFactory(SocketAddress address, String authority) { |
| 362 | this.address = address; |
| 363 | this.authority = authority; |
| 364 | } |
| 365 | |
| 366 | @Override |
Kun Zhang | edd57c9 | 2015-10-27 12:47:29 -0700 | [diff] [blame] | 367 | public NameResolver newNameResolver(URI notUsedUri, Attributes params) { |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 368 | return new NameResolver() { |
| 369 | @Override |
| 370 | public String getServiceAuthority() { |
| 371 | return authority; |
| 372 | } |
| 373 | |
| 374 | @Override |
| 375 | public void start(final Listener listener) { |
Lukasz Strzalkowski | 23652c5 | 2016-08-19 16:56:56 +0200 | [diff] [blame] | 376 | listener.onUpdate(Collections.singletonList( |
| 377 | ResolvedServerInfoGroup.builder().add(new ResolvedServerInfo(address)).build()), |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 378 | Attributes.EMPTY); |
| 379 | } |
| 380 | |
| 381 | @Override |
| 382 | public void shutdown() {} |
| 383 | }; |
| 384 | } |
Kun Zhang | 1624715 | 2015-12-08 17:52:05 -0800 | [diff] [blame] | 385 | |
| 386 | @Override |
| 387 | public String getDefaultScheme() { |
| 388 | return DIRECT_ADDRESS_SCHEME; |
| 389 | } |
Kun Zhang | 942f4c9 | 2015-09-04 17:21:44 -0700 | [diff] [blame] | 390 | } |
Carl Mastrangelo | 8493498 | 2017-01-11 12:46:47 -0800 | [diff] [blame] | 391 | |
| 392 | /** |
| 393 | * Returns the correctly typed version of the builder. |
| 394 | */ |
| 395 | private T thisT() { |
| 396 | @SuppressWarnings("unchecked") |
| 397 | T thisT = (T) this; |
| 398 | return thisT; |
| 399 | } |
zhangkun | 8d6d12e | 2014-10-15 13:04:19 -0700 | [diff] [blame] | 400 | } |