lywsvip/hsweb-framework
1
0
Fork 0
mirror of https://github.com/hs-web/hsweb-framework.git synced 2026-06-05 20:33:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 优化上下文逻辑

Browse Source
This commit is contained in:
zhouhao
2023-07-04 18:19:41 +08:00
parent c5fd2faad7
commit 08eeaacdbf
1 changed files with 9 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
hsweb-logging/hsweb-access-logging-aop/src/main/java/org/hswebframework/web/logging/aop/ReactiveAopAccessLoggerSupport.java
View File

@@ -67,27 +67,23 @@ public class ReactiveAopAccessLoggerSupport extends StaticMethodMatcherPointcutA
});
}
private Mono<RequestInfo> currentRequestInfo() {
return Mono
.deferContextual(context -> {
if (context.hasKey(RequestInfo.class)) {
RequestInfo info = context.get(RequestInfo.class);
ReactiveLogger.log(context, ctx -> info.setContext(new HashMap<>(ctx)));
return Mono.just(info);
}
return Mono.empty();
});
private Mono<RequestInfo> currentRequestInfo(ContextView context) {
if (context.hasKey(RequestInfo.class)) {
RequestInfo info = context.get(RequestInfo.class);
ReactiveLogger.log(context, ctx -> info.setContext(new HashMap<>(ctx)));
return Mono.just(info);
}
return Mono.empty();
}
protected Flux<?> wrapFluxResponse(Flux<?> flux, AccessLoggerInfo loggerInfo) {
return Flux.deferContextual(ctx -> this
.currentRequestInfo()
.currentRequestInfo(ctx)
.doOnNext(loggerInfo::putAccessInfo)
.then(beforeRequest(loggerInfo))
.thenMany(flux)
.doOnError(loggerInfo::setException)
.doFinally(signal -> completeRequest(loggerInfo, ctx)))
.contextWrite(ReactiveLogger.start("accessLogId", loggerInfo.getId()));
.doFinally(signal -> completeRequest(loggerInfo, ctx)));
}
private Mono<Void> beforeRequest(AccessLoggerInfo loggerInfo) {