Allow library user to configure Jetty directly

logzio · Nov 21, 2018 · 5604bd2 · 5604bd2
1 parent f387bf7
commit 5604bd2
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/main/java/io/logz/guice/jersey/JerseyModule.java b/src/main/java/io/logz/guice/jersey/JerseyModule.java
@@ -5,22 +5,30 @@
 import com.google.inject.Provider;
 import com.google.inject.servlet.ServletModule;
 import io.logz.guice.jersey.configuration.JerseyConfiguration;
+import org.eclipse.jetty.server.Server;
 
 import java.util.Objects;
+import java.util.function.Consumer;
 
 public class JerseyModule extends AbstractModule {
 
     private final JerseyConfiguration jerseyConfiguration;
+    private Consumer<Server> jettyConfigurer;
 
     public JerseyModule(JerseyConfiguration jerseyConfiguration) {
+        this(jerseyConfiguration, server -> {});
+    }
+
+    public JerseyModule(JerseyConfiguration jerseyConfiguration, Consumer<Server> jettyConfigurer) {
         this.jerseyConfiguration = Objects.requireNonNull(jerseyConfiguration);
+        this.jettyConfigurer = jettyConfigurer;
     }
 
     protected void configure() {
         Provider<Injector> injectorProvider = getProvider(Injector.class);
 
         install(new ServletModule());
-        bind(JerseyServer.class).toInstance(new JerseyServer(jerseyConfiguration, injectorProvider::get));
+        bind(JerseyServer.class).toInstance(new JerseyServer(jerseyConfiguration, injectorProvider::get, jettyConfigurer));
         bind(JerseyConfiguration.class).toInstance(jerseyConfiguration);
     }
 

diff --git a/src/main/java/io/logz/guice/jersey/JerseyServer.java b/src/main/java/io/logz/guice/jersey/JerseyServer.java
@@ -16,6 +16,7 @@
 import javax.servlet.DispatcherType;
 import java.util.EnumSet;
 import java.util.List;
+import java.util.function.Consumer;
 import java.util.function.Supplier;
 
 public class JerseyServer {
@@ -28,9 +29,16 @@ public class JerseyServer {
 
     JerseyServer(JerseyConfiguration jerseyConfiguration,
                  Supplier<Injector> injectorSupplier) {
+        this(jerseyConfiguration, injectorSupplier, jettyServer -> {});
+    }
+
+    JerseyServer(JerseyConfiguration jerseyConfiguration,
+                 Supplier<Injector> injectorSupplier,
+                 Consumer<Server> jettyConfigurer) {
         this.jerseyConfiguration = jerseyConfiguration;
         this.injectorSupplier = injectorSupplier;
         this.server = new Server();
+        jettyConfigurer.accept(server);
 
         configureServer();
     }