thrift/swift:ThriftMethodProcessor代码分析

thrift是一种跨语言的RPC框架,为了保证在各种语言下都能正确表述,IDL语言在设计的时候就只能选取各种语言的共性。 比如对于null,虽然在java中定义一个字段为Integer类型,那么这个字段就可以为null,但thrift不允许primitive类型的数据字段为null,因为在c/c++的struct的字段中没有null的概念。做为方法的参数传递,primitive类型也是同样的道理。 所以thrift client 发送调用服务方法时,如果方法的参数类型为Integer,你即使将该参数设置为null,服务端收到的值也不是null,而是0.

经过反复跟踪我总算找到将null参数转为0的位置,就在服务端com.facebook.swift.service.ThriftMethodProcessor的私有方法readArguments中,下面是readArguments的代码,注意代码中作者添加的中文注释

private Object[] readArguments(TProtocol in)
            throws Exception
    {
        try {
            int numArgs = method.getParameterTypes().length;
            // 初始化参数数组,初始值都为null
            Object[] args = new Object[numArgs];
            TProtocolReader reader = new TProtocolReader(in);

            // 先从收到的数据库根据fieldId 将不为null的服务方法参数解析出来,
            // 为null的参数不在收到的数据中出现
            // 所以这个循环结束时,不为null的参数已经被正确解析并保存到args的对应位置了
            // client填null(原本就是null)的参数对应的位置就是null,
            reader.readStructBegin();
            while (reader.nextField()) {
                short fieldId = reader.getFieldId();

                ThriftCodec<?> codec = parameterCodecs.get(fieldId);
                if (codec == null) {
                    // unknown field
                    reader.skipFieldData();
                }
                else {
                    // 将参数调用对应用的解码器ThriftCodec解析成正确的数据类型,
                    // 存入arg数组对应的位置
                    args[thriftParameterIdToJavaArgumentListPositionMap.get(fieldId)] = reader.readField(codec);
                }
            }
            reader.readStructEnd();

            // 在下面这个循环中,对args数组中为null的参数进行检查,
            // 如果是primitive类型则调用 Defaults.defaultValue方法返回缺省值
            // 比如对于Integer类型缺省值就是0
            // 如果删除掉这一整段循环代码,service端收到的Integer参数就可以是null
            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());
        }
    }

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏魂祭心

原 Type System Overvie

3318
来自专栏编程

Golang中defer 的五个坑-第三部分

译注:全文总共有四篇,本文为同系列文章的第三篇 本文将侧重于讲解使用 defer 的一些技巧 如果你对 defer 的基本操作还没有清晰的认识,请先阅读这篇文章...

1925
来自专栏大内老A

Enterprise Library深入解析与灵活应用(8):WCF与Exception Handling AppBlock集成[上]

在《WCF技术剖析(卷1)》的最后一章,我给出了一个具体的应用WCF的分布式应用实例,我把这个实例命名为PetShop。在这个例子中,我利用WCF的扩展实现了一...

18310
来自专栏纯洁的微笑

jvm系列(一):java类的加载机制

1、什么是类的加载 类的加载指的是将类的.class文件中的二进制数据读入到内存中,将其放在运行时数据区的方法区内,然后在堆区创建一个 java.lang.Cl...

3596
来自专栏java一日一条

浅谈JVM及原理

JVM, 中文名是Java虚拟机, 正如它的名字, 是一个虚拟机器,来模拟通用的物理机。 JVM是一个标准,一套规范, 规定了.class文件在其内部运行的相...

562
来自专栏PhpZendo

PHP 垃圾回收与内存管理指引

php 的变量存储在「zval」变量容器(数据结构)中,「zval」属性包含如下信息:

290
来自专栏SDNLAB

Open vSwitch系列之数据结构解析深入分析ofpbuf

上一篇我们分析了hmap,hamp可以说是Open vSwitch中基石结构,很多Open vSwitch中数据结构都依赖hmap。本篇我们来分析一下ofpbu...

3428
来自专栏黑白安全

php代码审计之弱类型引发的灾难

有人说php是世界上最好的语言,这可能是对开发人员来说,确实有这方面的特点,因为它开发起来不像其他语言那样麻烦,就比如:弱类型,它不需要像java等语言那样明确...

532
来自专栏后端技术探索

反射机制、依赖注入、控制反转

反向: dll->类[方法,属性]. 从已经有的dll文件反编译得到其中的一些可用的方法.

702
来自专栏喔家ArchiSelf

全栈必备 Java基础

那一年,从北邮毕业,同一年,在大洋的彼岸诞生了一门对软件业将产生重大影响的编程语言,它就是——Java。1998年的时候,开始学习Java1.2,并在Java ...

684

扫码关注云+社区