SessionManager.java

/* 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.bunjlabs.fuga.sessions;

import com.bunjlabs.fuga.FugaApp;
import com.bunjlabs.fuga.configuration.Configuration;
import com.bunjlabs.fuga.foundation.Context;
import com.bunjlabs.fuga.foundation.Cookie;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
import java.util.UUID;

public class SessionManager {

    private final Map<UUID, Session> sessions = new HashMap<>();
    private final Configuration configuration;

    /**
     * Create new session manager with specified fuga application.
     *
     * @param app Fuga application.
     */
    public SessionManager(FugaApp app) {
        this.configuration = app.getConfiguration();

    }

    /**
     * Returns session by the session id.
     *
     * @param sessionId Unique session id.
     * @return session by the session id.
     */
    public Session getSession(UUID sessionId) {
        return sessions.get(sessionId);
    }

    /**
     * Returns session by the session id.
     *
     * @param sessionId Unique session id.
     * @return session by the session id.
     */
    public Session getSession(String sessionId) {
        return sessions.get(UUID.fromString(sessionId));
    }

    /**
     * Returns session by the request context.
     *
     * @param ctx Request context.
     * @return session.
     */
    public Session getSession(Context ctx) {
        Optional<Cookie> sessionCookieOpt = ctx.request().cookie(configuration.get("fuga.sessions.cookie"));
        Session session = null;

        if (sessionCookieOpt.isPresent()) {
            session = sessions.get(UUID.fromString(sessionCookieOpt.get().value()));
        }

        if (session == null) {
            UUID sessionId = UUID.randomUUID();
            session = new Session(sessionId);

            sessions.put(sessionId, session);

            ctx.response().cookie(
                    new Cookie(configuration.get("fuga.sessions.cookie"), session.getSessionId().toString())
                    .path("/")
            );
        }
        return session;
    }

    /**
     * Update all available sessions.
     */
    public void update() {
        sessions.entrySet().stream().forEach((e) -> {
            Session session = e.getValue();
            long currentTime = new Date().getTime();
            if (currentTime - session.getUpdateTime() >= session.getTimeout() * 60000) {
                sessions.remove(e.getKey());
            }
        });
    }

}