Spring @async 子线程上下文多租户

Question

我有一个应用程序，它向我的 java 应用程序发送一些 REST 调用。我必须在 @Async 模式下运行最后一个命令，以便用户可以在有异步任务执行另一个 SOAP 调用时继续使用应用程序。

问题是我已经在数据库中自动装配了 TenantContext 和几个租户标识符。在执行异步任务时，它具有我的主线程的租户上下文，并为错误的租户保存数据。

这是我的 JpaTransactionManager，它为数据库中的每个事务调用：

@Autowired
private TenantContext tenantContext;

@Autowired
private Flyway flyway;

@Autowired
private Environment environment;

@Override
protected void doBegin(final Object transaction, final TransactionDefinition definition)
{
    super.doBegin(transaction, definition);
    final EntityManagerHolder entityManagerHolder = (EntityManagerHolder) TransactionSynchronizationManager.getResource(getEntityManagerFactory());
    final EntityManager entityManager = entityManagerHolder.getEntityManager();

    String subAccountId = "";
    if (environment.getActiveProfiles().length == 0 || !environment.getActiveProfiles()[0].equals("production"))
    {
        subAccountId = "SCAN4CLOUD";
    } else
    {
        subAccountId = tenantContext.getTenant().getAccount().getId().toUpperCase();
    }

    entityManager.setProperty("tenant", subAccountId);
}

我尝试使用以下类拦截异步调用以设置正确的租户上下文。

@Configuration
@EnableAsync
public class AsyncConfig extends AsyncConfigurerSupport
{
@Autowired
private TenantContext tenantContext;

@Autowired
private Environment environment;

private HashMap<String, ContextAwarePoolExecutor> tenantThreadPoolTaskExecutor = new HashMap<String, ContextAwarePoolExecutor>();

@Override
@Bean
@Scope(value = "request", proxyMode = ScopedProxyMode.TARGET_CLASS)
public Executor getAsyncExecutor()
{

    String subAccountId = "";

    if (environment.getActiveProfiles().length == 0 || !environment.getActiveProfiles()[0].equals("production"))
    {
        subAccountId = "SCAN4CLOUD";
    } else
    {
        subAccountId = tenantContext.getTenant().getAccount().getId().toUpperCase();
    }

    if (!tenantThreadPoolTaskExecutor.containsKey(subAccountId))
    {
        tenantThreadPoolTaskExecutor.put(subAccountId, new ContextAwarePoolExecutor(tenantContext));
    }

    return tenantThreadPoolTaskExecutor.get(subAccountId);
}

}

ContextAwarePoolExecutor：

public class ContextAwarePoolExecutor extends ThreadPoolTaskExecutor
{
    private TenantContext tenantContext;

    public ContextAwarePoolExecutor(TenantContext tenantContext)
    {
        this.tenantContext = tenantContext;
    }

    @Override
    public <T> Future<T> submit(Callable<T> task)
    {
        return super.submit(new ContextAwareCallable(task, tenantContext));
    }

    @Override
    public <T> ListenableFuture<T> submitListenable(Callable<T> task)
    {
        return super.submitListenable(new ContextAwareCallable(task, tenantContext));
    }
}

ContextAwareCallable：

public class ContextAwareCallable<T> implements Callable<T>
{
private Callable<T> task;
private TenantContext tenantContext;

public ContextAwareCallable(Callable<T> task, TenantContext tenantContext)
{
    this.task = task;
    this.tenantContext = tenantContext;
}

@Override
public T call() throws Exception
{
    if (tenantContext != null)
    {
        return tenantContext.execute(tenantContext.getTenant().getId(), task);
    }

    return task.call();
    }
}

但它仍然没有给我父线程的正确租户。

对此有何建议或其他方法？

谢谢， 没有Ta

Answer 1

Tenantcontext 类来自 SAP：import com.sap.cloud.account.TenantContext； https://appdoc.app/artifact/com.sap.cloud/neo-java-web-api/1.105.21/com/sap/cloud/account/TenantContext.html

import java.util.concurrent.Callable;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import com.sap.cloud.account.TenantContext;

public class ContextAwareCallable<T> implements Callable<T>
{
    private Callable<T> task;
    private TenantContext tenantContext;
    private String tenantId;
    private RequestAttributes context;

    public ContextAwareCallable(Callable<T> task, TenantContext tenantContext, RequestAttributes context)
    {
        this.task = task;
        this.tenantContext = tenantContext;
        this.context = context;
        if (tenantContext != null)
        {
            this.tenantId = tenantContext.getTenant().getId();
        }
    }

    @Override
    public T call() throws Exception
    {
        if (context != null)
        {
            RequestContextHolder.setRequestAttributes(context);
        }

        if (tenantId != null)
        {
            return tenantContext.execute(tenantId, task);
        }

        return task.call();
    }
}

Answer 2

我通过在 contextawarecallable 构造函数中直接添加tenantid 解决了这个问题。现在它正在工作。