From 3873ef57b1c21b8be462789a540bfddf8de3362a Mon Sep 17 00:00:00 2001 From: Harry Barber Date: Fri, 18 Nov 2022 15:14:55 +0000 Subject: [PATCH] Re-export service from root --- .../server/smithy/generators/ServerServiceGenerator.kt | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/generators/ServerServiceGenerator.kt b/codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/generators/ServerServiceGenerator.kt index 0715629d53..ca6bd0fe5a 100644 --- a/codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/generators/ServerServiceGenerator.kt +++ b/codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/generators/ServerServiceGenerator.kt @@ -12,6 +12,7 @@ import software.amazon.smithy.rust.codegen.core.rustlang.RustMetadata import software.amazon.smithy.rust.codegen.core.rustlang.RustModule import software.amazon.smithy.rust.codegen.core.rustlang.RustWriter import software.amazon.smithy.rust.codegen.core.rustlang.Visibility +import software.amazon.smithy.rust.codegen.core.rustlang.rust import software.amazon.smithy.rust.codegen.core.smithy.CodegenContext import software.amazon.smithy.rust.codegen.core.smithy.RustCrate import software.amazon.smithy.rust.codegen.core.smithy.generators.protocol.ProtocolSupport @@ -40,6 +41,12 @@ open class ServerServiceGenerator( * which assigns a symbol location to each shape. */ fun render() { + rustCrate.lib { + val serviceName = codegenContext.serviceShape.id.name.toString() + rust("##[doc(inline, hidden)]") + rust("pub use crate::service::$serviceName;") + } + rustCrate.withModule(RustModule.operation(Visibility.PRIVATE)) { ServerProtocolTestGenerator(codegenContext, protocolSupport, protocolGenerator).render(this) }