thrift/swift:服务端数据类型和client端数据类型之间的直接转换

2019-05-25 21:49:18 浏览数 (1)

版权声明:本文为博主原创文章,转载请注明源地址。 https://cloud.tencent.com/developer/article/1433556

需求描述

我们知道,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数据对象转换数据流的基本功能:

代码语言:javascript复制
/** 从一个数据对象Object,读取ThriftFieldMetadata指定的字段的值 */
com.facebook.swift.codec.internal.reflection.AbstractReflectionThriftCodec#getFieldValue(Object instance, ThriftFieldMetadata 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

代码语言:javascript复制
/** 构造{@code metadata}指定类型的实例并填充字段 */
public static <T>T constructStruct(Map<Short, Object> data,ThriftStructMetadata metadata) throws Exception;
/** 填充{@code instance}实例的字段 */
public static <T>T fillStructField(Map<Short, Object> data,ThriftStructMetadata metadata,T instance) 
        throws Exception
/** 获取{@code field}指定的字段值 */
public static Object getFieldValue(Object instance, ThriftFieldMetadata field) throws Exception ;
/** 根据{@code metadata}类型数据获取{@code instance}实例所有的字段值 */
public static Map<Short, Object> getFiledValues(Object instance, ThriftStructMetadata metadata);

应用示例

以下是调用上面的方法实现的转换器com.google.common.base.Function接口

代码语言:javascript复制
import static com.google.common.base.Preconditions.*;
import java.util.Map;

import com.facebook.swift.codec.metadata.ThriftStructMetadata;
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 ThriftStructMetadata leftMetadata;
    /** 目标数据类型的thrift数据结构描述对象 */
    private final ThriftStructMetadata rightMetadata;
    /** 构造方法 */
    public ThriftStructTransformer(Class<L> left, Class<R> right) {
        // 根据数据类型生成thrift数据结构描述对象
        this.leftMetadata= CATALOG.getThriftStructMetadata(checkNotNull(left,"left is null"));
        this.rightMetadata= CATALOG.getThriftStructMetadata(checkNotNull(right,"right is null"));
    }
    /** 源类型到目标数据类型转换实现 */
    @Override
    public R apply(L input) {
        if(null == input){
            return null;
        }
        try {
            // 读取所有字段值
            Map<Short, Object> data = getFiledValues(input,leftMetadata);
            // 根据字段值构造目标数据对象
            return constructStruct(data,rightMetadata);
        } catch (Exception e) {
            Throwables.throwIfUnchecked(e);
            throw new RuntimeException(e);
        }
    }
    @Override
    public String toString() {
        StringBuilder builder = new StringBuilder()
                .append("ThriftStructTransformer [leftClass=")
                .append(leftMetadata.getStructType())
                .append(", rightClass=")
                .append(rightMetadata.getStructType())
                .append("]");
        return builder.toString();
    }
}

完整代码参见:

https://gitee.com/l0km/common-java/blob/master/common-thrift/src/main/java/net/gdface/thrift/ThriftStructTransformer.java

0 人点赞