需求描述
我们知道,thrift/swift
框架中,服务端的数据类型与client的类型是相互独立的,比如服务端有一个ClassA
类型,到了client端,同样也会生成一个同名的类。服务端和client相互通信时,client的 ClassA
会被转成数据流(二进制或HTML或JSON…),通过网络传输到服务端,服务端收到数据流后再转换服务端的ClassA
,反之亦然。这个过程是由thrift框架自动完成的,对应用程序来说是透明的。
那么当我们希望client端应用程序与服务端共用同一个数据类型,也就是服务端的ClassA
时,就需要一种从 client ClassA
到服务端ClassA
之间的直接转换,thrift/swift
框架并没有提供这样的直接转换机制,该怎么办呢?
解决方案
首先需要找到thrift/swift
将数据对象转换成数据流的地方。
研究了swift
的代码,我在swift-codec-0.14.2.jar包中找到数据转换机制实现的地方,下面几个方法实现thrift数据对象转换数据流的基本功能:
/** 从一个数据对象Object,读取ThriftField@R_301_338@data指定的字段的值 */
com.facebook.swift.codec.internal.reflection.AbstractReflectionThriftCodec#getFieldValue(Object instance,ThriftField@R_301_338@data field)
/** 根据Map中字段值构造一个数据对象 */
com.facebook.swift.codec.internal.reflection.ReflectionThriftStructCodec#constructStruct(Map<Short,Object> data)
/** 从数据流protocol读取字段值转成数据对象 T */
T com.facebook.swift.codec.internal.reflection.ReflectionThriftStructCodec.read(TProtocol protocol)
/** 将数据对象instance写入数据流protocol */
void com.facebook.swift.codec.internal.reflection.ReflectionThriftStructCodec.write(T instance,TProtocol protocol)
通过这几个方法看懂了swift的数据对象转换机制,就可以参照这个机制实现服务端数据类型和client数据类型的直接转换。
具体代码参数gitee仓库中的代码:
https://gitee.com/l0km/common-java/blob/master/common-thrift/src/main/java/net/gdface/thrift/ThriftUtils.java
/** 构造{@code @R_301_338@data}指定类型的实例并填充字段 */
public static <T>T constructStruct(Map<Short,Object> data,ThriftStruct@R_301_338@data @R_301_338@data) throws Exception;
/** 填充{@code instance}实例的字段 */
public static <T>T fillStructField(Map<Short,ThriftStruct@R_301_338@data @R_301_338@data,T instance)
throws Exception
/** 获取{@code field}指定的字段值 */
public static Object getFieldValue(Object instance,ThriftField@R_301_338@data field) throws Exception ;
/** 根据{@code @R_301_338@data}类型数据获取{@code instance}实例所有的字段值 */
public static Map<Short,Object> getFiledValues(Object instance,ThriftStruct@R_301_338@data @R_301_338@data);
应用示例
以下是调用上面的方法实现的转换器com.google.common.base.Function
接口
import static com.google.common.base.Preconditions.*;
import java.util.Map;
import com.facebook.swift.codec.@R_301_338@data.ThriftStruct@R_301_338@data;
import com.google.common.base.Function;
import com.google.common.base.Throwables;
import static net.gdface.thrift.ThriftUtils.*;
/** * 有{@link com.facebook.swift.codec.ThriftStruct}注释的类型之间的转换 * @author guyadong * * @param <L> 源数据类型,client或服务端java bean数据类型 * @param <R> 目标数据类型,L对应的服务或client端java bean数据类型 */
public class ThriftStructTransformer<L,R> implements Function<L,R>{
/** 源数据类型的thrift数据结构描述对象 */
private final ThriftStruct@R_301_338@data left@R_301_338@data;
/** 目标数据类型的thrift数据结构描述对象 */
private final ThriftStruct@R_301_338@data right@R_301_338@data;
/** 构造方法 */
public ThriftStructTransformer(Class<L> left,Class<R> right) {
// 根据数据类型生成thrift数据结构描述对象
this.left@R_301_338@data= CATALOG.getThriftStruct@R_301_338@data(checkNotNull(left,"left is null"));
this.right@R_301_338@data= CATALOG.getThriftStruct@R_301_338@data(checkNotNull(right,"right is null"));
}
/** 源类型到目标数据类型转换实现 */
@Override
public R apply(L input) {
if(null == input){
return null;
}
try {
// 读取所有字段值
Map<Short,Object> data = getFiledValues(input,left@R_301_338@data);
// 根据字段值构造目标数据对象
return constructStruct(data,right@R_301_338@data);
} catch (Exception e) {
Throwables.throwIfUnchecked(e);
throw new RuntimeException(e);
}
}
@Override
public String toString() {
StringBuilder builder = new StringBuilder()
.append("ThriftStructTransformer [leftClass=")
.append(left@R_301_338@data.getStructType())
.append(",rightClass=")
.append(right@R_301_338@data.getStructType())
.append("]");
return builder.toString();
}
}
完整代码参见:
https://gitee.com/l0km/common-java/blob/master/common-thrift/src/main/java/net/gdface/thrift/ThriftStructTransformer.java