feat: 增加新的ContextHolder上下文支持

Author: zhou-hao

hsweb-authorization/hsweb-authorization-api/src/main/java/org/hswebframework/web/authorization/token/UserTokenHolder.java

@@ -1,6 +1,7 @@
 package org.hswebframework.web.authorization.token;
 
 
+import org.hswebframework.web.context.ContextHolder;
 import org.hswebframework.web.context.ContextUtils;
 
 /**
@@ -12,7 +13,7 @@ private UserTokenHolder() {
     }
 
     public static UserToken currentToken() {
-        return ContextUtils.currentContext().get(UserToken.class).orElse(null);
+        return ContextHolder.current().getOrDefault(UserToken.class,null);
     }
 
     public static UserToken setCurrent(UserToken token) {

hsweb-core/src/main/java/org/hswebframework/web/context/Context.java

@@ -4,6 +4,7 @@
 import java.util.Optional;
 import java.util.function.Supplier;
 
+@Deprecated
 public interface Context {
 
     default <T> Optional<T> get(Class<T> key) {

hsweb-core/src/main/java/org/hswebframework/web/context/ContextHolder.java

@@ -0,0 +1,78 @@
+package org.hswebframework.web.context;
+
+import lombok.SneakyThrows;
+import reactor.core.publisher.Mono;
+import reactor.util.context.Context;
+import reactor.util.context.ContextView;
+
+import java.io.Closeable;
+import java.lang.reflect.UndeclaredThrowableException;
+import java.util.Collections;
+import java.util.Comparator;
+import java.util.List;
+import java.util.ServiceLoader;
+import java.util.concurrent.Callable;
+import java.util.function.Function;
+import java.util.stream.Collectors;
+
+public class ContextHolder {
+
+    private static final List<ContextHolderSupport> supports;
+
+    static {
+        supports = ServiceLoader
+            .load(ContextHolderSupport.class)
+            .stream()
+            .map(ServiceLoader.Provider::get)
+            .collect(Collectors.toList());
+        supports.add(new ThreadLocalContextHolderSupport());
+        supports.sort(Comparator.comparingInt(ContextHolderSupport::order));
+    }
+
+    public static Closeable makeCurrent(ContextView context) {
+        for (ContextHolderSupport support : supports) {
+            if (support.isSupport()) {
+                return support.makeCurrent(context);
+            }
+        }
+        throw new UnsupportedOperationException();
+
+    }
+
+    @SneakyThrows
+    public static <T> T doInContext(Context context, Callable<T> call) {
+        try (Closeable ignore = makeCurrent(context)) {
+            return call.call();
+        } catch (UndeclaredThrowableException e) {
+            throw e.getCause();
+        }
+    }
+
+    public static <T> Mono<T> currentReactive(Function<ContextView, Mono<T>> handler) {
+        return Mono.deferContextual(ctx -> handler.apply(current().putAll(ctx)));
+    }
+
+    public static Context current() {
+        for (ContextHolderSupport support : supports) {
+            if (support.isSupport()) {
+                return support.current();
+            }
+        }
+        throw new UnsupportedOperationException();
+    }
+
+    public interface ContextHolderSupport {
+        boolean isSupport();
+
+        Closeable makeCurrent(ContextView context);
+
+        void clean();
+
+        Context current();
+
+        default int order() {
+            return Integer.MIN_VALUE;
+        }
+    }
+
+}

hsweb-core/src/main/java/org/hswebframework/web/context/ThreadLocalContextHolderSupport.java

@@ -0,0 +1,45 @@
+package org.hswebframework.web.context;
+
+import reactor.util.context.Context;
+import reactor.util.context.ContextView;
+
+import java.io.Closeable;
+
+/**
+ * 基于 ThreadLocal 的上下文持有器支持实现
+ * 适用于传统平台线程环境
+ */
+public class ThreadLocalContextHolderSupport implements ContextHolder.ContextHolderSupport {
+
+    private static final ThreadLocal<Context> contextHolder = ThreadLocal.withInitial(Context::empty);
+
+    @Override
+    public boolean isSupport() {
+        return true;
+    }
+
+    @Override
+    public Closeable makeCurrent(ContextView context) {
+        Context previous = contextHolder.get();
+        Context newContext = previous.putAll(context);
+        contextHolder.set(newContext);
+
+        return () -> contextHolder.set(previous);
+    }
+
+    @Override
+    public void clean() {
+        contextHolder.remove();
+    }
+
+    @Override
+    public Context current() {
+        Context context = contextHolder.get();
+        return context != null ? context : Context.empty();
+    }
+
+    @Override
+    public int order() {
+        return Integer.MAX_VALUE; // 最低优先级，作为回退选项
+    }
+}