Fix GraphQL errors feature for streaming content types

Author: kikoveiga

integrations/dd-sdk-android-okhttp/src/main/kotlin/com/datadog/android/okhttp/DatadogInterceptor.kt

@@ -14,6 +14,7 @@ import com.datadog.android.api.feature.FeatureSdkCore
 import com.datadog.android.core.InternalSdkCore
 import com.datadog.android.core.configuration.Configuration
 import com.datadog.android.core.sampling.Sampler
+import com.datadog.android.internal.network.HttpSpec
 import com.datadog.android.okhttp.internal.RumResourceAttributesProviderCompatibilityAdapter
 import com.datadog.android.okhttp.internal.buildResourceId
 import com.datadog.android.okhttp.internal.graphql.OkHttpGraphQLAdapter
@@ -295,8 +296,8 @@ open class DatadogInterceptor internal constructor(
                 // manually rebuild the mimetype as `toString()` can also include the charsets
                 it.type + "/" + it.subtype
             }
-            val isStream = contentType in STREAM_CONTENT_TYPES
-            val isWebSocket = !response.header(WEBSOCKET_ACCEPT_HEADER, null).isNullOrBlank()
+            val isStream = HttpSpec.ContentType.isStream(contentType)
+            val isWebSocket = !response.header(HttpSpec.Header.WEBSOCKET_ACCEPT_HEADER, null).isNullOrBlank()
             if (body == null || isStream || isWebSocket) {
                 return null
             }
@@ -437,15 +438,6 @@ open class DatadogInterceptor internal constructor(
     // endregion
 
     internal companion object {
-        internal val STREAM_CONTENT_TYPES = setOf(
-            "text/event-stream",
-            "application/grpc",
-            "application/grpc+proto",
-            "application/grpc+json"
-        )
-
-        internal const val WEBSOCKET_ACCEPT_HEADER = "Sec-WebSocket-Accept"
-
         internal const val WARN_RUM_DISABLED =
             "You set up a DatadogInterceptor for %s, but RUM features are disabled. " +
                 "Make sure you initialized the Datadog SDK with a valid Application Id, " +

integrations/dd-sdk-android-okhttp/src/main/kotlin/com/datadog/android/okhttp/internal/graphql/OkHttpGraphQLAdapter.kt

@@ -9,6 +9,7 @@ package com.datadog.android.okhttp.internal.graphql
 import androidx.annotation.WorkerThread
 import com.datadog.android.api.InternalLogger
 import com.datadog.android.internal.network.GraphQLHeaders
+import com.datadog.android.internal.network.HttpSpec
 import com.datadog.android.okhttp.internal.OkHttpHttpResponseInfo
 import com.datadog.android.okhttp.internal.OkHttpRequestInfo
 import com.datadog.android.rum.RumAttributes
@@ -68,7 +69,13 @@ internal class OkHttpGraphQLAdapter(
         graphqlAttributes: Map<String, Any?>,
         internalLogger: InternalLogger
     ): Map<String, Any> {
-        if (graphqlAttributes.isEmpty()) return emptyMap()
+        // Streaming responses surface GraphQL errors per-frame, not as a top-level `errors` array.
+        // Draining their bodies via peekBody().string() would block until the (potentially unbounded) body completes.
+        val body = response.body
+        val mimeType = body?.contentType()?.let { it.type + "/" + it.subtype }
+        val isStream = HttpSpec.ContentType.isStream(mimeType)
+        val isWebSocket = !response.header(HttpSpec.Header.WEBSOCKET_ACCEPT_HEADER, null).isNullOrBlank()
+        if (graphqlAttributes.isEmpty() || body == null || isStream || isWebSocket) return emptyMap()
         return try {
             val responseInfo = OkHttpHttpResponseInfo(response, internalLogger)
 

integrations/dd-sdk-android-okhttp/src/test/kotlin/com/datadog/android/okhttp/DatadogInterceptorTest.kt

@@ -303,7 +303,9 @@ internal class DatadogInterceptorTest : TracingInterceptorNotSendingSpanTest() {
         @IntForgery(min = 200, max = 300) statusCode: Int
     ) {
         // Given
-        val mimeType = forge.anElementFrom(DatadogInterceptor.STREAM_CONTENT_TYPES)
+        val mimeType = forge.anElementFrom(
+            HttpSpec.ContentType.values().filter(HttpSpec.ContentType::isStream)
+        )
         fakeMediaType = mimeType.toMediaType()
         stubChain(mockChain, statusCode)
         val expectedStartAttrs = emptyMap<String, Any?>()

integrations/dd-sdk-android-okhttp/src/test/kotlin/com/datadog/android/okhttp/internal/graphql/OkHttpGraphQLAdapterTest.kt

@@ -275,5 +275,35 @@ internal class OkHttpGraphQLAdapterTest {
         )
     }
 
+    @Test
+    fun `M return empty and not read body W extractGraphQLErrorAttributes() {streaming response}`(
+        @StringForgery fakeOpName: String
+    ) {
+        // Given
+        val streamContentType = forge.anElementFrom(
+            HttpSpec.ContentType.values().filter(HttpSpec.ContentType::isStream)
+        )
+        val unreadableBody = object : ResponseBody() {
+            override fun contentType(): MediaType = streamContentType.toMediaType()
+            override fun contentLength(): Long = -1L
+            override fun source(): BufferedSource {
+                throw AssertionError(
+                    "extractGraphQLErrorAttributes must not read body of streaming response"
+                )
+            }
+        }
+        val request = Request.Builder()
+            .url("https://example.com/graphql")
+            .build()
+        val response = forge.anOkHttpResponse(request, 200) { body(unreadableBody) }
+        val graphqlAttributes = mapOf<String, Any?>(RumAttributes.GRAPHQL_OPERATION_NAME to fakeOpName)
+
+        // When
+        val result = testedHelper.extractGraphQLErrorAttributes(response, graphqlAttributes, mockInternalLogger)
+
+        // Then
+        assertThat(result).isEmpty()
+    }
+
     // endregion
 }