|
|
|
@ -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) { |
|
|
|
|