com.github.mcollovati.vertx.vaadin.VertxWrappedSession Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vertx-vaadin-flow Show documentation
An adapter to run Vaadin Flow applications on Vertx
The newest version!
/*
 * The MIT License
 * Copyright © 2016-2020 Marco Collovati ([email protected])
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
package com.github.mcollovati.vertx.vaadin;

import jakarta.servlet.ServletContext;
import jakarta.servlet.http.HttpSession;
import jakarta.servlet.http.HttpSessionBindingEvent;
import jakarta.servlet.http.HttpSessionBindingListener;
import java.util.Collections;
import java.util.Enumeration;
import java.util.Map;
import java.util.Objects;
import java.util.Optional;
import java.util.Set;

import com.vaadin.flow.server.WrappedSession;
import lombok.AccessLevel;
import lombok.RequiredArgsConstructor;
import lombok.experimental.Delegate;

import com.github.mcollovati.vertx.web.ExtendedSession;

import static java.util.stream.Collectors.toMap;

/**
 * Created by marco on 16/07/16.
 */
@RequiredArgsConstructor(access = AccessLevel.PUBLIC)
public class VertxWrappedSession implements WrappedSession {

    private final ExtendedSession delegate;

    public ExtendedSession getVertxSession() {
        return delegate;
    }

    @Override
    public int getMaxInactiveInterval() {
        return Long.valueOf(delegate.timeout()).intValue() / 1000;
    }

    @Override
    public void setMaxInactiveInterval(int interval) {
        throw new UnsupportedOperationException("Setting max interval is not supported on Vert.x session");
    }

    @Override
    public Object getAttribute(String name) {
        checkSessionState();
        return delegate.get(name);
    }

    @Override
    // TODO: fire HttpSessionAttributeListener
    public void setAttribute(String name, Object value) {
        checkSessionState();
        if (value == null) {
            removeAttribute(name);
        } else {
            tryCastHttpSessionBindingListener(delegate.get(name))
                    .ifPresent(oldValue -> oldValue.valueUnbound(createHttpSessionBindingEvent(name, oldValue)));
            delegate.put(name, value);
            tryCastHttpSessionBindingListener(value)
                    .ifPresent(listener -> listener.valueBound(createHttpSessionBindingEvent(name, value)));
        }
    }

    @Override
    public Set getAttributeNames() {
        checkSessionState();
        return delegate.data().keySet();
    }

    @Override
    // TODO: catch HttpSessionBindingListener exceptions?
    public void invalidate() {
        checkSessionState();
        Map toUnbind = delegate.data().entrySet().stream()
                .filter(entry -> HttpSessionBindingListener.class.isInstance(entry.getValue()))
                .collect(toMap(Map.Entry::getKey, e -> HttpSessionBindingListener.class.cast(e.getValue())));
        delegate.destroy();
        toUnbind.forEach((name, listener) -> listener.valueUnbound(createHttpSessionBindingEvent(name, listener)));
        toUnbind.clear();
    }

    @Override
    public String getId() {
        return delegate.id();
    }

    @Override
    public long getCreationTime() {
        return delegate.createdAt();
    }

    @Override
    public long getLastAccessedTime() {
        checkSessionState();
        return delegate.lastAccessed();
    }

    @Override
    public boolean isNew() {
        checkSessionState();
        return delegate.lastAccessed() == 0L;
    }

    @Override
    public void removeAttribute(String name) {
        checkSessionState();
        tryCastHttpSessionBindingListener(delegate.remove(name))
                .ifPresent(oldValue -> oldValue.valueUnbound(createHttpSessionBindingEvent(name, oldValue)));
    }

    private void checkSessionState() {
        if (delegate.isDestroyed()) {
            throw new IllegalStateException("Session already invalidated");
        }
    }

    private Optional tryCastHttpSessionBindingListener(Object value) {
        return Optional.ofNullable(value)
                .filter(HttpSessionBindingListener.class::isInstance)
                .map(HttpSessionBindingListener.class::cast);
    }

    private HttpSessionBindingEvent createHttpSessionBindingEvent(String name, Object value) {
        return new HttpSessionBindingEvent(new VertxHttpSession(this), name, value);
    }
}

class VertxHttpSession implements HttpSession {

    @Delegate(excludes = Exclusions.class)
    private VertxWrappedSession delegate;

    VertxHttpSession(ExtendedSession session) {
        this(new VertxWrappedSession(Objects.requireNonNull(session)));
    }

    public VertxHttpSession(VertxWrappedSession session) {
        this.delegate = Objects.requireNonNull(session);
    }

    @Override
    public ServletContext getServletContext() {
        return null;
    }

    @Override
    public Enumeration getAttributeNames() {
        return Collections.enumeration(delegate.getAttributeNames());
    }

    private interface Exclusions {
        Set getAttributeNames();
    }
}