gtsoft/gtsoft-snail-job-server
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects 1 Releases Wiki Activity

feat:(grpc): 修复reqId丢失问题

Browse Source
This commit is contained in:
opensnail 2024-10-24 10:08:13 +08:00
parent 94a1329220
commit 7a199ae21a
1 changed files with 1 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/rpc/server/GrpcRequestHandlerActor.java
View File

@ -77,6 +77,7 @@ public class GrpcRequestHandlerActor extends AbstractActor {
String body = grpcSnailJobRequest.getBody();
Object[] objects = JsonUtil.parseObject(body, Object[].class);
request.setArgs(objects);
request.setReqId(grpcSnailJobRequest.getReqId());
snailJobRpcResult = doProcess(uri, JsonUtil.toJsonString(request), headersMap);
if (Objects.isNull(snailJobRpcResult)) {
snailJobRpcResult = new SnailJobRpcResult(StatusEnum.NO.getStatus(), "服务端异常", null,
@ -147,19 +148,5 @@ public class GrpcRequestHandlerActor extends AbstractActor {
throw new SnailJobServerException("No matching handler found. Path:[{}] method:[{}]", builder.getPathStr());
}
/**
* write response
*/
private void writeResponse(ChannelHandlerContext ctx, boolean keepAlive, String responseJson) {
FullHttpResponse response = new DefaultFullHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.OK,
Unpooled.copiedBuffer(responseJson, CharsetUtil.UTF_8));
response.headers().set(HttpHeaderNames.CONTENT_TYPE,
HttpHeaderValues.APPLICATION_JSON);
response.headers().set(HttpHeaderNames.CONTENT_LENGTH, response.content().readableBytes());
if (keepAlive) {
response.headers().set(HttpHeaderNames.CONNECTION, HttpHeaderValues.KEEP_ALIVE);
}
ctx.writeAndFlush(response);
}
}