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.
com.facebook.swift.service.ThriftMethodProcessorCustom Maven / Gradle / Ivy
package com.facebook.swift.service;
import com.facebook.nifty.core.RequestContext;
import com.facebook.nifty.core.RequestContexts;
import com.facebook.swift.codec.ThriftCodec;
import com.facebook.swift.codec.ThriftCodecManager;
import com.facebook.swift.codec.internal.TProtocolReader;
import com.facebook.swift.codec.internal.TProtocolWriter;
import com.facebook.swift.codec.metadata.ThriftFieldMetadata;
import com.facebook.swift.codec.metadata.ThriftType;
import com.facebook.swift.service.ContextChain;
import com.facebook.swift.service.ThriftServiceProcessor;
import com.facebook.swift.service.metadata.ThriftMethodMetadata;
import com.google.common.base.Defaults;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableMap.Builder;
import com.google.common.primitives.Primitives;
import com.google.common.reflect.TypeToken;
import com.google.common.util.concurrent.FutureCallback;
import com.google.common.util.concurrent.Futures;
import com.google.common.util.concurrent.ListenableFuture;
import com.google.common.util.concurrent.SettableFuture;
import org.apache.thrift.TApplicationException;
import org.apache.thrift.protocol.TMessage;
import org.apache.thrift.protocol.TMessageType;
import org.apache.thrift.protocol.TProtocol;
import org.apache.thrift.protocol.TProtocolException;
import org.weakref.jmx.Managed;
import com.google.common.base.Preconditions;
import javax.annotation.concurrent.ThreadSafe;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Type;
import java.util.Map;
import static org.apache.thrift.TApplicationException.INTERNAL_ERROR;
/**
* 定制{@link ThriftMethodProcessor},收到的primitive封装类型(Boolean,Integer...)如果是{@code null}不会被转成default 0.
* 通过调用 {@link #setUseDefaultValueIfPrimitiveWrap(boolean)} 方法设置为{@code true}可以禁用该特性
* @author guyadong
*
*/
@ThreadSafe
public class ThriftMethodProcessorCustom extends ThriftMethodProcessor
{
private final String name;
private final String serviceName;
private final String qualifiedName;
private final Object service;
private final Method method;
private final String resultStructName;
private final boolean oneway;
private final ImmutableList parameters;
private final Map> parameterCodecs;
private final Map thriftParameterIdToJavaArgumentListPositionMap;
private final ThriftCodec successCodec;
private final Map, ExceptionProcessor> exceptionCodecs;
/**
* 全局开关> builder = ImmutableMap.builder();
for (ThriftFieldMetadata fieldMetadata : methodMetadata.getParameters()) {
@SuppressWarnings("unused")
Builder> p = builder.put(fieldMetadata.getId(), codecManager.getCodec(fieldMetadata.getThriftType()));
}
parameterCodecs = builder.build();
// Build a mapping from thrift parameter ID to a position in the formal argument list
ImmutableMap.Builder parameterOrderingBuilder = ImmutableMap.builder();
short javaArgumentPosition = 0;
for (ThriftFieldMetadata fieldMetadata : methodMetadata.getParameters()) {
@SuppressWarnings("unused")
Builder p = parameterOrderingBuilder.put(fieldMetadata.getId(), javaArgumentPosition++);
}
thriftParameterIdToJavaArgumentListPositionMap = parameterOrderingBuilder.build();
ImmutableMap.Builder, ExceptionProcessor> exceptions = ImmutableMap.builder();
for (Map.Entry entry : methodMetadata.getExceptions().entrySet()) {
Class type = TypeToken.of(entry.getValue().getJavaType()).getRawType();
ExceptionProcessor processor = new ExceptionProcessor(entry.getKey(), codecManager.getCodec(entry.getValue()));
@SuppressWarnings("unused")
Builder, ExceptionProcessor> p = exceptions.put(type, processor);
}
exceptionCodecs = exceptions.build();
successCodec = (ThriftCodec) codecManager.getCodec(methodMetadata.getReturnType());
}
@Managed
@Override
public String getName()
{
return name;
}
@Override
public Class getServiceClass() {
return service.getClass();
}
@Override
public String getServiceName()
{
return serviceName;
}
@Override
public String getQualifiedName()
{
return qualifiedName;
}
@Override
public ListenableFuture process(TProtocol in, final TProtocol out, final int sequenceId, final ContextChain contextChain)
throws Exception
{
// read args
contextChain.preRead();
Object[] args = readArguments(in);
contextChain.postRead(args);
final RequestContext requestContext = RequestContexts.getCurrentContext();
in.readMessageEnd();
// invoke method
final ListenableFuture invokeFuture = invokeMethod(args);
final SettableFuture resultFuture = SettableFuture.create();
Futures.addCallback(invokeFuture, new FutureCallback()
{
@Override
public void onSuccess(Object result)
{
if (oneway) {
@SuppressWarnings("unused")
boolean s = resultFuture.set(true);
}
else {
RequestContext oldRequestContext = RequestContexts.getCurrentContext();
RequestContexts.setCurrentContext(requestContext);
// write success reply
try {
contextChain.preWrite(result);
writeResponse(out,
sequenceId,
TMessageType.REPLY,
"success",
(short) 0,
successCodec,
result);
contextChain.postWrite(result);
@SuppressWarnings("unused")
boolean s = resultFuture.set(true);
}
catch (Exception e) {
// An exception occurred trying to serialize a return value onto the output protocol
@SuppressWarnings("unused")
boolean b = resultFuture.setException(e);
}
finally {
RequestContexts.setCurrentContext(oldRequestContext);
}
}
}
@Override
public void onFailure(Throwable t)
{
RequestContext oldRequestContext = RequestContexts.getCurrentContext();
RequestContexts.setCurrentContext(requestContext);
try {
contextChain.preWriteException(t);
if (!oneway) {
ExceptionProcessor exceptionCodec = exceptionCodecs.get(t.getClass());
if (exceptionCodec != null) {
// write expected exception response
writeResponse(
out,
sequenceId,
TMessageType.REPLY,
"exception",
exceptionCodec.getId(),
exceptionCodec.getCodec(),
t);
contextChain.postWriteException(t);
} else {
// unexpected exception
TApplicationException applicationException =
ThriftServiceProcessor.writeApplicationException(
out,
method.getName(),
sequenceId,
INTERNAL_ERROR,
"Internal error processing " + method.getName(),
t);
contextChain.postWriteException(applicationException);
}
}
@SuppressWarnings("unused")
boolean b = resultFuture.set(true);
}
catch (Exception e) {
// An exception occurred trying to serialize an exception onto the output protocol
@SuppressWarnings("unused")
boolean b = resultFuture.setException(e);
}
finally {
RequestContexts.setCurrentContext(oldRequestContext);
}
}
});
return resultFuture;
}
private ListenableFuture invokeMethod(Object[] args)
{
try {
Object response = method.invoke(service, args);
if (response instanceof ListenableFuture) {
return (ListenableFuture) response;
}
return Futures.immediateFuture(response);
}
catch (IllegalAccessException | IllegalArgumentException e) {
// These really should never happen, since the method metadata should have prevented it
return Futures.immediateFailedFuture(e);
}
catch (InvocationTargetException e) {
Throwable cause = e.getCause();
if (cause != null) {
return Futures.immediateFailedFuture(cause);
}
return Futures.immediateFailedFuture(e);
}
}
private Object[] readArguments(TProtocol in)
throws Exception
{
try {
int numArgs = method.getParameterTypes().length;
Object[] args = new Object[numArgs];
TProtocolReader reader = new TProtocolReader(in);
// Map incoming arguments from the ID passed in on the wire to the position in the
// java argument list we expect to see a parameter with that ID.
reader.readStructBegin();
while (reader.nextField()) {
short fieldId = reader.getFieldId();
ThriftCodec codec = parameterCodecs.get(fieldId);
if (codec == null) {
// unknown field
reader.skipFieldData();
}
else {
// Map the incoming arguments to an array of arguments ordered as the java
// code for the handler method expects to see them
args[thriftParameterIdToJavaArgumentListPositionMap.get(fieldId)] = reader.readField(codec);
}
}
reader.readStructEnd();
// Walk through our list of expected parameters and if no incoming parameters were
// mapped to a particular expected parameter, fill the expected parameter slow with
// the default for the parameter type.
// 根据全局开关判断是否对null替换为default value
if(Boolean.TRUE.equals(useDefaultValueIfPrimitiveWrap)){
int argumentPosition = 0;
for (ThriftFieldMetadata argument : parameters) {
if (args[argumentPosition] == null) {
Type argumentType = argument.getThriftType().getJavaType();
if (argumentType instanceof Class) {
Class argumentClass = (Class) argumentType;
argumentClass = Primitives.unwrap(argumentClass);
args[argumentPosition] = Defaults.defaultValue(argumentClass);
}
}
argumentPosition++;
}
}
return args;
}
catch (TProtocolException e) {
// TProtocolException is the only recoverable exception
// Other exceptions may have left the input stream in corrupted state so we must
// tear down the socket.
throw new TApplicationException(TApplicationException.PROTOCOL_ERROR, e.getMessage());
}
}
private void writeResponse(TProtocol out,
int sequenceId,
byte responseType,
String responseFieldName,
short responseFieldId,
ThriftCodec responseCodec,
T result) throws Exception
{
out.writeMessageBegin(new TMessage(name, responseType, sequenceId));
TProtocolWriter writer = new TProtocolWriter(out);
writer.writeStructBegin(resultStructName);
writer.writeField(responseFieldName, (short) responseFieldId, responseCodec, result);
writer.writeStructEnd();
out.writeMessageEnd();
out.getTransport().flush();
}
private static final class ExceptionProcessor
{
private final short id;
private final ThriftCodec codec;
@SuppressWarnings("unchecked")
private ExceptionProcessor(short id, ThriftCodec coded)
{
this.id = id;
this.codec = (ThriftCodec) coded;
}
public short getId()
{
return id;
}
public ThriftCodec getCodec()
{
return codec;
}
}
/**
* 设置对于primitive wrap类型是否将{@code null}转为default value，默认为{@code false}
