misc: update isEnabledFor of s3-200-errors customization (#1346)

awslabs · Jul 3, 2024 · e0bd701 · e0bd701
1 parent de6e8f8
commit e0bd701
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...rc/main/kotlin/aws/sdk/kotlin/codegen/customization/s3/S3ErrorWith200StatusIntegration.kt b/...rc/main/kotlin/aws/sdk/kotlin/codegen/customization/s3/S3ErrorWith200StatusIntegration.kt
@@ -38,7 +38,11 @@ private object S3HandleError200ResponseMiddleware : ProtocolMiddleware {
         // Instead of playing whack-a-mole broadly apply this interceptor to everything but streaming responses
         // which adds a small amount of overhead to response processing.
         val output = ctx.model.expectShape(op.output.get())
-        return output.members().none { it.isStreaming || ctx.model.expectShape(it.target).isStreaming }
+        return output.members().none {
+            val isBlob = it.isBlobShape || ctx.model.expectShape(it.target).isBlobShape
+            val isStreaming = it.isStreaming || ctx.model.expectShape(it.target).isStreaming
+            isBlob && isStreaming
+        }
     }
 
     override val name: String = "Handle200ErrorsInterceptor"