All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.eclipse.jetty.ee8.servlet.BaseHolder Maven / Gradle / Ivy

There is a newer version: 2.0.31
Show newest version
//
// ========================================================================
// Copyright (c) 1995 Mort Bay Consulting Pty Ltd and others.
//
// This program and the accompanying materials are made available under the
// terms of the Eclipse Public License v. 2.0 which is available at
// https://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
// which is available at https://www.apache.org/licenses/LICENSE-2.0.
//
// SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
// ========================================================================
//
package org.eclipse.jetty.ee8.servlet;

import java.io.IOException;
import java.util.function.BiFunction;
import javax.servlet.ServletContext;
import javax.servlet.UnavailableException;
import org.eclipse.jetty.ee8.nested.ContextHandler;
import org.eclipse.jetty.ee8.nested.ContextHandler.APIContext;
import org.eclipse.jetty.util.Loader;
import org.eclipse.jetty.util.annotation.ManagedAttribute;
import org.eclipse.jetty.util.component.AbstractLifeCycle;
import org.eclipse.jetty.util.component.Dumpable;
import org.eclipse.jetty.util.thread.AutoLock;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * AbstractHolder
 *
 * Base class for all servlet-related classes that may be lazily instantiated  (eg servlet, filter,
 * listener), and/or require metadata to be held regarding their origin
 * (web.xml, annotation, programmatic api etc).
 *
 * @param  the type of holder
 */
public abstract class BaseHolder extends AbstractLifeCycle implements Dumpable {

    private static final Logger LOG = LoggerFactory.getLogger(BaseHolder.class);

    private final AutoLock _lock = new AutoLock();

    private final Source _source;

    private Class _class;

    private String _className;

    private T _instance;

    private ServletHandler _servletHandler;

    protected BaseHolder(Source source) {
        _source = source;
    }

    public Source getSource() {
        return _source;
    }

    AutoLock lock() {
        return _lock.lock();
    }

    boolean lockIsHeldByCurrentThread() {
        return _lock.isHeldByCurrentThread();
    }

    /**
     * Do any setup necessary after starting
     *
     * @throws Exception if unable to initialize
     */
    public void initialize() throws Exception {
        if (!isStarted())
            throw new IllegalStateException("Not started: " + this);
    }

    @SuppressWarnings("unchecked")
    @Override
    public void doStart() throws Exception {
        //if no class already loaded and no classname, make permanently unavailable
        if (_class == null && (_className == null || _className.isEmpty()))
            throw new UnavailableException("No class in holder " + toString());
        //try to load class
        if (_class == null) {
            try {
                _class = Loader.loadClass(_className);
                if (LOG.isDebugEnabled())
                    LOG.debug("Holding {} from {}", _class, _class.getClassLoader());
            } catch (Exception e) {
                LOG.warn("Unable to load class {}", _className, e);
                throw new UnavailableException("Class loading error for holder " + toString());
            }
        }
    }

    @Override
    public void doStop() throws Exception {
        if (_instance == null)
            _class = null;
    }

    @ManagedAttribute(value = "Class Name", readonly = true)
    public String getClassName() {
        return _className;
    }

    public Class getHeldClass() {
        return _class;
    }

    /**
     * @return Returns the servletHandler.
     */
    public ServletHandler getServletHandler() {
        return _servletHandler;
    }

    /**
     * @param servletHandler The {@link ServletHandler} that will handle requests dispatched to this servlet.
     */
    public void setServletHandler(ServletHandler servletHandler) {
        _servletHandler = servletHandler;
    }

    /**
     * @param className The className to set.
     */
    public void setClassName(String className) {
        _className = className;
        _class = null;
    }

    /**
     * @param held The class to hold
     */
    public void setHeldClass(Class held) {
        _class = held;
        if (held != null) {
            _className = held.getName();
        }
    }

    protected void illegalStateIfContextStarted() {
        if (_servletHandler != null) {
            ServletContext context = _servletHandler.getServletContext();
            if ((context instanceof APIContext) && ((APIContext) context).getContextHandler().isStarted())
                throw new IllegalStateException("Started");
        }
    }

    protected void setInstance(T instance) {
        try (AutoLock l = lock()) {
            _instance = instance;
            if (instance == null)
                setHeldClass(null);
            else
                setHeldClass((Class) instance.getClass());
        }
    }

    protected T getInstance() {
        try (AutoLock l = lock()) {
            return _instance;
        }
    }

    protected T createInstance() throws Exception {
        try (AutoLock l = lock()) {
            ServletContext ctx = getServletContext();
            if (ctx == null)
                return getHeldClass().getDeclaredConstructor().newInstance();
            if (ServletContextHandler.ServletAPIContext.class.isAssignableFrom(ctx.getClass()))
                return ((ServletContextHandler.ServletAPIContext) ctx).createInstance(this);
            return null;
        }
    }

    public ServletContext getServletContext() {
        ServletContext scontext = null;
        //try the ServletHandler first
        if (getServletHandler() != null)
            scontext = getServletHandler().getServletContext();
        if (scontext != null)
            return scontext;
        //try the ContextHandler next
        APIContext ctx = ContextHandler.getCurrentContext();
        if (ctx != null) {
            ContextHandler contextHandler = ctx.getContextHandler();
            if (contextHandler != null)
                return contextHandler.getServletContext();
        }
        return null;
    }

    /**
     * @return True if this holder was created for a specific instance.
     */
    public boolean isInstance() {
        try (AutoLock l = lock()) {
            return _instance != null;
        }
    }

    /**
     * Wrap component using component specific Wrapper Function beans.
     *
     * @param component the component to optionally wrap
     * @param wrapperFunctionType the bean class type to look for in the {@link ServletContextHandler}
     * @param function the BiFunction to execute for each {@code wrapperFunctionType} Bean found (passing in the component and component type)
     * @param  the "wrapper function" implementation. (eg: {@code ServletHolder.WrapperFunction} or {@code FilterHolder.WrapperFunction}, etc)
     * @return the component that has passed through all Wrapper Function beans found.
     */
    protected  T wrap(final T component, final Class wrapperFunctionType, final BiFunction function) {
        T ret = component;
        ServletContextHandler contextHandler = getServletHandler().getServletContextHandler();
        if (contextHandler == null) {
            APIContext context = ContextHandler.getCurrentContext();
            contextHandler = (ServletContextHandler) (context == null ? null : context.getContextHandler());
        }
        if (contextHandler != null) {
            for (W wrapperFunction : contextHandler.getBeans(wrapperFunctionType)) {
                ret = function.apply(wrapperFunction, ret);
            }
        }
        return ret;
    }

    protected T unwrap(final T component) {
        T ret = component;
        while (ret instanceof Wrapped) {
            // noinspection unchecked,rawtypes
            ret = (T) ((Wrapped) ret).getWrapped();
        }
        return ret;
    }

    @Override
    public void dump(Appendable out, String indent) throws IOException {
        Dumpable.dumpObject(out, this);
    }

    @Override
    public String dump() {
        return Dumpable.dump(this);
    }

    interface Wrapped {

        C getWrapped();
    }
}




© 2015 - 2025 Weber Informatics LLC | Privacy Policy