thrift:拦截器ThriftEventHandler获取调用参数

之前写过一篇博客《spring:拦截器(HandlerInterceptor)中获取POST请求参数》
介绍了如何在在spring的拦截器一层获取POST请求参数。
当我在thrift环境使用ThriftEventHandler拦截请求获取请求参数，也遇到了类似的问题：

这个阶段，因为请求方法的参数还没有被反序列化，所以要想获取HTTP的请求参数，就要自己从HttpServletRequest中获取。要从请求内容里获取参数，就要从TProtocol里获取到，然后解析出数据就可以了。
  但是，TProtocol读取数据是单向的，一次性的，如果在ThriftEventHandler获取了TProtocol中的数据，后续方法调用解析参数时再读取TProtocol就会失败，因为输入流结束了。
  所以拦截器(ThriftEventHandler)中获取请求参数本身并不是问题，问题就是要解决TProtocol的输入流能被多次读取问题。
  解决的思路与spring拦截的差不多，就是将TProtocol基于TProtocolDecorator封装起来，用封装后的BufferedProtocol替换掉整个请求链路中的TProtocol。
  要实现的BufferedProtocol类负责将TProtocol中的完整数据读取出来保存下来，以便重复读取。
  这样才能确保TProtocol可以多次被读取。
  
以下为BufferedProtocol封装类实现

org.apache.thrift.protocol.BufferedProtocol.java

import org.apache.thrift.transport.AutoExpandingBufferWriteTransport;
import org.apache.thrift.transport.TMemoryInputTransport;
import org.apache.thrift.transport.TTransport;
import org.apache.thrift.transport.TTransportException;

import java.util.Arrays;

/**
 * 基于{@link TProtocolDecorator}实现{@link TTransport}可重复读取
 * 
 * @author guyadong
 * @since 1.3.0
 *
 */
public class BufferedProtocol extends TProtocolDecorator {
	/**
	 * 装饰实例
	 * */
	public final TProtocol delegate;
	/**
	 * 从装饰{@link #delegate}中读取的数据缓冲区
	 */
	private final byte[] buffer;

	private BufferedProtocol(TProtocol protocol) {
		this(protocol,null);
	}
	private BufferedProtocol(TProtocol protocol,byte[] buffer) {
		super(protocol);
		this.delegate = protocol;
		try {
			/**
			 * 读取输入的{@link TTransport}所有数据创建{@link TMemoryInputTransport}实例
			 */
			this.buffer = null == buffer ? copy(protocol.getTransport()) : buffer;
			this.trans_ = new TMemoryInputTransport(this.buffer);
			this.delegate.trans_ = this.trans_;
		} catch (TTransportException e) {
			throw new RuntimeException(e);
		}
	}
	private static byte[] copy(TTransport transport) throws TTransportException {
		try (AutoExpandingBufferWriteTransport autoExpandingTransport = new AutoExpandingBufferWriteTransport(1024,
				1.5)) {
			int readCount;
			byte[] buffer = new byte[1024];
			while ((readCount = transport.read(buffer, 0, buffer.length)) > 0) {
				autoExpandingTransport.write(buffer, 0, readCount);
			}
			return Arrays.copyOfRange(autoExpandingTransport.getBuf().array(), 0, autoExpandingTransport.getPos());
		}
	}
	/**
	 * 复位 TTransport, 以支持从 {@link #buffer} 重复读取
	 */
	@Override
	public void reset() {
		delegate.reset();
		((TMemoryInputTransport)delegate.trans_).reset(buffer);
	}
	public static BufferedProtocol of(TProtocol protocol) {
		return new BufferedProtocol(protocol);
	}
}

进一步还要重写ThriftServiceProcessor，将TProtocol替换为BufferedProtocol,这样才能生效。
详细代码不贴了，参见项目仓库。
https://gitee.com/l0km/xthrift/blob/master/xthrift-service/src/main/java/com/facebook/swift/service/XthriftMethodProcessor.java

完整代码

完整代码参见码云仓库：https://gitee.com/l0km/xthrift