-
Notifications
You must be signed in to change notification settings - Fork 197
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
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 --> awslabs/aws-sdk-rust#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._
- Loading branch information
1 parent
4230687
commit d288ef9
Showing
3 changed files
with
59 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,12 @@ | ||
--- | ||
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. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
45 changes: 45 additions & 0 deletions
45
...tware/amazon/smithy/rustsdk/customize/transcribestreaming/TranscribeStreamingDecorator.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,45 @@ | ||
/* | ||
* 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() | ||
} | ||
} | ||
} |