Microsoft/thrifty:RPC方法返回NULL的异常处理

迈不过友情╰ 2022-03-28 13:46 293阅读 0赞

我们知道：thrift框架是不允许返回值为null的，如果返回值为null,client端会抛出异常,我在之前用[facebook/swift][facebook_swift]框架时就遇到了这个问题，这是当时解决问题的记录[《thrift:返回null的解决办法》][thrift_null],现在使用[Microsoft/thrifty][Microsoft_thrifty]框架实现的客户端同样也存在这个问题。  
下面是thifty-compiler生成的client端存根代码的receive方法的部分片段：

@Override
        protected PersonBean receive(Protocol protocol, MessageMetadata metadata) throws Exception {
        
          PersonBean result = null;
          .......
          if (result != null) {
        
            return result;
          } else if (ex1 != null) {
        
            throw ex1;
          } else {
        
            throw new ThriftException(ThriftException.Kind.MISSING_RESULT, "Missing result");
          }
        }
      }

可以看到，返回结果为null时，会抛出类型为`MISSING_RESULT`的`ThriftException`异常。  
知道了原因，解决问题的方法有了：

/**
    		* 当前调用的回调函数，由当前接口方法设置
    		*/
    		final ServiceMethodCallback<Integer> callback = null;
    		/** 创建一个侦听器对象，用于检测socket关闭状态 */
    		final AsyncClientBase.Listener closeListener = new AsyncClientBase.Listener(){
        
    		   @Override
    		   public void onTransportClosed() {
        
    		   }
    		
    		   @Override
    		   public void onError(Throwable error) {
        
    		       // 如果关闭时有异常，则将异常转给callback对象，
    		       // 当方法返回值为null时抛出的ThriftException异常会在这里被拦截发给callback对象
    		       callback.onError(error);
    		   }        
    		};
    		
            SocketTransport transport = new SocketTransport.Builder(host,port).build();
            transport.connect();
            Protocol protocol = new BinaryProtocol(transport);
            /* force set private field 'strictWrite' to true */
            Field field = BinaryProtocol.class.getDeclaredField("strictWrite");
            field.setAccessible(true);
            field.set(protocol, true);
            // 创建client端接口实例
            final mypackage.MyPrjClient service = new mypackage.MyPrjClient(protocol,closeListener);
            
            // 创建callback对象
    		callback = new ServiceMethodCallback<Integer>(){
        
    	        @Override
    	        public void onSuccess(Integer result) {
        
    	            // do something       
    	            try {
        
    	            	// 关闭连接
    	                service.close();
    	            } catch (IOException e) {
        
    	            }
    	        }
    	
    	        @Override 
    	        public void onError(Throwable error) {
        
    	        	// 对象ThriftException异常，判断类型是否为MISSING_RESULT，是则调用onSuccess正常返回null       
    	            if(error instanceof ThriftException ){
        
    	                if(((ThriftException)error).kind == ThriftException.Kind.MISSING_RESULT  ){
        
    	                    this.onSuccess(null);
    	                }
    	            }
    	            // do something
    	            try {
        
    	                // 关闭连接
    	                service.close();
    	            } catch (IOException e) {
        
    	            }
    	        }};
    	    // 执行异步接口调用
            service.callMethod(123,callback);

[facebook_swift]: https://github.com/facebook/swift
[thrift_null]: https://blog.csdn.net/10km/article/details/78491695
[Microsoft_thrifty]: https://github.com/Microsoft/thrifty