Java tutorial
/** * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you 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 org.apache.drill.exec.rpc.control; import io.netty.buffer.ByteBuf; import io.netty.channel.socket.SocketChannel; import java.util.UUID; import org.apache.drill.exec.memory.BufferAllocator; import org.apache.drill.exec.proto.BitControl.RpcType; import org.apache.drill.exec.proto.CoordinationProtos.DrillbitEndpoint; import org.apache.drill.exec.rpc.RemoteConnection; import org.apache.drill.exec.rpc.RpcBus; import org.apache.drill.exec.rpc.RpcOutcomeListener; import com.google.common.io.Closeables; import com.google.protobuf.MessageLite; public class ControlConnection extends RemoteConnection { static final org.slf4j.Logger logger = org.slf4j.LoggerFactory.getLogger(ControlConnection.class); private final RpcBus<RpcType, ControlConnection> bus; private final BufferAllocator allocator; private volatile DrillbitEndpoint endpoint; private volatile boolean active = false; private final UUID id; public ControlConnection(String name, SocketChannel channel, RpcBus<RpcType, ControlConnection> bus, BufferAllocator allocator) { super(channel, name); this.bus = bus; this.id = UUID.randomUUID(); this.allocator = allocator; } void setEndpoint(DrillbitEndpoint endpoint) { assert this.endpoint == null : "Endpoint should only be set once (only in the case in incoming server requests)."; this.endpoint = endpoint; active = true; } protected DrillbitEndpoint getEndpoint() { return endpoint; } public <SEND extends MessageLite, RECEIVE extends MessageLite> void send( RpcOutcomeListener<RECEIVE> outcomeListener, RpcType rpcType, SEND protobufBody, Class<RECEIVE> clazz, ByteBuf... dataBodies) { bus.send(outcomeListener, this, rpcType, protobufBody, clazz, dataBodies); } public <SEND extends MessageLite, RECEIVE extends MessageLite> void sendUnsafe( RpcOutcomeListener<RECEIVE> outcomeListener, RpcType rpcType, SEND protobufBody, Class<RECEIVE> clazz, ByteBuf... dataBodies) { bus.send(outcomeListener, this, rpcType, protobufBody, clazz, true, dataBodies); } public void disable() { active = false; } @Override public boolean isActive() { return active; } @Override public int hashCode() { final int prime = 31; int result = 1; result = prime * result + ((id == null) ? 0 : id.hashCode()); return result; } @Override public boolean equals(Object obj) { if (this == obj) { return true; } if (obj == null) { return false; } if (getClass() != obj.getClass()) { return false; } ControlConnection other = (ControlConnection) obj; if (id == null) { if (other.id != null) { return false; } } else if (!id.equals(other.id)) { return false; } return true; } public void shutdownIfClient() { if (bus.isClient()) { Closeables.closeQuietly(bus); } } @Override public BufferAllocator getAllocator() { return allocator; } }