Merge pull request eclipse-vertx#433 from EmadAlblueshi/routing-conte…

…xt-remove

add RoutingContext#remove

vertx-web/src/main/java/io/vertx/ext/web/RoutingContext.java

@@ -119,6 +119,16 @@ public interface RoutingContext {
    */
   <T> T get(String key);
 
+  /**
+   * Remove some data from the context. The data is available in any handlers that receive the context.
+   *
+   * @param key  the key for the data
+   * @param <T>  the type of the data
+   * @return  the previous data associated with the key
+   * @throws ClassCastException if the data is not of the expected type
+   */
+  <T> T remove(String key);
+
   /**
    * @return all the context data as a map
    */

vertx-web/src/main/java/io/vertx/ext/web/impl/RoutingContextDecorator.java

@@ -102,6 +102,11 @@ public <T> T get(String key) {
     return decoratedContext.get(key);
   }
 
+  @Override
+  public <T> T remove(String key) {
+    return decoratedContext.remove(key);
+  }
+
   @Override
   public String getAcceptableContentType() {
     return decoratedContext.getAcceptableContentType();

vertx-web/src/main/java/io/vertx/ext/web/impl/RoutingContextImpl.java

@@ -145,6 +145,13 @@ public <T> T get(String key) {
     return (T)obj;
   }
 
+  @Override
+  @SuppressWarnings("unchecked")
+  public <T> T remove(String key) {
+    Object obj = getData().remove(key);
+    return (T)obj;
+  }
+
   @Override
   public Map<String, Object> data() {
     return getData();

vertx-web/src/main/java/io/vertx/ext/web/impl/RoutingContextWrapper.java

@@ -83,6 +83,11 @@ public <T> T get(String key) {
     return inner.get(key);
   }
 
+  @Override
+  public <T> T remove(String key) {
+    return inner.remove(key);
+  }
+
   @Override
   public Map<String, Object> data() {
     return inner.data();

vertx-web/src/test/java/io/vertx/ext/web/impl/RoutingContextImplTest.java

@@ -83,4 +83,20 @@ public void test_object_as_json_yields_json_object() throws Exception {
         }, HttpResponseStatus.OK.code(), HttpResponseStatus.OK.reasonPhrase(), null);
     }
 
+    @Test
+    public void test_remove_data() throws Exception {
+        router.route().handler(event -> {
+            String foo = event.getBodyAsJson().encode();
+            event.put("foo", foo);
+            String removedFoo = event.remove("foo");
+            assertEquals(removedFoo, foo);
+            assertNull(event.get("foo"));
+            event.response().end();
+        });
+        testRequest(HttpMethod.POST, "/", req -> {
+            req.setChunked(true);
+            req.write(Buffer.buffer("{ \"foo\": \"bar\" }"));
+        }, HttpResponseStatus.OK.code(), HttpResponseStatus.OK.reasonPhrase(), null);
+    }
+
 }