Java tutorial
// Copyright 2018 The Bazel Authors. All rights reserved. // // 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 com.google.devtools.build.lib.remote; import io.grpc.CallOptions; import io.grpc.ClientCall; import io.grpc.ManagedChannel; import io.grpc.MethodDescriptor; import io.netty.util.AbstractReferenceCounted; import io.netty.util.ReferenceCounted; import java.util.concurrent.TimeUnit; /** A wrapper around a {@link io.grpc.ManagedChannel} exposing a reference count. * When instantiated the reference count is 1. {@link ManagedChannel#shutdown()} will be called * on the wrapped channel when the reference count reaches 0. * * See {@link ReferenceCounted} for more information about reference counting. */ class ReferenceCountedChannel extends ManagedChannel implements ReferenceCounted { private final ManagedChannel channel; private final AbstractReferenceCounted referenceCounted = new AbstractReferenceCounted() { @Override protected void deallocate() { channel.shutdown(); } @Override public ReferenceCounted touch(Object o) { return this; } }; public ReferenceCountedChannel(ManagedChannel channel) { this.channel = channel; } @Override public ManagedChannel shutdown() { throw new UnsupportedOperationException("Don't call shutdown() directly, but use release() " + "instead."); } @Override public boolean isShutdown() { return channel.isShutdown(); } @Override public boolean isTerminated() { return channel.isTerminated(); } @Override public ManagedChannel shutdownNow() { throw new UnsupportedOperationException( "Don't call shutdownNow() directly, but use release() " + "instead."); } @Override public boolean awaitTermination(long l, TimeUnit timeUnit) throws InterruptedException { return channel.awaitTermination(l, timeUnit); } @Override public <RequestT, ResponseT> ClientCall<RequestT, ResponseT> newCall( MethodDescriptor<RequestT, ResponseT> methodDescriptor, CallOptions callOptions) { return channel.<RequestT, ResponseT>newCall(methodDescriptor, callOptions); } @Override public String authority() { return channel.authority(); } @Override public int refCnt() { return referenceCounted.refCnt(); } @Override public ReferenceCountedChannel retain() { referenceCounted.retain(); return this; } @Override public ReferenceCountedChannel retain(int increment) { referenceCounted.retain(increment); return this; } @Override public ReferenceCounted touch() { referenceCounted.touch(); return this; } @Override public ReferenceCounted touch(Object hint) { referenceCounted.touch(hint); return this; } @Override public boolean release() { return referenceCounted.release(); } @Override public boolean release(int decrement) { return referenceCounted.release(decrement); } }