package io.undertow.servlet.core;
import java.io.IOException;
import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import io.undertow.servlet.UndertowServletLogger;
import io.undertow.servlet.UndertowServletMessages;
import io.undertow.servlet.api.DeploymentManager;
import io.undertow.servlet.api.FilterInfo;
import io.undertow.servlet.api.InstanceHandle;
import io.undertow.servlet.spec.FilterConfigImpl;
import io.undertow.servlet.spec.ServletContextImpl;
public class ManagedFilter implements Lifecycle {
private final FilterInfo filterInfo;
private final ServletContextImpl servletContext;
private volatile boolean started = false;
private volatile Filter filter;
private volatile InstanceHandle<? extends Filter> handle;
public ManagedFilter(final FilterInfo filterInfo, final ServletContextImpl servletContext) {
this.filterInfo = filterInfo;
this.servletContext = servletContext;
}
public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
if(servletContext.getDeployment().getDeploymentState() != DeploymentManager.State.STARTED) {
throw UndertowServletMessages.MESSAGES.deploymentStopped(servletContext.getDeployment().getDeploymentInfo().getDeploymentName());
}
if (!started) {
start();
}
getFilter().doFilter(request, response, chain);
}
private Filter getFilter() throws ServletException {
if (filter == null) {
createFilter();
}
return filter;
}
public void createFilter() throws ServletException {
synchronized (this) {
if (filter == null) {
try {
handle = filterInfo.getInstanceFactory().createInstance();
} catch (Exception e) {
throw UndertowServletMessages.MESSAGES.couldNotInstantiateComponent(filterInfo.getName(), e);
}
Filter filter = handle.getInstance();
new LifecyleInterceptorInvocation(servletContext.getDeployment().getDeploymentInfo().getLifecycleInterceptors(), filterInfo, filter, new FilterConfigImpl(filterInfo, servletContext)).proceed();
this.filter = filter;
}
}
}
public synchronized void start() throws ServletException {
if (!started) {
started = true;
}
}
public synchronized void stop() {
started = false;
if (handle != null) {
try {
new LifecyleInterceptorInvocation(servletContext.getDeployment().getDeploymentInfo().getLifecycleInterceptors(), filterInfo, filter).proceed();
} catch (Exception e) {
UndertowServletLogger.ROOT_LOGGER.failedToDestroy(filterInfo, e);
}
handle.release();
}
filter = null;
handle = null;
}
@Override
public boolean isStarted() {
return started;
}
public FilterInfo getFilterInfo() {
return filterInfo;
}
@Override
public String toString() {
return "ManagedFilter{" +
"filterInfo=" + filterInfo +
'}';
}
public void forceInit() throws ServletException {
if (filter == null) {
createFilter();
}
}
}