Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright (c) 2015 Uber Technologies, Inc.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
package com.uber.tchannel.api;
import com.uber.tchannel.api.errors.TChannelConnectionTimeout;
import com.uber.tchannel.api.errors.TChannelError;
import com.uber.tchannel.api.errors.TChannelNoPeerAvailable;
import com.uber.tchannel.api.handlers.HealthCheckRequestHandler;
import com.uber.tchannel.api.handlers.RequestHandler;
import com.uber.tchannel.channels.Connection;
import com.uber.tchannel.channels.PeerManager;
import com.uber.tchannel.channels.SubPeer;
import com.uber.tchannel.errors.ErrorType;
import com.uber.tchannel.handlers.OutRequest;
import com.uber.tchannel.handlers.ResponseRouter;
import com.uber.tchannel.headers.ArgScheme;
import com.uber.tchannel.headers.TransportHeaders;
import com.uber.tchannel.messages.JSONSerializer;
import com.uber.tchannel.messages.JsonRequest;
import com.uber.tchannel.messages.JsonResponse;
import com.uber.tchannel.messages.RawRequest;
import com.uber.tchannel.messages.RawResponse;
import com.uber.tchannel.messages.Request;
import com.uber.tchannel.messages.Response;
import com.uber.tchannel.messages.Serializer;
import com.uber.tchannel.messages.ThriftRequest;
import com.uber.tchannel.messages.ThriftResponse;
import com.uber.tchannel.messages.ThriftSerializer;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Random;
public final class SubChannel {
private final String service;
private final TChannel topChannel;
private final PeerManager peerManager;
private final long initTimeout;
private final Connection.Direction preferredDirection;
private final List peers = new ArrayList<>();
private final Map requestHandlers = new HashMap<>();
private final Serializer serializer = new Serializer(new HashMap() {
{
put(ArgScheme.JSON, new JSONSerializer());
put(ArgScheme.THRIFT, new ThriftSerializer());
}
});
public SubChannel(String service, TChannel topChannel) {
this.service = service;
this.topChannel = topChannel;
this.peerManager = topChannel.getPeerManager();
this.initTimeout = topChannel.getInitTimeout();
this.preferredDirection = Connection.Direction.NONE;
}
public SubChannel(String service, TChannel topChannel, Connection.Direction preferredDirection) {
this.service = service;
this.topChannel = topChannel;
this.peerManager = topChannel.getPeerManager();
this.initTimeout = topChannel.getInitTimeout();
this.preferredDirection = preferredDirection;
}
public String getServiceName() {
return this.service;
}
public PeerManager getPeerManager() {
return this.peerManager;
}
public SubChannel register(String endpoint, RequestHandler requestHandler) {
requestHandlers.put(endpoint, requestHandler);
return this;
}
public SubChannel registerHealthHanlder() {
return registerHealthHandler(new HealthCheckRequestHandler());
}
public SubChannel registerHealthHandler(HealthCheckRequestHandler healthHandler) {
return register("Meta::health", healthHandler);
}
public RequestHandler getRequestHandler(String endpoint) {
return requestHandlers.get(endpoint);
}
public Connection.Direction getPreferredDirection() {
return preferredDirection;
}
public SubChannel setPeers(List peers) {
for (InetSocketAddress peer : peers) {
this.peers.add(new SubPeer(peer, this));
}
return this;
}
private final Random random = new Random();
public SubPeer choosePeer(OutRequest outRequest) {
SubPeer res = null;
if (peers.size() == 0) {
return null;
}
int start = new Random().nextInt(peers.size());
int i = start;
boolean stop = false;
do {
i = (i + 1) % peers.size();
SubPeer peer = peers.get(i);
stop = peer.updateScore(outRequest);
if (stop || res == null) {
res = peer;
} else if (peer.getScore() > res.getScore()) {
res = peer;
}
} while (!stop && i != start);
outRequest.setUsedPeer(res.getRemoteAddress());
return res;
}
public Connection connect(OutRequest outRequest) {
SubPeer peer = choosePeer(outRequest);
if (peer == null) {
return null;
}
Connection conn = peer.getPreferredConnection();
if (conn == null) {
conn = peer.connectTo();
}
return conn;
}
public boolean sendOutRequest(OutRequest outRequest) {
boolean res = false;
while (true) {
if (!outRequest.shouldRetry()) {
outRequest.setFuture();
break;
}
if (sendOutRequest(outRequest, connect(outRequest))) {
res = true;
break;
}
}
return res;
}
public TFuture> send(
ThriftRequest request,
InetAddress host,
int port
) throws TChannelError {
// Set the "cn" header
// TODO: should make "cn" an option
request.setTransportHeader(TransportHeaders.CALLER_NAME_KEY, this.topChannel.getServiceName());
return sendRequest(request, host, port);
}
public TFuture> send(
ThriftRequest request
) throws TChannelError {
return send(request, null, 0);
}
public TFuture> send(
JsonRequest request,
InetAddress host,
int port
) {
// Set the "cn" header
// TODO: should make "cn" an option
request.setTransportHeader(TransportHeaders.CALLER_NAME_KEY, this.topChannel.getServiceName());
return sendRequest(request, host, port);
}
public TFuture> send(
JsonRequest request
) {
return send(request, null, 0);
}
public TFuture send(
RawRequest request,
InetAddress host,
int port
) {
// Set the "cn" header
// TODO: should make "cn" an option
request.setTransportHeader(TransportHeaders.CALLER_NAME_KEY, this.topChannel.getServiceName());
return sendRequest(request, host, port);
}
public TFuture send(
RawRequest request
) {
return send(request, null, 0);
}
protected TFuture sendRequest(
Request request,
InetAddress host,
int port
) {
OutRequest outRequest = new OutRequest(this, request);
if (host != null) {
Connection conn = peerManager.findOrNew(new InetSocketAddress(host, port));
// No retry for direct connections
outRequest.disableRetry();
if (!sendOutRequest(outRequest, conn)) {
outRequest.setFuture();
}
} else if (peers.size() == 0) {
outRequest.setLastError(ErrorType.BadRequest, new TChannelNoPeerAvailable());
outRequest.setFuture();
} else {
sendOutRequest(outRequest);
}
return outRequest.getFuture();
}
protected boolean sendOutRequest(
OutRequest outRequest,
Connection connection
) {
Request request = outRequest.getRequest();
// Validate if the ArgScheme is set correctly
if (request.getArgScheme() == null) {
request.setArgScheme(ArgScheme.RAW);
outRequest.setLastError(ErrorType.BadRequest, "Expect call request to have Arg Scheme specified");
outRequest.setFuture();
return false;
}
// Set the default retry flag if it is not set
if (request.getRetryFlags() == null) {
request.setRetryFlags("c");
}
long initTimeout = this.initTimeout;
if (initTimeout <= 0) {
initTimeout = request.getTimeout();
}
if (connection == null) {
outRequest.setLastError(ErrorType.BadRequest, new TChannelNoPeerAvailable());
outRequest.setFuture();
return false;
} else if (!connection.waitForIdentified(initTimeout)) {
connection.clean();
if (connection.lastError() != null) {
outRequest.setLastError(ErrorType.NetworkError, connection.lastError());
} else {
outRequest.setLastError(ErrorType.NetworkError, new TChannelConnectionTimeout());
}
return false;
}
// Get a response router for our outbound channel
ResponseRouter router = connection.channel().pipeline().get(ResponseRouter.class);
return router.expectResponse(outRequest);
}
}
