| package com.facebook.thrift; |
| import com.facebook.thrift.TApplicationException; |
| import com.facebook.thrift.TSerializer; |
| import com.facebook.thrift.transport.TTransport; |
| import com.facebook.thrift.transport.TSocket; |
| import com.facebook.thrift.transport.TFramedTransport; |
| import com.facebook.thrift.transport.TTransportException; |
| import com.facebook.thrift.protocol.TBinaryProtocol; |
| import com.facebook.thrift.protocol.TSimpleJSONProtocol; |
| import java.util.HashMap; |
| import java.util.HashSet; |
| import java.util.ArrayList; |
| public class OverloadNonblockingServer { |
| public static void main(String[] args) throws Exception { |
| int msg_size_mb = Integer.parseInt(args[0]); |
| int msg_size = msg_size_mb * 1024 * 1024; |
| TSocket socket = new TSocket("localhost", 9090); |
| TBinaryProtocol binprot = new TBinaryProtocol(socket); |
| binprot.writeI32(msg_size); |
| for (int i = 0; i < msg_size_mb; i++) { |
| binprot.writeBinary(new byte[1024 * 1024]); |