cd.connect.spring.jersey.JerseyModule Maven / Gradle / Ivy

Go to download
package cd.connect.spring.jersey;


import cd.connect.spring.servlet.ServletModule;
import org.glassfish.jersey.server.ResourceConfig;
import org.glassfish.jersey.servlet.ServletContainer;
import org.glassfish.jersey.servlet.ServletProperties;
import org.springframework.context.ApplicationContext;

import javax.servlet.ServletContext;
import java.util.ServiceLoader;
import java.util.stream.Stream;

/**
 * Loads the predefined (if it exists) service class for the JerseyApplications and
 * falls back to our own if it doesn't.
 *
 * @author Richard Vowles - https://plus.google.com/+RichardVowles
 */
abstract public class JerseyModule extends ServletModule {
	@Override
	protected void postProcess(ServletContext servletContext, ApplicationContext ctx) {
		JerseyApplication jerseyResourceConfig = null;

		for(JerseyApplication app : ServiceLoader.load(JerseyApplication.class)) {
			if (jerseyResourceConfig != null) {
				throw new RuntimeException("Too many possible JerseyApplications, please include the dependency of one, not > 1.");
			} else if (!(app instanceof ResourceConfig)) {
				throw new RuntimeException(String.format("JerseyApplication `{}` is not an instance of ResourceConfig.", app.getClass().getName()));
			} else {
				jerseyResourceConfig = app;
			}
		}

		if (jerseyResourceConfig == null) {
			jerseyResourceConfig = new JerseyApplicationBase();
		}

		jerseyResourceConfig.init(ctx, registerResources());

		ResourceConfig base = ResourceConfig.class.cast(jerseyResourceConfig);

		servlet(new ServletContainer(base), s ->
			s.priority(100).name(this.getClass().getName() + "-servlet")
				.url(getUrlServletPrefix())
				.param(ServletProperties.FILTER_CONTEXT_PATH, getUrlServletPrefix()));
	}

	abstract protected Stream> registerResources();

	abstract protected String getUrlServletPrefix();
}