import java.io.File; import java.io.IOException; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.atomic.AtomicInteger; import org.apache.http.HttpEntity; import org.apache.http.client.methods.CloseableHttpResponse; import org.apache.http.client.methods.HttpPost; import org.apache.http.entity.ContentType; import org.apache.http.entity.mime.MultipartEntityBuilder; import org.apache.http.entity.mime.content.FileBody; import org.apache.http.entity.mime.content.StringBody; import org.apache.http.impl.client.CloseableHttpClient; import org.apache.http.impl.client.HttpClients; import org.apache.http.util.EntityUtils; public class ClientMultipartFormPostTest { private static ExecutorService pool = Executors.newFixedThreadPool(300); public static void main(String[] args) throws Exception { final String path = args[0];//文件地址 final String url = args[1]; //调用的URL final int i_len = Integer.parseInt(args[2]);//线程总数 final int j_len = Integer.parseInt(args[3]);//每个线程的请求数(暂时没用到) final AtomicInteger c = new AtomicInteger(0); final long s = System.currentTimeMillis(); for (int i = 0; i < i_len; i++) { new Thread(new Runnable() { @Override public void run() { while (true) { try { upLoadLogMultiThread(url,path); int cc = c.addAndGet(1); if (cc % 1000 == 0) { System.out.println(String.format("c: %d, t: %d", cc, (System.currentTimeMillis() - s))); } } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } } }).start(); } } public static void upLoadLogMultiThread(String url,String path) throws IOException{ CloseableHttpClient httpclient = HttpClients.createDefault(); try { HttpPost httppost = new HttpPost(url); FileBody bin = new FileBody(new File(path)); StringBody comment = new StringBody("A binary file of some kind", ContentType.TEXT_PLAIN); HttpEntity reqEntity = MultipartEntityBuilder.create() .addPart("bin", bin) .addPart("comment", comment) .build(); httppost.setEntity(reqEntity); CloseableHttpResponse response = httpclient.execute(httppost); try { HttpEntity resEntity = response.getEntity(); EntityUtils.consume(resEntity); } finally { response.close(); } } finally { httpclient.close(); } } } |