• Home
• software.amazon.smithy.kotlin
• smithy-kotlin-codegen
• 0.7.6
• source code
• StandardRetryMiddleware.kt

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

software.amazon.smithy.kotlin.codegen.retries.StandardRetryMiddleware.kt Maven / Gradle / Ivy

/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0.
 */

package software.amazon.smithy.kotlin.codegen.retries

import software.amazon.smithy.kotlin.codegen.core.KotlinWriter
import software.amazon.smithy.kotlin.codegen.core.RuntimeTypes
import software.amazon.smithy.kotlin.codegen.rendering.protocol.ProtocolGenerator
import software.amazon.smithy.kotlin.codegen.rendering.protocol.ProtocolMiddleware
import software.amazon.smithy.model.shapes.OperationShape

/**
 * Adds retry wrappers around operation invocations.
 */
class StandardRetryMiddleware : ProtocolMiddleware {
    override val name: String = RuntimeTypes.Http.Middlware.Retry.name

    override fun render(ctx: ProtocolGenerator.GenerationContext, op: OperationShape, writer: KotlinWriter) {
        writer.addImport(RuntimeTypes.Http.Middlware.Retry)
        writer.addImport(RuntimeTypes.Core.Retries.Impl.StandardRetryPolicy)

        writer.write(
            "op.install(#T(config.retryStrategy, StandardRetryPolicy.Default))",
            RuntimeTypes.Http.Middlware.Retry
        )
    }
}