1.控制OkHttp线程池数量

unknown
1 parent e945a1ed
Showing 3 changed files with 103 additions and 31 deletions Show diff stats
build.gradle
src/main/java/com/taover/util/UtilHttpByOkHttp.java
src/test/java/OkHttpDebug.java
@@ -59,7 +59,7 @@ uploadArchives {
                 authentication(userName: NEXUS_USERNAME, password: NEXUS_PASSWORD)
             }
             pom.project {
-                version '1.1.112'
+                version '1.1.114'
                 artifactId ARTIFACT_Id
                 groupId GROUP_ID
                 packaging TYPE
@@ -24,8 +24,7 @@ import okhttp3.ResponseBody;
 public class UtilHttpByOkHttp {
 	//防止线程并发，使用threadlocal管理client
-	private static ThreadLocal<OkHttpClient> localHttpClientForHttp = new ThreadLocal<OkHttpClient>();
-	private static ThreadLocal<OkHttpClient> localHttpClientForHttps = new ThreadLocal<OkHttpClient>();
+	private static Map<String, OkHttpClient> localHttpClient = new HashMap<String, OkHttpClient>();
 	public static final String METHOD_GET = "GET";
 	public static final String METHOD_POST = "POST";
@@ -131,38 +130,38 @@ public class UtilHttpByOkHttp {
 	}
 	public static OkHttpClient getHttpClient(String url, int timeoutInSecond){
-		OkHttpClient client = null;
+		OkHttpClient client = null;		
 		boolean isHttps = url.trim().toLowerCase().startsWith("https");
-		if(isHttps){
-			client = localHttpClientForHttps.get();
-		}else{
-			client = localHttpClientForHttp.get();
-		}	
-		
-		if(client != null){
-			return client;
-		}
-		
-		if(isHttps){
-			client = buildOkHttpClientForHttps(timeoutInSecond);
-		}else{
-			OkHttpClient.Builder builder = new OkHttpClient.Builder();
-			builder.connectTimeout(timeoutInSecond, TimeUnit.SECONDS)
-	            .readTimeout(timeoutInSecond, TimeUnit.SECONDS)
-	            .writeTimeout(timeoutInSecond,TimeUnit.SECONDS)
-	            .retryOnConnectionFailure(true);
-			client = builder.build();
-		}
-		
-		if(isHttps){
-			localHttpClientForHttps.set(client);
-		}else{
-			localHttpClientForHttp.set(client);
-		}
-		
+		String mapKey = getHttpClientMapKey(isHttps, timeoutInSecond);
+		synchronized (localHttpClient) {
+			client = localHttpClient.get(mapKey);
+			if(client != null){
+				return client;
+			}
+			
+			if(isHttps){
+				client = buildOkHttpClientForHttps(timeoutInSecond);
+			}else{
+				OkHttpClient.Builder builder = new OkHttpClient.Builder();
+				builder.connectTimeout(timeoutInSecond, TimeUnit.SECONDS)
+		            .readTimeout(timeoutInSecond, TimeUnit.SECONDS)
+		            .writeTimeout(timeoutInSecond,TimeUnit.SECONDS)
+		            .retryOnConnectionFailure(true);
+				client = builder.build();
+			}
+			localHttpClient.put(mapKey, client);
+		}		
 		return client;
 	}
+	private static String getHttpClientMapKey(boolean isHttps, int timeoutInSecond) {
+		if(isHttps) {
+			return "HTTPS_"+timeoutInSecond;
+		}else {
+			return "HTTP_"+timeoutInSecond;
+		}		
+	}
+
 	/**
 	 * 并发请求复用连接池，避免内存溢出
 	 * @param okHttpClient
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.Map;
+
+import com.taover.util.UtilHttpByOkHttp;
 public class OkHttpDebug {
+	public static void main(String args[]) {
+		//loadRequestPerMinute(10);
+				
+		//testWarePushOrder(100);
+		
+		multiRequest(10);
+	}
+
+	private static void multiRequest(int count) {
+		for(int sequence=0; sequence<count; ++sequence) {
+			new RequestThread(sequence).start();
+		}
+	}
+	
+	private static void testWarePushOrder(int pushCount) {
+		Map<String, Object> params = new HashMap<String, Object>();		
+		for(int i=0; i<pushCount; ++i) {
+			try {
+				if(i%2 == 0) {
+					params.put("wareId", 20);
+				}else {
+					params.put("wareId", 34);
+				}				
+				Thread.sleep(3000);
+				long starTime = System.currentTimeMillis();
+				String response = UtilHttpByOkHttp.sendPostForm("http://localhost/api/schedule/ware/order", null, params);
+				long endTime = System.currentTimeMillis();
+				System.out.println("success["+i+"]["+(endTime-starTime)+"]:"+response);
+			} catch (Exception e) {
+				e.printStackTrace();
+			}
+		}
+	}
+
+	private static void loadRequestPerMinute(int i) {
+		double gapSecond = 60.0/i;
+		int sequence = 0;
+		while(true) {
+			++sequence;
+			new RequestThread(sequence).start();
+			try {
+				Thread.sleep((long)(gapSecond*1000));
+			} catch (InterruptedException e) {
+				e.printStackTrace();
+			}
+		}
+	}
+}
+
+
+class RequestThread extends Thread{
+	int sequence = 0;
+	public RequestThread(int sequence) {
+		this.sequence = sequence;
+	}
+	@Override
+	public void run() {
+		try {
+			long starTime = System.currentTimeMillis();
+			UtilHttpByOkHttp.sendGet("http://api.wxorder.taover.com/v1/wxorderorder?page=1&size=100&controlStatus=", Collections.singletonMap("Authorization", "bearer;eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ0ZW5hbnQiOiIxNiIsInVzZXJuYW1lIjoiMTM2MjEwNTEyMzAiLCJ1c2VyaWQiOiIxNyIsImlzcyI6ImFkbWluIiwiYXVkIjoiMDk4ZjZiY2Q0NjIxZDM3M2NhZGU0ZTgzMjYyN2I0ZjYiLCJleHAiOjE2MDI4NDM5MzIsIm5iZiI6MTYwMjY3MTEzMn0.PbXwPqpFYQB7dwAeB9ELr6dho29k_YZCtcVoECeo-wM"), 60);
+			Thread.sleep(2000);
+			long endTime = System.currentTimeMillis();
+			System.out.println("success["+this.sequence+"]:"+(endTime-starTime));
+		} catch (Exception e) {
+			System.out.println("fail");
+		}
+	}
 }
+
	@@ -59,7 +59,7 @@ uploadArchives {		@@ -59,7 +59,7 @@ uploadArchives {
59	authentication(userName: NEXUS_USERNAME, password: NEXUS_PASSWORD)	59	authentication(userName: NEXUS_USERNAME, password: NEXUS_PASSWORD)
60	}	60	}
61	pom.project {	61	pom.project {
62	- version '1.1.112'	62	+ version '1.1.114'
63	artifactId ARTIFACT_Id	63	artifactId ARTIFACT_Id
64	groupId GROUP_ID	64	groupId GROUP_ID
65	packaging TYPE	65	packaging TYPE
		1	+import java.util.Collections;
		2	+import java.util.HashMap;
		3	+import java.util.Map;
		4	+
		5	+import com.taover.util.UtilHttpByOkHttp;
1		6
2	public class OkHttpDebug {	7	public class OkHttpDebug {
		8	+ public static void main(String args[]) {
		9	+ //loadRequestPerMinute(10);
		10	+
		11	+ //testWarePushOrder(100);
		12	+
		13	+ multiRequest(10);
		14	+ }
		15	+
		16	+ private static void multiRequest(int count) {
		17	+ for(int sequence=0; sequence<count; ++sequence) {
		18	+ new RequestThread(sequence).start();
		19	+ }
		20	+ }
		21	+
		22	+ private static void testWarePushOrder(int pushCount) {
		23	+ Map<String, Object> params = new HashMap<String, Object>();
		24	+ for(int i=0; i<pushCount; ++i) {
		25	+ try {
		26	+ if(i%2 == 0) {
		27	+ params.put("wareId", 20);
		28	+ }else {
		29	+ params.put("wareId", 34);
		30	+ }
		31	+ Thread.sleep(3000);
		32	+ long starTime = System.currentTimeMillis();
		33	+ String response = UtilHttpByOkHttp.sendPostForm("http://localhost/api/schedule/ware/order", null, params);
		34	+ long endTime = System.currentTimeMillis();
		35	+ System.out.println("success["+i+"]["+(endTime-starTime)+"]:"+response);
		36	+ } catch (Exception e) {
		37	+ e.printStackTrace();
		38	+ }
		39	+ }
		40	+ }
		41	+
		42	+ private static void loadRequestPerMinute(int i) {
		43	+ double gapSecond = 60.0/i;
		44	+ int sequence = 0;
		45	+ while(true) {
		46	+ ++sequence;
		47	+ new RequestThread(sequence).start();
		48	+ try {
		49	+ Thread.sleep((long)(gapSecond*1000));
		50	+ } catch (InterruptedException e) {
		51	+ e.printStackTrace();
		52	+ }
		53	+ }
		54	+ }
		55	+}
		56	+
		57	+
		58	+class RequestThread extends Thread{
		59	+ int sequence = 0;
		60	+ public RequestThread(int sequence) {
		61	+ this.sequence = sequence;
		62	+ }
3		63
		64	+ @Override
		65	+ public void run() {
		66	+ try {
		67	+ long starTime = System.currentTimeMillis();
		68	+ UtilHttpByOkHttp.sendGet("http://api.wxorder.taover.com/v1/wxorderorder?page=1&size=100&controlStatus=", Collections.singletonMap("Authorization", "bearer;eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ0ZW5hbnQiOiIxNiIsInVzZXJuYW1lIjoiMTM2MjEwNTEyMzAiLCJ1c2VyaWQiOiIxNyIsImlzcyI6ImFkbWluIiwiYXVkIjoiMDk4ZjZiY2Q0NjIxZDM3M2NhZGU0ZTgzMjYyN2I0ZjYiLCJleHAiOjE2MDI4NDM5MzIsIm5iZiI6MTYwMjY3MTEzMn0.PbXwPqpFYQB7dwAeB9ELr6dho29k_YZCtcVoECeo-wM"), 60);
		69	+ Thread.sleep(2000);
		70	+ long endTime = System.currentTimeMillis();
		71	+ System.out.println("success["+this.sequence+"]:"+(endTime-starTime));
		72	+ } catch (Exception e) {
		73	+ System.out.println("fail");
		74	+ }
		75	+ }
4	}	76	}
		77	+