package org.springframework.boot.actuate.endpoint.web;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.Map;
import javax.servlet.Servlet;
import javax.servlet.ServletRegistration.Dynamic;
import org.springframework.beans.BeanUtils;
import org.springframework.util.Assert;
import org.springframework.util.StringUtils;
public final class EndpointServlet {
private final Servlet servlet;
private final Map<String, String> initParameters;
private final int loadOnStartup;
public EndpointServlet(Class<? extends Servlet> servlet) {
this(instantiateClass(servlet));
}
private static Servlet instantiateClass(Class<? extends Servlet> servlet) {
Assert.notNull(servlet, "Servlet must not be null");
return BeanUtils.instantiateClass(servlet);
}
public EndpointServlet(Servlet servlet) {
this(servlet, Collections.emptyMap(), -1);
}
private EndpointServlet(Servlet servlet, Map<String, String> initParameters, int loadOnStartup) {
Assert.notNull(servlet, "Servlet must not be null");
this.servlet = servlet;
this.initParameters = Collections.unmodifiableMap(initParameters);
this.loadOnStartup = loadOnStartup;
}
public EndpointServlet withInitParameter(String name, String value) {
Assert.hasText(name, "Name must not be empty");
return withInitParameters(Collections.singletonMap(name, value));
}
public EndpointServlet withInitParameters(Map<String, String> initParameters) {
Assert.notNull(initParameters, "InitParameters must not be null");
boolean hasEmptyName = initParameters.keySet().stream().anyMatch((name) -> !StringUtils.hasText(name));
Assert.isTrue(!hasEmptyName, "InitParameters must not contain empty names");
Map<String, String> mergedInitParameters = new LinkedHashMap<>(this.initParameters);
mergedInitParameters.putAll(initParameters);
return new EndpointServlet(this.servlet, mergedInitParameters, this.loadOnStartup);
}
public EndpointServlet withLoadOnStartup(int loadOnStartup) {
return new EndpointServlet(this.servlet, this.initParameters, loadOnStartup);
}
Servlet getServlet() {
return this.servlet;
}
Map<String, String> getInitParameters() {
return this.initParameters;
}
int getLoadOnStartup() {
return this.loadOnStartup;
}
}