Skip to content

Commit 469584a

Browse files
authored
Increase MaxMessageSize for Thrift Transport (#305)
1 parent 154db98 commit 469584a

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

dumper/lib-ext-hive-metastore/src/main/java/com/google/edwmigration/dumper/ext/hive/metastore/HiveMetastoreThriftClient.java

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
import javax.annotation.Nonnegative;
2525
import javax.annotation.Nonnull;
2626
import javax.annotation.concurrent.NotThreadSafe;
27+
import org.apache.thrift.TConfiguration;
2728
import org.apache.thrift.protocol.TBinaryProtocol;
2829
import org.apache.thrift.protocol.TProtocol;
2930
import org.apache.thrift.transport.TSocket;
@@ -131,7 +132,14 @@ public HiveMetastoreThriftClient build() throws TTransportException {
131132
// the TTransport here with a TSaslClientTransport parameterized accordingly. This hasn't been
132133
// done yet
133134
// in the interest of expediency.
134-
TTransport transport = new TSocket(host, port);
135+
TTransport transport =
136+
new TSocket(
137+
new TConfiguration(
138+
Integer.MAX_VALUE,
139+
TConfiguration.DEFAULT_MAX_FRAME_SIZE,
140+
TConfiguration.DEFAULT_RECURSION_DEPTH),
141+
host,
142+
port);
135143
TProtocol protocol = new TBinaryProtocol(transport);
136144
transport.open();
137145

0 commit comments

Comments
 (0)