注册新的undertow SessionManager

Question

我有wildfly 8.1服务器运行。我有我自己的SessionManager实现io.undertow.server.session.SessionManager。我想配置系统使用我的会话管理器。

我应该在哪里以及如何配置/添加我的会话管理器的新设置?

Answer 1

public class StartupBeanExtension implements Extension, ServletExtension {
    @Override
    public void handleDeployment(DeploymentInfo deployment, ServletContext context) {
        boolean sessionPersistenceEnabled = Boolean.parseBoolean(BeanUtils.getBean(PropertyResolver.class).getValue("UAM.SessionPersistenceEnabled"));
        if (sessionPersistenceEnabled) {
            System.out.println("Overriding default InMemorySessionManager...[" + deployment.getDeploymentName() + ", " + deployment.getDisplayName() + "]");
            deployment.setSessionManagerFactory(new UAMSessionManagerFactory());
        } else {
            System.out.println("InMemorySessionManager IS NOT OVERIDED!");
        }
    }    
}
public class UAMSessionManagerFactory implements SessionManagerFactory {
    @Override
    public SessionManager createSessionManager(Deployment deployment) {
        UAMSessionManager ss = new UAMSessionManager(deployment.getDeploymentInfo().getDeploymentName());
        return ss;
    }
}
public class UAMSessionManager extends InMemorySessionManager {
    public UAMSessionManager(String deploymentName) {
        super(deploymentName);
        UAMSessionListener uamSessionListener = new UAMSessionListener();
        super.registerSessionListener(uamSessionListener);
        System.out.println("New session manager created. Listener activated.");
    }
    // create session
    public Session createSession(final HttpServerExchange serverExchange, final SessionConfig config, String sessionID) {
        config.setSessionId(serverExchange, sessionID);
        Session session = super.createSession(serverExchange, config);
        return session;
    }
    // get session
    public Session getSession(final HttpServerExchange serverExchange, final SessionConfig config) {
        final String sessionId = config.findSessionId(serverExchange);
        Session session = getSession(sessionId);
        if (session == null) {
            // DO SOMETHING TO CREATE SESSION OR RESTORE IT FROM DB
            try {
                UAMService uam = getUAMService();
                if (uam != null) {
                    Sessions storedSession = uam.getSession(sessionId);
                    if (storedSession != null) {
                        String storedSessionId = storedSession.getSessionId();
                        // create new session with storedSessionID
                        session = createSession(serverExchange, config, storedSessionId);
                        // SET session attributes if needed from storedSession to new one
                    } else {
                        // let InMemorySessionManager create new session
                        return null;
                    }
                }
            } catch (Exception ex) {
            }
        }
        return session;
    }
}
public class UAMSessionListener implements SessionListener {
    @Override
    public void sessionCreated(Session session, HttpServerExchange exchange) {
    }
    @Override
    public void sessionDestroyed(Session session, HttpServerExchange exchange, SessionDestroyedReason reason) {
    }
    @Override
    public void attributeAdded(Session session, String name, Object value) {
        UAMService uamService = getUAMService();
        if (uamService != null) {
            Sessions storedSession = uamService.getSession(session.getId());
            boolean isNew = false;
            if (storedSession == null) {
                storedSession = new Sessions();
                storedSession.setSessionId(session.getId());
                storedSession.setActvityDate(new Date());
                isNew = true;
            }
            // STORE SOME INFO FROM value and update/create it in storage
            uamService.updateSession(storedSession, isNew);
        }
    }
    @Override
    public void attributeUpdated(Session session, String name, Object newValue, Object oldValue) {
    }
    @Override
    public void attributeRemoved(Session session, String name, Object oldValue) {
    }
    @Override
    public void sessionIdChanged(Session session, String oldSessionId) {
    }
}

要用另一个SessionManager覆盖默认的InMemmorySessionManager，应该执行以下步骤:

1. 开发SessionManager，实现io.undertow.server.session.SessionManager
2. 开发SessionManagerFactory，实现io. below .servlet.api.SessionManagerFactory
3. 开发启动扩展，实现io.undertow.servlet。在handleDeployment(Deployment)方法中用新的sessionManagerFactory更改sessionManagerFactory。
4. 通过添加./META-INF/services/io.undertow.servlet来注册新的servleextension。ServletExtension文件(文件应包含新ServletExtension的名称)。例如com.my.utils.StartupExtension)