Improve possible extension of filter class.

Chavaillaz · Jan 20, 2024 · 201b777 · 201b777
1 parent 5ba9520
commit 201b777
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/main/java/com/chavaillaz/jakarta/rs/LoggedFilter.java b/src/main/java/com/chavaillaz/jakarta/rs/LoggedFilter.java
@@ -97,8 +97,8 @@ protected static <A extends Annotation> Optional<A> getAnnotation(ResourceInfo r
      *
      * @return The annotation type
      */
-    protected Class<? extends Logged> getAnnotation() {
-        return Logged.class;
+    protected Optional<Logged> getAnnotation() {
+        return getAnnotation(resourceInfo, Logged.class);
     }
 
     /**
@@ -167,7 +167,7 @@ public void filter(ContainerRequestContext requestContext, ContainerResponseCont
      * @param requestContext The context of the request
      */
     protected void logRequestBody(ContainerRequestContext requestContext) {
-        getAnnotation(resourceInfo, getAnnotation())
+        getAnnotation()
                 .map(Logged::requestBody)
                 .filter(loggingActivated -> loggingActivated && requestContext.hasEntity())
                 .ifPresent(logging -> MDC.put(REQUEST_BODY, getRequestBody(requestContext)));
@@ -179,7 +179,7 @@ protected void logRequestBody(ContainerRequestContext requestContext) {
      * @param responseContext THe context of the response
      */
     protected void logResponseBody(ContainerResponseContext responseContext) {
-        getAnnotation(resourceInfo, getAnnotation())
+        getAnnotation()
                 .map(Logged::responseBody)
                 .filter(loggingActivated -> loggingActivated && responseContext.hasEntity())
                 .ifPresent(logging -> MDC.put(RESPONSE_BODY, getResponseBody(responseContext)));