自定义Tomcat的session管理器代替StandardManager

最新推荐文章于 2023-07-17 19:28:36 发布

FarawayHighAway

最新推荐文章于 2023-07-17 19:28:36 发布

阅读量1.7k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： java日记文章标签： Tomcat session StandardManager

本文链接：https://blog.youkuaiyun.com/l18767118724/article/details/83617322

java日记专栏收录该内容

10 篇文章

订阅专栏

本文深入探讨了如何在Spring Boot中自定义Session管理器，通过继承LightSessionManager类，实现了一个不进行Session验证且不持久化的Session管理方案。此管理器在找不到Session时会自动生成，后续可通过Redis等外部服务验证Session的有效性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

首先要理解EmbeddedServletContainerAutoConfiguration 配置的过程

请查看

https://blog.youkuaiyun.com/l18767118724/article/details/83589173

这里就是要加深对 WebSocketContainerCustomizer这个类的认识。

这个类会在 org.springframework.boot.context.embedded.tomcat.TomcatEmbeddedServletContainerFactory#configureContext方法被调用前使用。根据这个流程，可以使用EmbeddedServletContainerAutoConfiguration中的创建的TomcatEmbeddedServletContainerFactory对象进行有效的操作。如下是修改session管理器。tomcat默认的是StandardManager

现在自定义自己的LightSessionManager。这个管理器不校验session，不持久化，只是生成。

@Configuration
@AutoConfigureAfter(EmbeddedServletContainerAutoConfiguration.class)
public class SessionManagerConfiguration {


    @Bean
    public SessionManagerCustomizer sessionManagerCustomizer() {
        return new SessionManagerCustomizer();
    }


    class SessionManagerCustomizer extends WebSocketContainerCustomizer<TomcatEmbeddedServletContainerFactory> {

        @Override
        protected void doCustomize(TomcatEmbeddedServletContainerFactory container) {
            container.addContextCustomizers(new TomcatContextCustomizer() {
                @Override
                public void customize(Context context) {
                    context.setManager(new LightSessionManager());
                }
            });
        }
    }

    /**
     * 重写session管理，不持久。如果没有则生成一个session。后面通过redis校验session是否有效。
     */
    class LightSessionManager extends ManagerBase {
        @Override
        public Session findSession(String id) throws IOException {
            Session session = super.findSession(id);
            if (Objects.isNull(session)) {
                session = super.createSession(id);
            }
            return session;
        }

        @Override
        public void load() {

        }

        @Override
        public void unload() {

        }

        @Override
        protected void startInternal() throws LifecycleException {
            super.startInternal();
            this.setState(LifecycleState.STARTING);
        }

        @Override
        protected void stopInternal() throws LifecycleException {
            this.setState(LifecycleState.STOPPING);
            super.stopInternal();
        }
    }
}