All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
me.legrange.services.jetty.JettyComponent Maven / Gradle / Ivy
package me.legrange.services.jetty;
import jakarta.servlet.DispatcherType;
import jakarta.ws.rs.ext.MessageBodyWriter;
import me.legrange.service.Component;
import me.legrange.service.ComponentException;
import me.legrange.service.Service;
import org.eclipse.jetty.server.Handler;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.handler.gzip.GzipHandler;
import org.eclipse.jetty.servlet.ServletContextHandler;
import org.eclipse.jetty.servlet.ServletHolder;
import org.glassfish.jersey.server.ResourceConfig;
import org.glassfish.jersey.server.ServerProperties;
import org.glassfish.jersey.servlet.ServletContainer;
import java.lang.reflect.InvocationTargetException;
import java.util.Collections;
import java.util.EnumSet;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import static java.lang.String.format;
import static me.legrange.log.Log.info;
import static me.legrange.log.Log.warning;
/**
* A component that adds Jetty HTTP functionality.
*
* @author gideon
*/
public class JettyComponent extends Component {
private Server server;
private ServletContextHandler context;
private boolean running = false;
private final Set jerseyProviders = new HashSet();
private final Map>> endpoints = new HashMap<>();
private JettyConfig config;
public JettyComponent(Service service) {
super(service);
}
@Override
public String getName() {
return "jetty";
}
@Override
public void start(JettyConfig config) throws ComponentException {
this.config = config;
if (config.isEnabled()) {
try {
context = makeContext();
server = new Server(config.getPort());
server.setHandler(gzip(context));
server.start();
info("Started Jetty server on port %d", config.getPort());
} catch (Exception ex) {
throw new ComponentException(ex.getMessage(), ex);
}
} else {
warning("Jetty server disabled");
}
}
@Override
public void stop() throws ComponentException {
if (config.isEnabled()) {
try {
server.stop();
} catch (Exception e) {
throw new ComponentException(e.getMessage(), e);
}
}
}
public void addEndpoints(String path, Set> endpoints) throws ComponentException {
ResourceConfig rc = new ResourceConfig(endpoints);
rc.addProperties(Collections.singletonMap("jersey.config.server.wadl.disableWadl", "true"));
rc.addProperties(Collections.singletonMap(ServerProperties.OUTBOUND_CONTENT_LENGTH_BUFFER, 0));
for (Object provider : jerseyProviders) {
rc.register(provider);
}
ServletHolder holder = new ServletHolder(new ServletContainer(rc));
context.addServlet(holder, path + "/*");
try {
context.start();
} catch (Exception ex) {
throw new ComponentException(ex.getMessage(), ex);
}
if (this.endpoints.containsKey(path)) {
info("Re-added %d endpoints on path %s", endpoints.size(), path);
} else {
info("Added %d new endpoint of type on path %s", endpoints.size(), path);
this.endpoints.put(path, endpoints);
}
running = true;
checkForMessageBodyWriter();
}
/**
* Add a new endpoint with the given path and endpoint class.
*
* @param path The path
* @param endpoint The endpoint class
*/
public void addEndpoint(String path, Class endpoint) throws ComponentException {
ResourceConfig rc = new ResourceConfig(endpoint);
checkForMessageBodyWriter();
for (Object provider : jerseyProviders) {
rc.register(provider);
}
ServletHolder holder = new ServletHolder(new ServletContainer(rc));
context.addServlet(holder, path + "/*");
try {
context.start();
} catch (Exception ex) {
throw new ComponentException(ex.getMessage(), ex);
}
if (endpoints.containsKey(path)) {
info("Re-added endpoint of type '%s' on '%s'", endpoint.getSimpleName(), path);
} else {
info("Added new endpoint of type '%s' on '%s'", endpoint.getSimpleName(), path);
endpoints.put(path, Collections.singleton(endpoint));
}
running = true;
}
public void addProvider(Class provider) throws ComponentException {
try {
addProvider(provider.getConstructor().newInstance());
} catch (NoSuchMethodException e) {
throw new ComponentException(format("No default constructor for provider class %s. Either pass an instance or a provider with a default constructor", provider.getSimpleName()), e);
} catch (IllegalAccessException | InvocationTargetException | InstantiationException e) {
throw new ComponentException(format("Error creating provider from class %s (%s)", provider.getSimpleName(), e.getMessage()), e);
}
}
public void addProvider(Object provider) throws ComponentException {
if (MessageBodyWriter.class.isInstance(provider)) {
Optional mbr = findMessageBodyWriter();
if (mbr.isPresent()) {
jerseyProviders.remove(mbr.get());
warning("Replacing message body writer '%s' with '%s'",
mbr.get().getClass().getSimpleName(),
provider.getClass().getName());
}
}
jerseyProviders.add(provider);
if (running) {
info("Re-adding %d endpoint(s) because of provider change", endpoints.size());
context = makeContext();
try {
server.stop();
server.setHandler(gzip(context));
server.start();
} catch (Exception ex) {
throw new ComponentException(ex.getMessage(), ex);
}
for (Map.Entry>> pair : endpoints.entrySet()) {
addEndpoints(pair.getKey(), pair.getValue());
}
}
}
/**
* Check for a MessageBodyWriter
*/
private void checkForMessageBodyWriter() throws ComponentException {
if (findMessageBodyWriter().isEmpty()) {
throw new ComponentException("No MessageBodyWriters were registered for serialization. Remember to register them using the addProvider methods");
}
}
private Optional findMessageBodyWriter() {
return jerseyProviders.stream()
.filter(p -> MessageBodyWriter.class.isInstance(p))
.map(p -> (MessageBodyWriter) p)
.findFirst();
}
private ServletContextHandler makeContext() {
context = new ServletContextHandler(ServletContextHandler.SESSIONS);
context.setContextPath("/");
if (config.isStandardErrorFilter()) {
context.addFilter(ErrorFilter.class, "/*", EnumSet.allOf(DispatcherType.class));
}
return context;
}
private Handler gzip(Handler context) {
if (config.isEnableGzip()) {
GzipHandler handler = new GzipHandler();
handler.setIncludedMethods("PUT", "POST", "GET");
handler.setInflateBufferSize(2048);
handler.setHandler(context);
return handler;
}
return context;
}
}
