You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@iotdb.apache.org by ji...@apache.org on 2020/12/01 11:51:55 UTC

[iotdb] 01/02: fix non socket transport

This is an automated email from the ASF dual-hosted git repository.

jiangtian pushed a commit to branch cluster_add_snappy
in repository https://gitbox.apache.org/repos/asf/iotdb.git

commit e515abd1195e5b48b47cf58d94a87ef64da77511
Author: jt <jt...@163.com>
AuthorDate: Tue Dec 1 18:54:13 2020 +0800

    fix non socket transport
---
 .../apache/iotdb/rpc/TimeoutChangeableTFastFramedTransport.java    | 6 +++++-
 .../apache/iotdb/rpc/TimeoutChangeableTSnappyFramedTransport.java  | 7 ++++++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/service-rpc/src/main/java/org/apache/iotdb/rpc/TimeoutChangeableTFastFramedTransport.java b/service-rpc/src/main/java/org/apache/iotdb/rpc/TimeoutChangeableTFastFramedTransport.java
index 4aa50a7..99f873d 100644
--- a/service-rpc/src/main/java/org/apache/iotdb/rpc/TimeoutChangeableTFastFramedTransport.java
+++ b/service-rpc/src/main/java/org/apache/iotdb/rpc/TimeoutChangeableTFastFramedTransport.java
@@ -45,7 +45,11 @@ public class TimeoutChangeableTFastFramedTransport extends TElasticFramedTranspo
 
     @Override
     public TTransport getTransport(TTransport trans) {
-      return new TimeoutChangeableTSnappyFramedTransport((TSocket) trans);
+      if (trans instanceof TSocket) {
+        return new TimeoutChangeableTSnappyFramedTransport((TSocket) trans);
+      } else {
+        return new TSnappyElasticFramedTransport(trans);
+      }
     }
   }
 }
diff --git a/service-rpc/src/main/java/org/apache/iotdb/rpc/TimeoutChangeableTSnappyFramedTransport.java b/service-rpc/src/main/java/org/apache/iotdb/rpc/TimeoutChangeableTSnappyFramedTransport.java
index b251f3f..57ae414 100644
--- a/service-rpc/src/main/java/org/apache/iotdb/rpc/TimeoutChangeableTSnappyFramedTransport.java
+++ b/service-rpc/src/main/java/org/apache/iotdb/rpc/TimeoutChangeableTSnappyFramedTransport.java
@@ -20,6 +20,7 @@
 package org.apache.iotdb.rpc;
 
 import java.net.SocketException;
+import org.apache.thrift.transport.TFastFramedTransport;
 import org.apache.thrift.transport.TSocket;
 import org.apache.thrift.transport.TTransport;
 import org.apache.thrift.transport.TTransportFactory;
@@ -44,7 +45,11 @@ public class TimeoutChangeableTSnappyFramedTransport extends TSnappyElasticFrame
   public static class Factory extends TTransportFactory {
     @Override
     public TTransport getTransport(TTransport trans) {
-      return new TimeoutChangeableTSnappyFramedTransport((TSocket) trans);
+      if (trans instanceof TSocket) {
+        return new TimeoutChangeableTSnappyFramedTransport((TSocket) trans);
+      } else {
+        return new TFastFramedTransport(trans);
+      }
     }
   }
 }