with operator for routes fix #345

Author: jknack

coverage-report/src/test/java/org/jooby/issues/Issue345.java

@@ -0,0 +1,34 @@
+package org.jooby.issues;
+
+import org.jooby.test.ServerFeature;
+import org.junit.Test;
+
+public class Issue345 extends ServerFeature {
+
+  {
+    get("/a", req -> req.route().attributes());
+
+    with(() -> {
+
+      get("/b", req -> req.route().attributes());
+
+      get("/c", req -> req.route().attributes());
+
+    }).attr("foo", "bar");
+
+    get("/d", req -> req.route().attributes());
+  }
+
+  @Test
+  public void withOperator() throws Exception {
+    request().get("/a")
+        .expect("{}");
+    request().get("/b")
+        .expect("{foo=bar}");
+    request().get("/c")
+        .expect("{foo=bar}");
+    request().get("/d")
+        .expect("{}");
+  }
+
+}

jooby/src/main/java/org/jooby/Jooby.java

@@ -143,6 +143,7 @@
 import com.typesafe.config.ConfigValue;
 import com.typesafe.config.ConfigValueFactory;
 
+import javaslang.Predicates;
 import javaslang.control.Try;
 
 /**
@@ -3124,6 +3125,20 @@ public Route.Definition sse(final String path, final Sse.Handler1 handler) {
     return appendDefinition(new Route.Definition("GET", path, handler)).consumes(MediaType.sse);
   }
 
+  @Override
+  public Route.Collection with(final Runnable callback) {
+    // hacky way of doing what we want... but we do simplify developer life
+    int size = this.bag.size();
+    callback.run();
+    // collect latest routes and apply collection attr
+    List<Route.Definition> local = this.bag.stream()
+        .skip(size)
+        .filter(Predicates.instanceOf(Route.Definition.class))
+        .map(r -> (Route.Definition) r)
+        .collect(Collectors.toList());
+    return new Route.Collection(local.toArray(new Route.Definition[local.size()]));
+  }
+
   /**
    * <h1>Bootstrap</h1>
    * <p>