|
@@ -20,7 +20,7 @@ import io.netty.util.CharsetUtil;
|
|
|
import org.slf4j.Logger;
|
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
|
|
-import java.util.concurrent.TimeUnit;
|
|
|
+import java.util.concurrent.*;
|
|
|
|
|
|
/**
|
|
|
* Copy from : https://github.com/xuxueli/xxl-rpc
|
|
@@ -43,6 +43,25 @@ public class EmbedServer {
|
|
|
// param
|
|
|
EventLoopGroup bossGroup = new NioEventLoopGroup();
|
|
|
EventLoopGroup workerGroup = new NioEventLoopGroup();
|
|
|
+ ThreadPoolExecutor bizThreadPool = new ThreadPoolExecutor(
|
|
|
+ 0,
|
|
|
+ 200,
|
|
|
+ 60L,
|
|
|
+ TimeUnit.SECONDS,
|
|
|
+ new LinkedBlockingQueue<Runnable>(2000),
|
|
|
+ new ThreadFactory() {
|
|
|
+ @Override
|
|
|
+ public Thread newThread(Runnable r) {
|
|
|
+ return new Thread(r, "xxl-rpc, EmbedServer bizThreadPool-" + r.hashCode());
|
|
|
+ }
|
|
|
+ },
|
|
|
+ new RejectedExecutionHandler() {
|
|
|
+ @Override
|
|
|
+ public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
|
|
|
+ throw new RuntimeException("xxl-job, EmbedServer bizThreadPool is EXHAUSTED!");
|
|
|
+ }
|
|
|
+ });
|
|
|
+
|
|
|
|
|
|
try {
|
|
|
// start server
|
|
@@ -56,7 +75,7 @@ public class EmbedServer {
|
|
|
.addLast(new IdleStateHandler(0, 0, 30 * 3, TimeUnit.SECONDS)) // beat 3N, close if idle
|
|
|
.addLast(new HttpServerCodec())
|
|
|
.addLast(new HttpObjectAggregator(5 * 1024 * 1024)) // merge request & reponse to FULL
|
|
|
- .addLast(new EmbedHttpServerHandler(executorBiz, accessToken));
|
|
|
+ .addLast(new EmbedHttpServerHandler(executorBiz, accessToken, bizThreadPool));
|
|
|
}
|
|
|
})
|
|
|
.childOption(ChannelOption.SO_KEEPALIVE, true);
|
|
@@ -121,9 +140,11 @@ public class EmbedServer {
|
|
|
|
|
|
private ExecutorBiz executorBiz;
|
|
|
private String accessToken;
|
|
|
- public EmbedHttpServerHandler(ExecutorBiz executorBiz, String accessToken) {
|
|
|
+ private ThreadPoolExecutor bizThreadPool;
|
|
|
+ public EmbedHttpServerHandler(ExecutorBiz executorBiz, String accessToken, ThreadPoolExecutor bizThreadPool) {
|
|
|
this.executorBiz = executorBiz;
|
|
|
this.accessToken = accessToken;
|
|
|
+ this.bizThreadPool = bizThreadPool;
|
|
|
}
|
|
|
|
|
|
@Override
|
|
@@ -137,14 +158,20 @@ public class EmbedServer {
|
|
|
boolean keepAlive = HttpUtil.isKeepAlive(msg);
|
|
|
String accessTokenReq = msg.headers().get(XxlJobRemotingUtil.XXL_JOB_ACCESS_TOKEN);
|
|
|
|
|
|
- // do invoke
|
|
|
- Object responseObj = process(httpMethod, uri, requestData, accessTokenReq);
|
|
|
+ // invoke
|
|
|
+ bizThreadPool.execute(new Runnable() {
|
|
|
+ @Override
|
|
|
+ public void run() {
|
|
|
+ // do invoke
|
|
|
+ Object responseObj = process(httpMethod, uri, requestData, accessTokenReq);
|
|
|
|
|
|
- // to json
|
|
|
- String responseJson = GsonTool.toJson(responseObj);
|
|
|
+ // to json
|
|
|
+ String responseJson = GsonTool.toJson(responseObj);
|
|
|
|
|
|
- // write response
|
|
|
- writeResponse(ctx, keepAlive, responseJson);
|
|
|
+ // write response
|
|
|
+ writeResponse(ctx, keepAlive, responseJson);
|
|
|
+ }
|
|
|
+ });
|
|
|
}
|
|
|
|
|
|
private Object process(HttpMethod httpMethod, String uri, String requestData, String accessTokenReq) {
|