lywsvip/hsweb-framework
1
0
Fork 0
mirror of https://github.com/hs-web/hsweb-framework.git synced 2026-06-04 11:52:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 优化上下文传递

Browse Source
This commit is contained in:
zhouhao
2025-08-04 13:55:38 +08:00
parent a7bbd59d01
commit 2dbb9c7cc0
4 changed files with 30 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
hsweb-authorization/hsweb-authorization-api/src/main/java/org/hswebframework/web/authorization/ReactiveAuthenticationHolder.java
View File

@@ -104,9 +104,7 @@ public final class ReactiveAuthenticationHolder {
}
Authentication get() {
return auth == null
? AuthenticationHolder.get().orElse(null)
: auth;
return auth;
}
}

10
hsweb-authorization/hsweb-authorization-api/src/main/java/org/hswebframework/web/authorization/context/AuthenticationThreadLocalAccessor.java
View File

@@ -3,13 +3,21 @@ package org.hswebframework.web.authorization.context;
import io.micrometer.context.ThreadLocalAccessor;
import org.hswebframework.web.authorization.Authentication;
import org.hswebframework.web.authorization.AuthenticationHolder;
import org.hswebframework.web.authorization.ReactiveAuthenticationHolder;
import javax.annotation.Nonnull;
public class AuthenticationThreadLocalAccessor implements ThreadLocalAccessor<Authentication> {
public class AuthenticationThreadLocalAccessor
implements ThreadLocalAccessor<Authentication> {
static final String KEY = "cp.hs.auth";
static {
ReactiveAuthenticationHolder.addSupplier(
new ThreadLocalReactiveAuthenticationSupplier()
);
}
@Override
@Nonnull
public Object key() {

18
hsweb-authorization/hsweb-authorization-api/src/main/java/org/hswebframework/web/authorization/context/ThreadLocalReactiveAuthenticationSupplier.java Normal file
View File

@@ -0,0 +1,18 @@
package org.hswebframework.web.authorization.context;
import org.hswebframework.web.authorization.Authentication;
import org.hswebframework.web.authorization.AuthenticationHolder;
import org.hswebframework.web.authorization.ReactiveAuthenticationSupplier;
import reactor.core.publisher.Mono;
class ThreadLocalReactiveAuthenticationSupplier implements ReactiveAuthenticationSupplier {
@Override
public Mono<Authentication> get(String userId) {
return Mono.empty();
}
@Override
public Mono<Authentication> get() {
return Mono.justOrEmpty(AuthenticationHolder.get());
}
}

3
hsweb-authorization/hsweb-authorization-api/src/test/java/org/hswebframework/web/authorization/context/AuthenticationThreadLocalAccessorTest.java
View File

@@ -26,9 +26,10 @@ class AuthenticationThreadLocalAccessorTest {
() -> Authentication
.currentReactive()
.subscribeOn(Schedulers.boundedElastic())
.contextCapture()
.block());
assertEquals(auth,auth2);
assertEquals(auth, auth2);
}
@Test