Add RequestContextResolver Feature (#204)

* now adds request resolver

* Update BaseControllerWriter.java

* add for other generators

* works

* lowercase b

* Update JsonBUtil.java

* pr comment

* add ServerContext class to hold both req and res

* prevent rebinding context

* add hibernate plugin

Author: SentryMan

.github/workflows/build.yml

@@ -10,7 +10,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        java_version: [11, 17, 19]
+        java_version: [11, 17, 20]
         os: [ubuntu-latest]
 
     steps:

http-api/pom.xml

@@ -15,4 +15,26 @@
     <tag>avaje-http-parent-1.19</tag>
   </scm>
 
+  <dependencies>
+    <dependency>
+      <groupId>io.avaje</groupId>
+      <artifactId>avaje-inject</artifactId>
+      <version>9.0</version>
+      <scope>provided</scope>
+      <optional>true</optional>
+    </dependency>
+  </dependencies>
+
+  <build>
+    <plugins>
+      <plugin>
+        <groupId>org.apache.maven.plugins</groupId>
+        <artifactId>maven-surefire-plugin</artifactId>
+        <version>3.0.0</version>
+        <configuration>
+          <forkCount>0</forkCount>
+        </configuration>
+      </plugin>
+    </plugins>
+  </build>
 </project>

http-api/src/main/java/io/avaje/http/api/Controller.java

@@ -23,4 +23,7 @@
 
   /** Specify the path mapping request to the controller. */
   String value() default "";
+
+  /** Specify if the http request context should be instrumented via RequestContextResolver */
+  boolean instrumentRequestContext() default false;
 }

http-api/src/main/java/io/avaje/http/api/Delete.java

@@ -19,13 +19,13 @@
  *
  * }</pre>
  */
-@Target(value=METHOD)
-@Retention(value=RUNTIME)
-@HttpMethod(value="DELETE")
+@Target(METHOD)
+@Retention(RUNTIME)
+@HttpMethod("DELETE")
 public @interface Delete {
 
-  /**
-   * Specify the path.
-   */
+  /** Specify the path. */
   String value() default "";
+
+  boolean instrumentRequestContext() default false;
 }

http-api/src/main/java/io/avaje/http/api/Get.java

@@ -49,9 +49,13 @@
  *
  * }</pre>
  */
-@Target(value=METHOD)
-@Retention(value=RUNTIME)
-@HttpMethod(value="GET")
+@Target(METHOD)
+@Retention(RUNTIME)
+@HttpMethod("GET")
 public @interface Get {
+
+  /** Specify the path. */
   String value() default "";
+
+  boolean instrumentRequestContext() default false;
 }

http-api/src/main/java/io/avaje/http/api/Patch.java

@@ -1,17 +1,19 @@
 package io.avaje.http.api;
 
-import java.lang.annotation.Retention;
-import java.lang.annotation.Target;
-
 import static java.lang.annotation.ElementType.METHOD;
 import static java.lang.annotation.RetentionPolicy.RUNTIME;
 
-/**
- * Marks a method that handles HTTP PATCH requests.
- */
-@Target(value = METHOD)
-@Retention(value = RUNTIME)
-@HttpMethod(value = "PATCH")
+import java.lang.annotation.Retention;
+import java.lang.annotation.Target;
+
+/** Marks a method that handles HTTP PATCH requests. */
+@Target(METHOD)
+@Retention(RUNTIME)
+@HttpMethod("PATCH")
 public @interface Patch {
+
+  /** Specify the path. */
   String value() default "";
+
+  boolean instrumentRequestContext() default false;
 }

http-api/src/main/java/io/avaje/http/api/Post.java

@@ -16,9 +16,13 @@
       ...
  *  }
  */
-@Target(value=METHOD)
-@Retention(value=RUNTIME)
-@HttpMethod(value="POST")
+@Target(METHOD)
+@Retention(RUNTIME)
+@HttpMethod("POST")
 public @interface Post {
+
+  /** Specify the path. */
   String value() default "";
+
+  boolean instrumentRequestContext() default false;
 }

http-api/src/main/java/io/avaje/http/api/Put.java

@@ -9,9 +9,13 @@
 /**
  * Marks a method that handles HTTP PUT requests.
  */
-@Target(value=METHOD)
-@Retention(value=RUNTIME)
-@HttpMethod(value="PUT")
+@Target(METHOD)
+@Retention(RUNTIME)
+@HttpMethod("PUT")
 public @interface Put {
+
+  /** Specify the path. */
   String value() default "";
+
+  boolean instrumentRequestContext() default false;
 }

http-api/src/main/java/io/avaje/http/api/context/RequestContextResolver.java

@@ -0,0 +1,49 @@
+package io.avaje.http.api.context;
+
+import java.util.Optional;
+import java.util.concurrent.Callable;
+import java.util.function.Supplier;
+
+/**
+ * The holder for the current request context that is bound to instrumented threads. Allowing lookup
+ * of the current request if it is present. The Default implementation uses ThreadLocal. If you are
+ * able, you should provide an implementation using ScopedValues.
+ */
+public interface RequestContextResolver {
+
+  /**
+   * Wraps the execution of the given callable in request context processing.
+   *
+   * @param ctx The request context
+   * @param callable The callable
+   * @param <T> The return type of the callable
+   * @return The return value of the callable
+   * @throws Exception if the callable throws an exception
+   */
+  <T> T callWith(ServerContext ctx, Callable<T> callable) throws Exception;
+
+  /**
+   * Wraps the execution of the given supplier in request context processing.
+   *
+   * @param ctx The request context
+   * @param supplier The supplier
+   * @param <T> The return type of the supplier
+   * @return The return value of the supplier
+   */
+  <T> T supplyWith(ServerContext ctx, Supplier<T> supplier);
+
+  /**
+   * Wraps the execution of the given runnable in request context processing.
+   *
+   * @param ctx The request context
+   * @param runnable The runnable
+   */
+  void runWith(ServerContext request, Runnable runnable);
+
+  /**
+   * Retrieve the current server context.
+   *
+   * @return The request context if it is present
+   */
+  Optional<ServerContext> currentRequest();
+}

http-api/src/main/java/io/avaje/http/api/context/ServerContext.java

@@ -0,0 +1,31 @@
+package io.avaje.http.api.context;
+
+/** Holder for the Server Request/Response Classes */
+public class ServerContext {
+
+  private final Object request;
+  private final Object response;
+
+  public ServerContext(Object req, Object res) {
+    request = req;
+    response = res;
+  }
+
+  /**
+   * Retrieve the current server request.
+   *
+   * @return The request
+   */
+  <T> T request() {
+    return (T) request;
+  }
+
+  /**
+   * Retrieve the current server response.
+   *
+   * @return The request
+   */
+  <T> T response() {
+    return (T) response;
+  }
+}