Fixes filter to handle only beagle requests

Signed-off-by: Hernand Azevedo <[email protected]>
ZupIT · Sep 26, 2022 · 9b7da7b · 9b7da7b
1 parent d444a92
commit 9b7da7b
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 19 deletions.
diff --git a/...ters/micronaut/src/main/kotlin/br/com/zup/beagle/micronaut/filter/BeaglePlatformFilter.kt b/...ters/micronaut/src/main/kotlin/br/com/zup/beagle/micronaut/filter/BeaglePlatformFilter.kt
@@ -45,16 +45,22 @@ class BeaglePlatformFilter(private val objectMapper: ObjectMapper) : OncePerRequ
 
     @Suppress("UNCHECKED_CAST")
     private fun treatResponse(wrappedResponse: MutableHttpResponse<*>, currentPlatform: String?) {
-        wrappedResponse.body.ifPresent {
-            if (it !is NettySystemFileCustomizableResponseType && it !is String) {
-                val jsonTree = this.objectMapper.readTree(
-                    this.objectMapper.writeValueAsString(it)
-                )
-                BeaglePlatformUtil.treatBeaglePlatform(
-                    currentPlatform,
-                    jsonTree
-                )
-                (wrappedResponse as MutableHttpResponse<String>).body(this.objectMapper.writeValueAsString(jsonTree))
+        if (currentPlatform != null) {
+            wrappedResponse.body.ifPresent {
+                if (it !is NettySystemFileCustomizableResponseType && it !is String) {
+                    val jsonTree = this.objectMapper.readTree(
+                        this.objectMapper.writeValueAsString(it)
+                    )
+                    BeaglePlatformUtil.treatBeaglePlatform(
+                        currentPlatform,
+                        jsonTree
+                    )
+                    (wrappedResponse as MutableHttpResponse<String>).body(
+                        this.objectMapper.writeValueAsString(
+                            jsonTree
+                        )
+                    )
+                }
             }
         }
     }

diff --git a/starters/spring/src/main/kotlin/br/com/zup/beagle/spring/filter/BeaglePlatformFilter.kt b/starters/spring/src/main/kotlin/br/com/zup/beagle/spring/filter/BeaglePlatformFilter.kt
@@ -29,20 +29,29 @@ import javax.servlet.http.HttpServletResponse
 
 class BeaglePlatformFilter(private val objectMapper: ObjectMapper) : Filter {
 
-    override fun doFilter(request: ServletRequest?, response: ServletResponse?, chain: FilterChain?) {
-        if (chain != null && request is HttpServletRequest && response is HttpServletResponse) {
-            request.setAttribute(
-                BeaglePlatformUtil.BEAGLE_PLATFORM_HEADER,
-                request.getHeader(BeaglePlatformUtil.BEAGLE_PLATFORM_HEADER)
-            )
-            val responseWrapper = ContentCachingResponseWrapper(response)
+    override fun doFilter(
+        request: ServletRequest?,
+        response: ServletResponse?,
+        chain: FilterChain?
+    ) {
+        val httpServletResponse = response as? HttpServletResponse
+        val httpServletRequest = request as? HttpServletRequest
+        val platformHeader = httpServletRequest?.getHeader(BeaglePlatformUtil.BEAGLE_PLATFORM_HEADER)
+        val isServletRequestResponse = request is HttpServletRequest && response is HttpServletResponse
+
+        if (chain != null && isServletRequestResponse && platformHeader != null) {
+            request.setAttribute(BeaglePlatformUtil.BEAGLE_PLATFORM_HEADER, platformHeader)
+            val responseWrapper = ContentCachingResponseWrapper(httpServletResponse)
             chain.doFilter(request, responseWrapper)
-            treatResponse(responseWrapper, request.getHeader(BeaglePlatformUtil.BEAGLE_PLATFORM_HEADER))
+            treatResponse(responseWrapper, platformHeader)
             responseWrapper.copyBodyToResponse()
         }
     }
 
-    private fun treatResponse(responseWrapper: ContentCachingResponseWrapper, currentPlatform: String?) {
+    private fun treatResponse(
+        responseWrapper: ContentCachingResponseWrapper,
+        currentPlatform: String?
+    ) {
         if (responseWrapper.contentType == MediaType.APPLICATION_JSON_VALUE) {
             val jsonData = this.objectMapper.readTree(responseWrapper.contentAsByteArray).also {
                 BeaglePlatformUtil.treatBeaglePlatform(currentPlatform, it)