thrift/swift:ThriftMethodProcessor代码分析

骑猪看日落 2022-06-03 03:12 54阅读 0赞

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());
            }
        }