从 2.1.4 升级到 3.4.3 的 Spring Boot 升级后，ThreadLocal get() 有时会返回 null

Question

这是代码：

public class ContextManagerImpl implements ContextManager {

    private static final ThreadLocal<Context> ctx = new ThreadLocal<Context>();
    @Override
    public Context getContext() {
        if(ctx.get() == null) {
            ctx.set(new Context("", ""));   // Dummy context. This should never happen
        }
        return ctx.get();
    }

    @Override
    public void begin(Context context){
        ctx.set(context);                            // Verified context passed is never null or blank 
    }

    @Override
    public void end() {
        if(ctx!= null) {
            ctx.remove();
        }
    }
}

public final class Context implements Serializable {
    private String s1;
    private String s2;
}

有 2 个线程正在使用具有不同线程本地上下文的此类。它在大多数情况下都能正常工作，但有时即使 begin() 正确设置了 ctx 中的值，getContext() 也会返回虚拟上下文。 我怀疑某处存在导致这种情况的竞争条件，但鉴于 threadlocal set() 和 get() 是线程安全的，并且 ctx 的初始化是在声明期间完成的，这永远不会发生。 注意：我已经将 spring boot 升级到 3.4.3 但仍然使用 JDK 8。

Answer 1

在 Spring Boot 从 2.1.4 升级到 3.4.3 后，ThreadLocal get() 有时会返回 null

如果 Spring 保证在 getContext() 之前调用 begin(...)，那么在我看来，一个线程正在调用 begin(...) 方法，而另一个线程正在调用 getContext()。如果是这种情况，getContext() 将返回 null。

ThreadLocal 旨在为每个线程提供一个不同的上下文。那是你要的吗？我不完全理解你的接线，但我怀疑你使用ThreadLocal。围绕线程的 Spring Boot 保证是什么？

我怀疑您应该改为将其设为 volatile 字段：

 private static volatile Context ctx; // maybe set to new Context("", "");
 // this should be called by spring boot
 @Override
 public Context begin(Context ctx) {
    this.ctx = ctx;
 }
 public Context getContext() {
    return ctx;
 }

如果您担心ContextManagerImpl 的多个实例使用相同的上下文初始化时会出现一些竞争条件，那么您可以改用AtomicReference：

 // may want to initialize with a dummy context
 private static AtomicReference <Context> ctxRef = new AtomicReference();
 // this should be called by spring boot
 @Override
 public Context begin(Context ctx) {
    ctxRef.compareAndSet(null, ctx);
 }
 public Context getContext() {
    return ctxRef.get();
 }