Unverified Commit d288ef9d authored by Landon James's avatar Landon James Committed by GitHub
Browse files

Remove stalled stream protection from transcribestreaming (#3831) 

## Motivation and Context
<!--- Why is this change required? What problem does it solve? -->
<!--- If it fixes an open issue, please link to the issue here -->
https://github.com/awslabs/aws-sdk-rust/issues/1181

## Checklist
- [x] For changes to the AWS SDK, generated SDK code, or SDK runtime
crates, I have created a changelog entry Markdown file in the
`.changelog` directory, specifying "aws-sdk-rust" in the `applies_to`
key.

----

_By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice._
parent 4230687c

.changelog/removessptranscribestreaming.md

0 → 100644
+12 −0
Original line number Diff line number Diff line 
---

applies_to:

  - aws-sdk-rust

authors:

  - landonxjames

references: ["aws-sdk-rust#1181"]

breaking: false

new_feature: false

bug_fix: true

---



Remove stalled stream protection from transcribe-streaming operations.

aws/sdk-codegen/src/main/kotlin/software/amazon/smithy/rustsdk/AwsCodegenDecorator.kt

+2 −0
Original line number Diff line number Diff line
@@ -28,6 +28,7 @@ import software.amazon.smithy.rustsdk.customize.s3control.S3ControlDecorator 
import software.amazon.smithy.rustsdk.customize.sso.SSODecorator

import software.amazon.smithy.rustsdk.customize.sts.STSDecorator

import software.amazon.smithy.rustsdk.customize.timestream.TimestreamDecorator

import software.amazon.smithy.rustsdk.customize.transcribestreaming.TranscribeStreamingDecorator

import software.amazon.smithy.rustsdk.endpoints.AwsEndpointsStdLib

import software.amazon.smithy.rustsdk.endpoints.OperationInputTestDecorator

import software.amazon.smithy.rustsdk.endpoints.RequireEndpointRules
@@ -88,6 +89,7 @@ val DECORATORS: List<ClientCodegenDecorator> = 
        SSODecorator().onlyApplyTo("com.amazonaws.sso#SWBPortalService"),

        TimestreamDecorator().onlyApplyTo("com.amazonaws.timestreamwrite#Timestream_20181101"),

        TimestreamDecorator().onlyApplyTo("com.amazonaws.timestreamquery#Timestream_20181101"),

        TranscribeStreamingDecorator().onlyApplyTo("com.amazonaws.transcribestreaming#Transcribe"),

        // Only build docs-rs for linux to reduce load on docs.rs

        listOf(

            DocsRsMetadataDecorator(

aws/sdk-codegen/src/main/kotlin/software/amazon/smithy/rustsdk/customize/transcribestreaming/TranscribeStreamingDecorator.kt

0 → 100644
+45 −0
Original line number Diff line number Diff line 
/*

 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.

 * SPDX-License-Identifier: Apache-2.0

 */



package software.amazon.smithy.rustsdk.customize.transcribestreaming



import software.amazon.smithy.model.Model

import software.amazon.smithy.model.shapes.OperationShape

import software.amazon.smithy.model.shapes.ServiceShape

import software.amazon.smithy.model.shapes.ShapeId

import software.amazon.smithy.model.transform.ModelTransformer

import software.amazon.smithy.rust.codegen.client.smithy.ClientRustSettings

import software.amazon.smithy.rust.codegen.client.smithy.customize.ClientCodegenDecorator

import software.amazon.smithy.rust.codegen.client.smithy.traits.IncompatibleWithStalledStreamProtectionTrait

import software.amazon.smithy.rust.codegen.core.util.letIf

import java.util.logging.Logger



/**

 * Top level decorator for TranscribeStreaming

 */

class TranscribeStreamingDecorator : ClientCodegenDecorator {

    private val operationsIncompatibleWithStalledStreamProtection =

        setOf(

            ShapeId.from("com.amazonaws.transcribestreaming#StartCallAnalyticsStreamTranscription"),

            ShapeId.from("com.amazonaws.transcribestreaming#StartMedicalStreamTranscription"),

            ShapeId.from("com.amazonaws.transcribestreaming#StartStreamTranscription"),

        )



    override val name: String = "TranscribeStreamingDecorator"

    override val order: Byte = 0

    private val logger = Logger.getLogger(javaClass.name)



    override fun transformModel(

        service: ServiceShape,

        model: Model,

        settings: ClientRustSettings,

    ): Model =

        ModelTransformer.create().mapShapes(model) { shape ->

            shape.letIf(shape.id in operationsIncompatibleWithStalledStreamProtection) {

                logger.info("Adding IncompatibleWithStalledStreamProtection trait to $it")

                (it as OperationShape).toBuilder().addTrait(IncompatibleWithStalledStreamProtectionTrait()).build()

            }

        }

}