Skip to content
This repository was archived by the owner on Apr 7, 2026. It is now read-only.
Prev Previous commit
Next Next commit
Moved InstanceConfigRpcCache to v1 package per comment
Changed to package private
  • Loading branch information
mbril committed Jan 22, 2020
commit cf0d792e95a14c3422d3e68a868564dbb0cbfa55
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,6 @@
import com.google.api.gax.rpc.WatchdogProvider;
import com.google.api.pathtemplate.PathTemplate;
import com.google.cloud.grpc.GrpcTransportOptions;
import com.google.cloud.spanner.InstanceConfigRpcCache;
import com.google.cloud.spanner.SpannerException;
import com.google.cloud.spanner.SpannerExceptionFactory;
import com.google.cloud.spanner.SpannerOptions;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,10 +14,12 @@
* limitations under the License.
*/

package com.google.cloud.spanner;
package com.google.cloud.spanner.spi.v1;

import com.google.cloud.spanner.spi.v1.GapicSpannerRpc;
import com.google.cloud.spanner.spi.v1.SpannerRpc;
import com.google.cloud.spanner.ErrorCode;
import com.google.cloud.spanner.SpannerException;
import com.google.cloud.spanner.SpannerExceptionFactory;
import com.google.cloud.spanner.SpannerOptions;
import com.google.common.cache.CacheBuilder;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
Expand All @@ -31,14 +33,14 @@
import java.util.logging.Level;
import java.util.logging.Logger;

public class InstanceConfigRpcCache {
class InstanceConfigRpcCache {

private final LoadingCache<InstanceName, SpannerRpc> cache;

private Logger logger = Logger.getLogger(InstanceConfigRpcCache.class.getName());
private final GapicSpannerRpc projectClient;

public InstanceConfigRpcCache(final GapicSpannerRpc projectClient) {
InstanceConfigRpcCache(final GapicSpannerRpc projectClient) {
this.projectClient = projectClient;
cache =
CacheBuilder.newBuilder()
Expand Down Expand Up @@ -85,13 +87,13 @@ public SpannerRpc load(InstanceName instanceName) throws SpannerException {
});
}

public SpannerRpc get(SessionName sessionName) {
SpannerRpc get(SessionName sessionName) {
InstanceName instanceName =
InstanceName.of(sessionName.getProject(), sessionName.getInstance());
return get(instanceName);
}

public SpannerRpc get(DatabaseName databaseName) {
SpannerRpc get(DatabaseName databaseName) {
InstanceName instanceName =
InstanceName.of(databaseName.getProject(), databaseName.getInstance());
return get(instanceName);
Expand All @@ -111,7 +113,7 @@ private SpannerRpc get(InstanceName instanceName) {
}
}

public void invalidateAll() {
void invalidateAll() {
for (SpannerRpc rpc : cache.asMap().values()) {
if (rpc == this.projectClient) continue;
rpc.shutdown();
Expand Down