Spring Boot内嵌Tomcat如何按URI生成不同前缀的会话ID

需求可行性 #

你的需求完全合理且可实现，核心问题是之前选择的多上下文实现方向走偏了，才会遇到404、动态路径无法匹配的问题，完全不需要拆分Tomcat上下文即可实现需求。

之前方案失效的原因 #

• 多上下文方案不适用：你的路径第一段是动态{customer}变量，本身无法映射到固定路径的Tomcat上下文；且独立上下文会隔离Spring容器，导致控制器、Bean无法共享，额外复杂度极高。
• 手动添加/newapp上下文报404的原因：你通过tomcat.addContext()创建的是全新的空Tomcat上下文，没有注册Spring的DispatcherServlet，也没有加载项目中@Controller的路径映射，请求进入该上下文后找不到对应处理逻辑，自然返回404。
• 动态委托上下文方案实现成本过高：你仅需要差异化会话ID前缀，不需要做上下文级别的会话隔离，完全不需要实现复杂的上下文委托逻辑。

最优实现方案 #

核心思路是复用Spring Boot默认创建的根上下文（保证原有MVC映射完全生效），通过请求标记+自定义会话ID生成器的轻量方案实现前缀差异化，无侵入、性能开销极低。

实现步骤 #

1. 定义ThreadLocal上下文存储当前请求的路径标记，避免参数传递污染Tomcat原生逻辑

public class SessionIdPrefixHolder {
    private static final ThreadLocal<Boolean> NEW_APP_FLAG = ThreadLocal.withInitial(() -> false);

    public static void markNewAppRequest() {
        NEW_APP_FLAG.set(true);
    }

    public static boolean isNewAppRequest() {
        return NEW_APP_FLAG.get();
    }

    public static void clear() {
        NEW_APP_FLAG.remove();
    }
}

2. 自定义Tomcat Valve，在请求处理最前端判断路径是否匹配/{customer}/newapp/**规则，标记到ThreadLocal，请求结束后清除标记避免线程复用导致的标记错乱

import org.apache.catalina.connector.Request;
import org.apache.catalina.connector.Response;
import org.apache.catalina.valves.ValveBase;
import javax.servlet.ServletException;
import java.io.IOException;
import java.util.regex.Pattern;

public class NewAppPathMarkValve extends ValveBase {
    // 匹配第一段为任意customer标识、第二段为newapp的路径
    private static final Pattern NEWAPP_PATH_PATTERN = Pattern.compile("^/[^/]+/newapp(/.*)?$");

    @Override
    public void invoke(Request request, Response response) throws IOException, ServletException {
        try {
            String requestUri = request.getRequestURI();
            if (NEWAPP_PATH_PATTERN.matcher(requestUri).matches()) {
                SessionIdPrefixHolder.markNewAppRequest();
            }
            getNext().invoke(request, response);
        } finally {
            SessionIdPrefixHolder.clear();
        }
    }
}

3. 自定义会话ID生成器，复用Tomcat原生StandardSessionIdGenerator的生成逻辑，仅根据标记添加前缀

import org.apache.catalina.util.StandardSessionIdGenerator;

public class PrefixSessionIdGenerator extends StandardSessionIdGenerator {
    private static final String NEWAPP_PREFIX = "NEWAPP_";

    @Override
    public String generateSessionId(String route) {
        String defaultSessionId = super.generateSessionId(route);
        if (SessionIdPrefixHolder.isNewAppRequest()) {
            return NEWAPP_PREFIX + defaultSessionId;
        }
        return defaultSessionId;
    }
}

4. 将自定义Valve和ID生成器注册到Spring Boot内嵌Tomcat的默认上下文，不需要重写getWebServer方法，只要注册自定义Bean即可

import org.apache.catalina.Manager;
import org.apache.catalina.core.StandardManager;
import org.springframework.boot.web.embedded.tomcat.TomcatServletWebServerFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class TomcatSessionConfig {
    @Bean
    public TomcatServletWebServerFactory tomcatFactory() {
        TomcatServletWebServerFactory factory = new TomcatServletWebServerFactory();
        // 注册路径标记Valve
        factory.addContextValves(new NewAppPathMarkValve());
        // 替换默认上下文的会话ID生成器
        factory.addContextCustomizers(context -> {
            Manager manager = context.getManager();
            if (manager instanceof StandardManager) {
                ((StandardManager) manager).setSessionIdGenerator(new PrefixSessionIdGenerator());
            }
        });
        return factory;
    }
}

方案优势 #

• 零侵入原有业务逻辑：不需要修改现有@Controller的路径映射，不需要调整业务代码，原有接口完全正常运行，不会出现404问题
• 兼容性强：完全复用Tomcat原生会话管理逻辑，会话过期、集群路由、Cookie写入等逻辑不受任何影响
• 性能开销极低：仅做一次正则匹配和ThreadLocal存取，对请求吞吐量几乎无影响
• 维护成本低：代码量少，逻辑清晰，后续调整前缀规则或者匹配路径只需要修改对应逻辑即可

注意事项 #

• 路径匹配规则可以根据实际业务调整，如果newapp段的位置不是固定在第二段，修改正则即可
• ThreadLocal的清除逻辑必须放在finally块中，避免Tomcat线程池复用线程时残留旧标记，导致会话ID前缀生成错误
• 下划线属于Servlet规范允许的会话ID合法字符，不会影响Cookie写入、会话解析的逻辑

内容的提问来源于stack exchange，提问作者Iacopo Pace